fix

intelowlproject · Nov 22, 2024 · 2535db4 · 2535db4
1 parent 5b09461
commit 2535db4
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/api_app/analyzers_manager/migrations/0131_analyzer_config_vt_sample_download.py b/api_app/analyzers_manager/migrations/0131_analyzer_config_vt_sample_download.py
@@ -31,4 +31,4 @@ class Migration(migrations.Migration):
         ("api_app", "0064_vt_sample_download"),
     ]
 
-    # operations = [migrations.RunPython(migrate, reverse_migrate)]
+    operations = [migrations.RunPython(migrate, reverse_migrate)]
diff --git a/api_app/migrations/0064_vt_sample_download.py b/api_app/migrations/0064_vt_sample_download.py
@@ -14,7 +14,7 @@ def migrate(apps, schema_editor):
     )
 
     # visualizer python module
-    vt_sample_visualizer_python_module, _ = PythonModule.objects.get_or_create(
+    PythonModule.objects.get_or_create(
         module="sample_download.SampleDownload",
         base_path=PythonModuleBasePaths.Visualizer.value,
     )

diff --git a/api_app/playbooks_manager/migrations/0056_download_sample_vt.py b/api_app/playbooks_manager/migrations/0056_download_sample_vt.py
@@ -35,4 +35,4 @@ class Migration(migrations.Migration):
         ("analyzers_manager", "0131_analyzer_config_vt_sample_download"),
     ]
 
-    # operations = [migrations.RunPython(migrate, reverse_migrate)]
+    operations = [migrations.RunPython(migrate, reverse_migrate)]
diff --git a/api_app/visualizers_manager/migrations/0039_sample_download.py b/api_app/visualizers_manager/migrations/0039_sample_download.py
@@ -12,11 +12,11 @@ def migrate(apps, schema_editor):
         description="Download a sample",
         python_module=PythonModule.objects.get(module="sample_download.SampleDownload"),
     )
-    visualizer_download_sample.playbooks.set(
-        PlaybookConfig.objects.filter(
+    visualizer_download_sample.playbooks.add(
+        *PlaybookConfig.objects.filter(
             analyzers=AnalyzerConfig.objects.get(name="DownloadFileFromUri")
-        )
-        + PlaybookConfig.objects.filter(
+        ),
+        *PlaybookConfig.objects.filter(
             analyzers=AnalyzerConfig.objects.get(name="VirusTotalv3SampleDownload")
         )
     )
@@ -35,4 +35,4 @@ class Migration(migrations.Migration):
         ("playbooks_manager", "0056_download_sample_vt"),
     ]
 
-    # operations = [migrations.RunPython(migrate, reverse_migrate)]
+    operations = [migrations.RunPython(migrate, reverse_migrate)]