Codebase list python-faraday / 78dcaf8 faraday / migrations / versions / 9c4091d1a09b_create_agent_table.py
78dcaf8

Tree @78dcaf8 (Download .tar.gz)

9c4091d1a09b_create_agent_table.py @78dcaf8 — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
"""create agent table

Revision ID: 9c4091d1a09b
Revises: 0d216660da28
Create Date: 2019-05-22 19:17:31.444968+00:00

"""

from alembic import op
import sqlalchemy as sa

revision = '9c4091d1a09b'
down_revision = 'be89aa03e35e'
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        'agent',
        sa.Column('id', sa.Integer, primary_key=True),
        sa.Column('token', sa.Text, nullable=False, unique=True),
        sa.Column('name', sa.Text, nullable=False),
        sa.Column('active', sa.Boolean, nullable=False),
        sa.Column('workspace_id', sa.Integer, nullable=False),
        # metadata
        sa.Column('create_date', sa.DateTime),
        sa.Column('update_date', sa.DateTime),
        sa.Column('creator_id', sa.Integer),
        sa.Column('update_user_id', sa.Integer),
    )

    op.create_foreign_key(
        'agent_creator_id_fkey',
        'agent',
        'faraday_user', ['creator_id'], ['id']
    )

    op.create_foreign_key(
        'agent_update_user_id_fkey',
        'agent',
        'faraday_user', ['update_user_id'], ['id']
    )

    op.create_foreign_key(
        'agent_workspace_id_fkey',
        'agent',
        'workspace', ['workspace_id'], ['id']
    )

    op.create_table(
        'agent_schedule',
        sa.Column('id', sa.Integer, primary_key=True),
        sa.Column('description', sa.Text, nullable=False),
        sa.Column('crontab', sa.Text, nullable=False),
        sa.Column('timezone', sa.Text, nullable=False),
        sa.Column('active', sa.Boolean, nullable=False),
        sa.Column('creator_id', sa.Integer),
        sa.Column('update_user_id', sa.Integer),
        sa.Column('create_date', sa.DateTime),
        sa.Column('update_date', sa.DateTime),
        sa.Column('workspace_id', sa.Integer, nullable=False),
        sa.Column('agent_id', sa.Integer, nullable=False),
        sa.Column('last_run', sa.DateTime),
    )

    op.create_foreign_key(
        'agent_schedule_agent_id_fkey',
        'agent_schedule',
        'agent', ['agent_id'], ['id']
    )

    op.create_foreign_key(
        'agent_schedule_id_fkey',
        'agent_schedule',
        'workspace', ['workspace_id'], ['id']
    )


def downgrade():
    op.drop_table('agent_schedule')
    op.drop_table('agent')