aboutsummaryrefslogtreecommitdiff
path: root/migrations/auth/20230912_01_BxrhE-add-system-resource.py
blob: 66c64619f150edd1e65bdc2c8961f1f670049863 (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
"""
Add 'system' resource.
"""

import uuid

import sqlite3
from yoyo import step

__depends__ = {'20230907_04_3LnrG-refactor-create-group-resources-table'}

def add_system_resource(conn):
    """Add a system resource."""
    conn.row_factory = sqlite3.Row
    cursor = conn.cursor()
    cursor.execute(
        "SELECT resource_category_id FROM resource_categories "
        "WHERE resource_category_key='system'")
    category_id = cursor.fetchone()["resource_category_id"]
    cursor.execute(
        "INSERT INTO "
        "resources(resource_id, resource_name, resource_category_id, public) "
        "VALUES(?, ?, ?, ?)",
        (str(uuid.uuid4()), "GeneNetwork System", category_id, "1"))

def delete_system_resource(conn):
    """Add a system resource."""
    conn.row_factory = sqlite3.Row
    cursor = conn.cursor()
    cursor.execute(
        "SELECT resource_category_id FROM resource_categories "
        "WHERE resource_category_key='system'")
    category_id = cursor.fetchone()["resource_category_id"]
    cursor.execute("DELETE FROM resources WHERE resource_category_id = ?",
                   (category_id,))

steps = [
    step(add_system_resource, delete_system_resource)
]