initial

tasklogger/models.py

@@ -0,0 +1,88 @@
+import logging
+from uuid import uuid4
+
+from django.db import models
+from django_middleware_global_request.middleware import get_request
+
+from lib.db import DateAwareMixin
+
+
+class TaskFactory:
+    def __new__(self, task_uuid=None, request=None, *args, **kwargs):
+        request = request or get_request()
+        if not request:
+            logging.error('No request found while creating task')
+        else:
+            session_uuid = request.session.get('current_task_uuid')
+            form_uuid = request.POST.get('task_uuid', request.GET.get('task_uuid'))
+            if form_uuid:
+                task_uuid = task_uuid or form_uuid
+                request.session['current_task_uuid'] = task_uuid
+            elif session_uuid:
+                task_uuid = task_uuid or session_uuid
+        task_uuid = task_uuid or kwargs.pop('uuid', None)
+        if not task_uuid:
+            task = Task.objects.create()
+        else:
+            task, _created = Task.objects.get_or_create(uuid=task_uuid)
+        if request:
+            request.session['current_task_uuid'] = str(task.uuid)
+        return task
+
+    @staticmethod
+    def reset_current_task(request=None):
+        request = request or get_request()
+        if request:
+            request.session.pop('current_task_uuid', None)
+        return True
+
+
+class Task(DateAwareMixin):
+    uuid = models.UUIDField(default=uuid4, editable=False, unique=True)
+    proxmox_upid = models.CharField(max_length=150, null=True, blank=True)
+    status = models.CharField(max_length=150, null=True, blank=True, default='running')
+
+    current_task = None
+
+    def __del__(self):
+        self.unset_as_current()
+
+    def unset_as_current(self, request=None):
+        request = request or get_request()
+        if request:
+            uuid = request.session.get('current_task_uuid')
+            if uuid == self.uuid:
+                request.session.pop('current_task_uuid', None)
+                return True
+            else:
+                logging.warning(f'Tried to unset {self.uuid} as current, but it is not the current task ({uuid}')
+        return False
+
+    def __str__(self):
+        return str(self.uuid)
+
+    def add_entry(self, message):
+        TaskEntry.objects.create(task=self, message=message)
+        return self
+
+    def task_is_stopped(self, result):
+        return result.get('status', 'running').lower() in ('ok', 'error', 'stopped')
+
+    def wrap_proxmox_function(self, proxmox_function, *args, **kwargs):
+        from lib.proxmox import Proxmox
+        self.proxmox_upid = proxmox_function(*args, **kwargs)
+        self.save()
+        if not self.proxmox_upid:
+            logging.warning(f'Could not get upid for {proxmox_function} - {args} {kwargs}')
+            return self
+        while True:
+            with Proxmox() as pm:
+                for message in pm.get_task_status(taskhash=self.proxmox_upid):
+                    self.add_entry(message)
+                    if self.task_is_stopped(message):
+                        return self
+
+
+class TaskEntry(DateAwareMixin):
+    task = models.ForeignKey(Task, on_delete=models.CASCADE, related_name='entries')
+    message = models.JSONField(null=True, blank=True)