aboutsummaryrefslogtreecommitdiff
path: root/migrations/auth/20240606_03_BY7Us-drop-group-roles-table.py
blob: 5add4d28100b5ec2a216a9cbd5af86a985d27484 (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
"""
Drop 'group_roles' table.
"""
import sqlite3
from yoyo import step

__depends__ = {'20240606_02_ubZri-create-resource-roles-table'}

def drop_group_roles(conn):
    """Delete `group_roles` and related data."""
    conn.row_factory = sqlite3.Row
    cursor = conn.cursor()
    cursor.execute("DELETE FROM user_roles WHERE role_id IN "
                   "(SELECT role_id FROM group_roles)")
    cursor.execute("DELETE FROM roles WHERE role_id IN"
                 "(SELECT role_id FROM group_roles)")
    cursor.execute("DROP TABLE IF EXISTS group_roles")
    cursor.close()

def restore_group_roles(conn):
    """Restore the `group_roles` table."""
    
    conn.row_factory = sqlite3.Row
    cursor = conn.cursor()
    cursor.execute(
        """
        CREATE TABLE group_roles(
          group_role_id TEXT PRIMARY KEY,
          group_id TEXT NOT NULL,
          role_id TEXT NOT NULL,
          UNIQUE (group_id, role_id),
          FOREIGN KEY(group_id) REFERENCES groups(group_id)
            ON UPDATE CASCADE ON DELETE RESTRICT,
          FOREIGN KEY(role_id) REFERENCES roles(role_id)
            ON UPDATE CASCADE ON DELETE RESTRICT
        ) WITHOUT ROWID
        """)
    cursor.execute(
        """
        CREATE INDEX idx_tbl_group_roles_cols_group_id
        ON group_roles(group_id)
        """)
    cursor.close()

steps = [
    step(drop_group_roles, restore_group_roles)
]