changes

monitorizer/inventory/migrations/0002_alter_domainscan_status.py

@@ -0,0 +1,28 @@
+# Generated by Django 5.0.4 on 2024-04-12 20:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("inventory", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="domainscan",
+            name="status",
+            field=models.CharField(
+                choices=[
+                    ("pending", "Pending"),
+                    ("running", "Running"),
+                    ("success", "Success"),
+                    ("error", "Error"),
+                ],
+                db_index=True,
+                default="pending",
+                max_length=32,
+            ),
+        ),
+    ]

monitorizer/inventory/models.py

@@ -79,7 +79,10 @@ class ScanStatus(models.TextChoices):
 
     domain = models.ForeignKey(SeedDomain, on_delete=models.CASCADE, blank=True)
     status = models.CharField(
-        choices=ScanStatus.choices, max_length=32, default=ScanStatus.PENDING
+        choices=ScanStatus.choices,
+        max_length=32,
+        default=ScanStatus.PENDING,
+        db_index=True,
     )
     command_tpl = models.ForeignKey(CommandTemplate, on_delete=models.CASCADE)
     command_tpl_vars = models.JSONField(null=True, blank=True)

monitorizer/inventory/tasks.py

@@ -27,6 +27,15 @@ def execute_scan_descriptor(descriptor_pk):
     descriptor = models.ScanAutoSubmitter.objects.get(pk=descriptor_pk)
     if not descriptor.domain.enabled:
         return
+
+    if (
+        models.DomainScan.objects.filter(
+            status=models.DomainScan.ScanStatus.PENDING
+        ).count()
+        > 1500
+    ):
+        return
+
     for command in descriptor.commands.all():
         models.DomainScan.objects.create(
             descriptor=descriptor,