Alter DNS malicious detector analyzers (#2753)

* alter_migration_file Signed-off-by: pranjalg1331 <pranjaloff13@gmail.com> * migration file Signed-off-by: pranjalg1331 <pranjaloff13@gmail.com> --------- Signed-off-by: pranjalg1331 <pranjaloff13@gmail.com>
intelowlproject · Feb 18, 2025 · 201390f · 201390f
1 parent 3198062
commit 201390f
Showing 1 changed file with 71 additions and 0 deletions.
diff --git a/api_app/analyzers_manager/migrations/0150_alter_dns_malicious_detectors.py b/api_app/analyzers_manager/migrations/0150_alter_dns_malicious_detectors.py
@@ -0,0 +1,71 @@
+from django.db import migrations
+
+
+def update_maximum_tlp(apps, schema_editor):
+    AnalyzerConfig = apps.get_model("analyzers_manager", "AnalyzerConfig")
+    try:
+        obj = AnalyzerConfig.objects.get(name="UltraDNS_Malicious_Detector")
+        obj.maximum_tlp = "CLEAR"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="DNS0_EU_Malicious_Detector")
+        obj.maximum_tlp = "CLEAR"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="Quad9_Malicious_Detector")
+        obj.maximum_tlp = "CLEAR"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="CloudFlare_Malicious_Detector")
+        obj.maximum_tlp = "CLEAR"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+
+def reverse_update_maximum_tlp(apps, schema_editor):
+    AnalyzerConfig = apps.get_model("analyzers_manager", "AnalyzerConfig")
+    try:
+        obj = AnalyzerConfig.objects.get(name="UltraDNS_Malicious_Detector")
+        obj.maximum_tlp = "AMBER"  # Reverting to the original value
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="DNS0_EU_Malicious_Detector")
+        obj.maximum_tlp = "AMBER"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="Quad9_Malicious_Detector")
+        obj.maximum_tlp = "AMBER"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+    try:
+        obj = AnalyzerConfig.objects.get(name="CloudFlare_Malicious_Detector")
+        obj.maximum_tlp = "AMBER"
+        obj.save()
+    except AnalyzerConfig.DoesNotExist:
+        pass
+
+
+class Migration(migrations.Migration):
+    dependencies = (("analyzers_manager", "0149_alter_die_analyzer"),)
+
+    operations = [
+        migrations.RunPython(update_maximum_tlp, reverse_update_maximum_tlp),
+    ]