summaryrefslogtreecommitdiffstats
path: root/gitsrht/alembic/versions/8d26b98e7d44_add_cascades_for_repo_deletion.py
blob: 212044b059f0ce6a7385b188d30bc292eead6093 (plain)
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
83
84
85
86
87
88
89
90
91
92
93
94
"""Add cascades for repo deletion

Revision ID: 8d26b98e7d44
Revises: ce3a03ec34a5
Create Date: 2018-07-11 07:15:41.973647

"""

# revision identifiers, used by Alembic.
revision = '8d26b98e7d44'
down_revision = 'ce3a03ec34a5'

from alembic import op
import sqlalchemy as sa


def upgrade():
    op.drop_constraint(
            constraint_name="redirect_new_repo_id_fkey",
            table_name="redirect",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="redirect_new_repo_id_fkey",
            source_table="redirect",
            referent_table="repository",
            local_cols=["new_repo_id"],
            remote_cols=["id"],
            ondelete="CASCADE")
    op.drop_constraint(
            constraint_name="access_repo_id_fkey",
            table_name="access",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="access_repo_id_fkey",
            source_table="access",
            referent_table="repository",
            local_cols=["repo_id"],
            remote_cols=["id"],
            ondelete="CASCADE")
    op.drop_constraint(
            constraint_name="webhook_repo_id_fkey",
            table_name="webhook",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="webhook_repo_id_fkey",
            source_table="webhook",
            referent_table="repository",
            local_cols=["repo_id"],
            remote_cols=["id"],
            ondelete="CASCADE")
    op.drop_constraint(
            constraint_name="webhook_oauth_token_id_fkey",
            table_name="webhook",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="webhook_oauth_token_id_fkey",
            source_table="webhook",
            referent_table="oauthtoken",
            local_cols=["oauth_token_id"],
            remote_cols=["id"],
            ondelete="CASCADE")


def downgrade():
    op.drop_constraint(
            constraint_name="redirect_new_repo_id_fkey",
            table_name="redirect",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="redirect_new_repo_id_fkey",
            source_table="redirect",
            referent_table="repository",
            local_cols=["new_repo_id"],
            remote_cols=["id"])
    op.drop_constraint(
            constraint_name="access_repo_id_fkey",
            table_name="access",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="access_repo_id_fkey",
            source_table="access",
            referent_table="repository",
            local_cols=["repo_id"],
            remote_cols=["id"])
    op.drop_constraint(
            constraint_name="webhook_repo_id_fkey",
            table_name="webhook",
            type_="foreignkey")
    op.create_foreign_key(
            constraint_name="webhook_repo_id_fkey",
            source_table="webhook",
            referent_table="repository",
            local_cols=["repo_id"],
            remote_cols=["id"])