delete scripts migration

validity/migrations/0010_squashed_initial.py

@@ -513,5 +513,4 @@ class Migration(migrations.Migration):
         ),
         migrations.RunPython(create_cf, delete_cf),
         migrations.RunPython(create_polling_datasource, delete_polling_datasource),
-        migrations.RunPython(setup_scripts, delete_scripts),
     ]

validity/migrations/0011_delete_scripts.py

@@ -0,0 +1,24 @@
+from django.db import migrations
+
+
+DATASOURCE_NAME = "validity_scripts"
+
+
+def delete_scripts(apps, schema_editor):
+    """
+    Delete DataSource and ScriptModule used for validity v1/v2
+    """
+    DataSource = apps.get_model("core", "DataSource")
+    ScriptModule = apps.get_model("extras", "ScriptModule")
+    db_alias = schema_editor.connection.alias
+    ScriptModule.objects.using(db_alias).filter(data_source__name=DATASOURCE_NAME).delete()
+    DataSource.objects.using(db_alias).filter(name=DATASOURCE_NAME).delete()
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("validity", "0011_squashed_initial"),
+    ]
+    operations = [
+        migrations.RunPython(delete_scripts, migrations.RunPython.noop),
+    ]

validity/views/report.py

@@ -1,5 +1,5 @@
 import functools
-from typing import Any, Dict, Iterable, Iterator
+from typing import Any, Iterable, Iterator
 
 from django.db.models.query import QuerySet
 from django.shortcuts import get_object_or_404
@@ -101,7 +101,7 @@ def get_table(self, **kwargs):
         table.configure(self.request)
         return table
 
-    def get_context_data(self, **kwargs: Any) -> Dict[str, Any]:
+    def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
         return super().get_context_data(**kwargs) | {
             "object": self.object,
             "tab": self.tab,