Added LoadFileSamePlaybook pivot

api_app/pivots_manager/migrations/0036_alter_extractedonenotefiles_loadfilesameplaybook.py

@@ -0,0 +1,48 @@
+from django.db import migrations
+
+
+def migrate(apps, schema_editor):
+    PythonModule = apps.get_model("api_app", "PythonModule")
+    Parameter = apps.get_model("api_app", "Parameter")
+    PluginConfig = apps.get_model("api_app", "PluginConfig")
+    PivotConfig = apps.get_model("pivots_manager", "PivotConfig")
+    pivot_to_update = PivotConfig.objects.get(name="ExtractedOneNoteFiles")
+    pm = PythonModule.objects.create(
+        health_check_schedule=None,
+        update_schedule=None,
+        module="load_file_same_playbook.LoadFileSamePlaybook",
+        base_path="api_app.pivots_manager.pivots",
+    )
+    param1 = Parameter.objects.create(
+        name="field_to_compare",
+        type="str",
+        description="Dotted path to the field",
+        is_secret=False,
+        required=True,
+        python_module=pm,
+    )
+    PluginConfig.objects.filter(pivot_config=pivot_to_update).delete()
+    pivot_to_update.python_module = pm
+    PluginConfig.objects.create(
+        parameter=param1,
+        value="stored_base64",
+        for_organization=False,
+        updated_at="2024-11-07T10:35:46.217160Z",
+        analyzer_config=None,
+        connector_config=None,
+        visualizer_config=None,
+        ingestor_config=None,
+        pivot_config=pivot_to_update,
+    )
+    pivot_to_update.full_clean()
+    pivot_to_update.save()
+
+
+class Migration(migrations.Migration):
+    atomic = False
+    dependencies = [
+        ("api_app", "0063_singleton_and_elastic_report"),
+        ("pivots_manager", "0035_pivot_config_phishingextractortoanalysis"),
+    ]
+
+    operations = [migrations.RunPython(migrate, migrations.RunPython.noop)]

api_app/pivots_manager/migrations/0037_alter_resubmitdownloadedfile_loadfilesameplaybook.py

@@ -0,0 +1,48 @@
+from django.db import migrations
+
+
+def migrate(apps, schema_editor):
+    PythonModule = apps.get_model("api_app", "PythonModule")
+    Parameter = apps.get_model("api_app", "Parameter")
+    PluginConfig = apps.get_model("api_app", "PluginConfig")
+    PivotConfig = apps.get_model("pivots_manager", "PivotConfig")
+    pivot_to_update = PivotConfig.objects.get(name="ResubmitDownloadedFile")
+    pm = PythonModule.objects.get(
+        health_check_schedule=None,
+        update_schedule=None,
+        module="load_file_same_playbook.LoadFileSamePlaybook",
+        base_path="api_app.pivots_manager.pivots",
+    )
+    param1 = Parameter.objects.get(
+        name="field_to_compare",
+        type="str",
+        description="Dotted path to the field",
+        is_secret=False,
+        required=True,
+        python_module=pm,
+    )
+    PluginConfig.objects.filter(pivot_config=pivot_to_update).delete()
+    pivot_to_update.python_module = pm
+    PluginConfig.objects.create(
+        parameter=param1,
+        value="stored_base64",
+        for_organization=False,
+        updated_at="2024-11-07T10:35:46.217160Z",
+        analyzer_config=None,
+        connector_config=None,
+        visualizer_config=None,
+        ingestor_config=None,
+        pivot_config=pivot_to_update,
+    )
+    pivot_to_update.full_clean()
+    pivot_to_update.save()
+
+
+class Migration(migrations.Migration):
+    atomic = False
+    dependencies = [
+        ("api_app", "0063_singleton_and_elastic_report"),
+        ("pivots_manager", "0036_alter_extractedonenotefiles_loadfilesameplaybook"),
+    ]
+
+    operations = [migrations.RunPython(migrate, migrations.RunPython.noop)]

api_app/pivots_manager/pivots/load_file_same_playbook.py

@@ -0,0 +1,15 @@
+from api_app.pivots_manager.models import PivotConfig
+from api_app.pivots_manager.pivots.load_file import LoadFile
+
+
+class LoadFileSamvmePlaybook(LoadFile):
+    field_to_compare: str
+
+    @classmethod
+    def update(cls) -> bool:
+        pass
+
+    def get_playbook_to_execute(self):
+        self._config: PivotConfig
+        # use the same playbook of the parent when resubmit a file
+        return self._job.get_root().playbook_to_execute