Migrations related to 9999881 (#2560)

Author: josh-feather

utils/db_migration/versions/add_on_delete_cascade_to_task_tags_.py

@@ -0,0 +1,33 @@
+# Copyright (C) 2010-2015 Cuckoo Foundation.
+# This file is part of Cuckoo Sandbox - http://www.cuckoosandbox.org
+# See the file 'docs/LICENSE' for copying permission.
+
+"""Add ON DELETE CASCADE to tasks_tags foreign keys
+
+Revision ID: 4e000e02a409
+Revises: c2bd0eb5e69d
+Create Date: 2025-04-11 09:58:42.957359
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '4e000e02a409'
+down_revision = 'c2bd0eb5e69d'
+
+from alembic import op
+
+
+def upgrade():
+     op.drop_constraint('tasks_tags_task_id_fkey', 'tasks_tags', type_='foreignkey')
+     op.create_foreign_key('tasks_tags_task_id_fkey', 'tasks_tags', 'tasks', ['task_id'], ['id'], ondelete='CASCADE')
+
+     op.drop_constraint('tasks_tags_tag_id_fkey', 'tasks_tags', type_='foreignkey')
+     op.create_foreign_key('tasks_tags_tag_id_fkey', 'tasks_tags', 'tags', ['tag_id'], ['id'], ondelete='CASCADE')
+
+
+def downgrade():
+     op.drop_constraint('tasks_tags_task_id_fkey', 'tasks_tags', type_='foreignkey')
+     op.create_foreign_key('tasks_tags_task_id_fkey', 'tasks_tags', 'tasks', ['task_id'], ['id'])
+
+     op.drop_constraint('tasks_tags_tag_id_fkey', 'tasks_tags', type_='foreignkey')
+     op.create_foreign_key('tasks_tags_tag_id_fkey', 'tasks_tags', 'tags', ['tag_id'], ['id'])

utils/db_migration/versions/from_0_6_to_1_1.py

@@ -34,7 +34,7 @@
     print("Unable to import alembic (install with `poetry run pip install alembic`)")
     sys.exit()
 
-sys.path.append(os.path.join("..", ".."))
+sys.path.append(os.path.join("..", "..", ".."))
 
 import lib.cuckoo.core.database as db
 from lib.cuckoo.common.config import Config

utils/db_migration/versions/from_1_1_to_1_2-added_states.py

@@ -34,7 +34,7 @@
     sys.exit()
 
 curdir = os.path.abspath(os.path.dirname(__file__))
-sys.path.append(os.path.join(curdir, "..", ".."))
+sys.path.append(os.path.join(curdir, "..", "..", ".."))
 
 import lib.cuckoo.core.database as db
 

utils/db_migration/versions/from_1_1_to_1_2-extend_file_type.py

@@ -22,7 +22,7 @@
 from alembic import op
 
 curdir = os.path.abspath(os.path.dirname(__file__))
-sys.path.append(os.path.join(curdir, "..", ".."))
+sys.path.append(os.path.join(curdir, "..", "..", ".."))
 
 import lib.cuckoo.core.database as db
 

utils/db_migration/versions/from_1_2_to_1_2-accuvant-add_statistics.py

@@ -34,7 +34,7 @@
     sys.exit()
 
 curdir = os.path.abspath(os.path.dirname(__file__))
-sys.path.append(os.path.join(curdir, "..", ".."))
+sys.path.append(os.path.join(curdir, "..", "..", ".."))
 
 import lib.cuckoo.core.database as db