add migration

Author: manuel-sommer

dojo/db_migrations/0229_merge_mobsf.py

@@ -0,0 +1,55 @@
+from django.db import migrations
+import logging
+
+
+logger = logging.getLogger(__name__)
+
+
+PARSER_REFERENCES = ['Mobsfscan Scan']
+
+
+def update_parser_test(test, parser_test_type) -> None:
+    if test.test_type.name in PARSER_REFERENCES or test.scan_type in PARSER_REFERENCES:
+        test.test_type = parser_test_type
+        test.scan_type = parser_test_type.name
+        test.save()
+
+
+# Update the found_by field to remove Mobsfscan Scan and add MobSF Scan
+def update_parser_finding(finding, newparser_test_type, parser_test_type) -> None:
+    # Check if nessus is in found by list and remove
+    if parser_test_type in finding.found_by.all():
+        finding.found_by.remove(parser_test_type.id)
+    # Check if tenable is already in list somehow before adding it
+    if newparser_test_type not in finding.found_by.all():
+        finding.found_by.add(newparser_test_type.id)
+    finding.save()
+
+
+# Update all finding objects that came from Mobsfscan Scan reports
+def forward_merge_parser(apps, schema_editor):
+    finding_model = apps.get_model('dojo', 'Finding')
+    test_type_model = apps.get_model('dojo', 'Test_Type')
+    # Get or create MobSF Scan Test Type and fetch the Mobsfscan Scan test types
+    newparser_test_type, _ = test_type_model.objects.get_or_create(name="MobSF Scan", defaults={"active": True})
+    parser_test_type = test_type_model.objects.filter(name="Mobsfscan Scan").first()
+    # Get all the findings found by Mobsfscan Scan
+    findings = finding_model.objects.filter(test__scan_type__in=PARSER_REFERENCES)
+    logger.warning(f'We identified {findings.count()} Mobsfscan Scan findings to migrate to MobSF Scan findings')
+    # Iterate over all findings and change
+    for finding in findings:
+        # Update the found by field
+        update_parser_finding(finding, newparser_test_type, parser_test_type)
+        # Update the test object
+        update_parser_test(finding.test, newparser_test_type)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('dojo', '0228_alter_jira_username_password'),
+    ]
+
+    operations = [
+        migrations.RunPython(forward_merge_parser),
+    ]