aboutsummaryrefslogtreecommitdiff
path: root/migrations/auth/20230912_01_BxrhE-add-system-resource.py
diff options
context:
space:
mode:
Diffstat (limited to 'migrations/auth/20230912_01_BxrhE-add-system-resource.py')
-rw-r--r--migrations/auth/20230912_01_BxrhE-add-system-resource.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/migrations/auth/20230912_01_BxrhE-add-system-resource.py b/migrations/auth/20230912_01_BxrhE-add-system-resource.py
new file mode 100644
index 0000000..66c6461
--- /dev/null
+++ b/migrations/auth/20230912_01_BxrhE-add-system-resource.py
@@ -0,0 +1,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)
+]