Reprocess old suse advisory in next import cycle

Signed-off-by: Keshav Priyadarshi <[email protected]>

Author: keshav-space

vulnerabilities/migrations/0075_alter_vulnerabilityseverity_options_and_more.py

@@ -2,12 +2,14 @@
 
 from django.db import migrations, models
 from aboutcode.pipeline import LoopProgress
+from datetime import datetime
+from datetime import timezone
 
 
 class Migration(migrations.Migration):
     """
     Add url field to the VulnerabilitySeverity model and remove reference foreignkey
-    relationship. Also, add new M2M 'severities' field in Vulnerability and 
+    relationship. Also, add new M2M 'severities' field in Vulnerability and
     AffectedByPackageRelatedVulnerability relationship.
 
     To achieve this following model changes and data migrations are applied in chronological order:
@@ -16,10 +18,12 @@ class Migration(migrations.Migration):
     - Add 'severities' field to Vulnerability.
     - Add 'url' field to VulnerabilitySeverity.
     - Data migration to remove corrupted SUSE scores.
+    - Data migration to enable reprocessing of old suse Advisory on next import.
     - Data migration to populate new VulnerabilitySeverity url field using reference.
     - Data migration to populate Vulnerability 'severities' M2M relationship.
     - Delete VulnerabilitySeverity reference field.
     """
+
     def remove_inaccurate_suse_score(apps, schema_editor):
         """
         Remove inaccurate suse severity scores.
@@ -35,6 +39,22 @@ def reverse_remove_inaccurate_suse_score(apps, schema_editor):
         """Reverse data migration not needed for inaccurate severity scores."""
         pass
 
+    def reprocess_suse_advisory_on_next_import(apps, schema_editor):
+        """Clear `date_imported` on old suse advisory to enable reprocessing on next suse import."""
+
+        print(f"\nEnable reprocess of old suse advisory on next import.")
+        Advisory = apps.get_model("vulnerabilities", "Advisory")
+        Advisory.objects.filter(
+            created_by="vulnerabilities.importers.suse_scores.SUSESeverityScoreImporter"
+        ).update(date_imported=None)
+
+    def reverse_reprocess_suse_advisory_on_next_import(apps, schema_editor):
+        """Populate `date_imported` on old suse advisory to prevent reprocessing on next suse import."""
+        Advisory = apps.get_model("vulnerabilities", "Advisory")
+        Advisory.objects.filter(
+            created_by="vulnerabilities.importers.suse_scores.SUSESeverityScoreImporter"
+        ).update(date_imported=datetime.now(timezone.utc))
+
     def populate_severity_model_with_url(apps, schema_editor):
         """Populate the new VulnerabilitySeverity `url` field using reference url"""
 
@@ -181,6 +201,10 @@ def reverse_populate_vulnerability_model_with_severities(apps, schema_editor):
             code=remove_inaccurate_suse_score,
             reverse_code=reverse_remove_inaccurate_suse_score,
         ),
+        migrations.RunPython(
+            code=reprocess_suse_advisory_on_next_import,
+            reverse_code=reverse_reprocess_suse_advisory_on_next_import,
+        ),
         migrations.RunPython(
             code=populate_severity_model_with_url,
             reverse_code=reverse_populate_severity_model_reference_using_url,