upload

2026-07-10 09:15:15 +02:00
parent b80f501d5a
commit 9d5ea53b5c
18 changed files with 2122 additions and 0 deletions
@@ -0,0 +1,90 @@
+from core.exceptions import JobFailed
+from django.utils import timezone
+from netbox.jobs import JobRunner, system_job
+
+from .choices import SyncStatusChoices
+from .models import VCenterEndpoint
+from .sync import VMwareImporter
+
+
+class SyncVCenterEndpointJob(JobRunner):
+    class Meta:
+        name = "VMware VM synchronization"
+
+    def run(self, *args, **kwargs):
+        endpoint = self.job.object
+        if endpoint is None:
+            endpoint_pk = kwargs.get("endpoint_pk")
+            endpoint = VCenterEndpoint.objects.get(pk=endpoint_pk)
+
+        if not endpoint.enabled:
+            endpoint.mark_failure("Endpoint is disabled.")
+            raise JobFailed("Endpoint is disabled.")
+
+        self.logger.info("Starting VMware sync for %s (%s)", endpoint.name, endpoint.host)
+        endpoint.mark_running()
+
+        try:
+            result = VMwareImporter(endpoint, self.logger).sync()
+        except Exception as exc:
+            endpoint.mark_failure(exc)
+            raise
+
+        if result.errors:
+            endpoint.last_sync_at = timezone.now()
+            endpoint.last_status = SyncStatusChoices.STATUS_FAILED
+            endpoint.last_vm_count = result.seen
+            endpoint.last_created_count = result.created
+            endpoint.last_updated_count = result.updated
+            endpoint.last_error_count = result.errors
+            endpoint.last_status = SyncStatusChoices.STATUS_FAILED
+            endpoint.last_message = f"Finished with {result.errors} VM error(s). Check the job log."
+            endpoint.set_next_sync(endpoint.last_sync_at)
+            endpoint.save(
+                update_fields=(
+                    "last_sync_at",
+                    "last_status",
+                    "last_vm_count",
+                    "last_created_count",
+                    "last_updated_count",
+                    "last_error_count",
+                    "last_message",
+                    "next_sync_at",
+                    "last_updated",
+                )
+            )
+            raise JobFailed(endpoint.last_message)
+
+        endpoint.mark_success(result)
+        self.logger.info(
+            "VMware sync finished for %s: %s VM(s), %s created, %s updated, %s skipped.",
+            endpoint.name,
+            result.synced,
+            result.created,
+            result.updated,
+            result.skipped,
+        )
+
+
+@system_job(interval=5)
+class ScheduleDueVCenterSyncsJob(JobRunner):
+    class Meta:
+        name = "Schedule due VMware VM synchronizations"
+
+    def run(self, *args, **kwargs):
+        now = timezone.now()
+        due_endpoints = VCenterEndpoint.objects.filter(
+            enabled=True,
+            sync_interval_minutes__isnull=False,
+        ).filter(next_sync_at__lte=now)
+
+        queued = 0
+        for endpoint in due_endpoints:
+            SyncVCenterEndpointJob.enqueue_once(instance=endpoint)
+            endpoint.last_status = SyncStatusChoices.STATUS_QUEUED
+            endpoint.last_message = "Automatic sync job queued."
+            endpoint.set_next_sync(now)
+            endpoint.save(update_fields=("last_status", "last_message", "next_sync_at", "last_updated"))
+            queued += 1
+
+        self.logger.info("Queued %s VMware sync job(s).", queued)