django-admin-background-task 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

bgtask/model_admin.py

@@ -1,7 +1,9 @@
 from datetime import timedelta
+from functools import wraps
 
 from django.contrib import admin
 from django.contrib.admin.utils import label_for_field
+from django.contrib.messages import INFO
 from django.db.models import Q
 from django.utils import timezone
 
@@ -12,8 +14,29 @@ class BGTaskModelAdmin(admin.ModelAdmin):
     # This is not overridden to avoid messing with the implicit logic for finding change list
     # templates that ModelAdmin uses. So you either need to specify this yourself on your
     # subclass or you need to extend from this in your custom template.
+    #
     # change_list_template = "bgtask/admin/change_list.html"
 
+    # ----------------------------------------------------------------------------------------------
+    # Class API
+    # ----------------------------------------------------------------------------------------------
+    @classmethod
+    def starts_task(cls, name, **task_kwargs):
+
+        def starts_task_decorator(func):
+
+            @wraps(func)
+            def starts_task_wrapper(self, request, *args, **kwargs):
+                bgtask = self.start_bgtask(name, **task_kwargs)
+                result = func(self, request, *args, bgtask=bgtask, **kwargs)
+                self.message_user(request, f"Dispatched task {name}", INFO)
+
+            func.bgtask_name = name
+
+            return starts_task_wrapper
+
+        return starts_task_decorator
+
     # ----------------------------------------------------------------------------------------------
     # API for subclasses
     # ----------------------------------------------------------------------------------------------
@@ -41,11 +64,26 @@ class BGTaskModelAdmin(admin.ModelAdmin):
     def _bgtask_namespace(self):
         return type(self).__module__ + "." + type(self).__name__
 
+    @staticmethod
+    def _extract_bgtask_name_from_admin_action(action):
+        # recurse through the potentially wrapped action until we find one that declares
+        # the bgtask_name
+        next_action = action
+        while True:
+            if hasattr(next_action, "bgtask_name"):
+                return next_action.bgtask_name
+
+            if not hasattr(next_action, "__wrapped__"):
+                return None
+
+            next_action = next_action.__wrapped__
+
     def _admin_bg_tasks(self, request):
         task_name_to_desc = {}
         for action, action_name, action_description in self.get_actions(request).values():
-            if hasattr(action, "bgtask_name"):
-                task_name_to_desc[action.bgtask_name] = action_description
+            bgtask_name = self._extract_bgtask_name_from_admin_action(action)
+            if bgtask_name is not None:
+                task_name_to_desc[bgtask_name] = action_description
 
         for name in getattr(self, "bgtask_names", []):
             task_name_to_desc[name] = name
@@ -58,10 +96,13 @@ class BGTaskModelAdmin(admin.ModelAdmin):
                 name__in=task_name_to_desc, namespace=self._bgtask_namespace
             )
             .filter(
-                Q(state=BackgroundTask.STATES.running)
+                (
+                    Q(state=BackgroundTask.STATES.running)
+                    & Q(started_at__gt=timezone.now() - timedelta(days=1))
+                )
                 | (
                     ~Q(state=BackgroundTask.STATES.not_started)
-                    & Q(completed_at__gt=timezone.now() - timedelta(minutes=30))
+                    & Q(completed_at__gt=timezone.now() - timedelta(hours=2))
                 )
             )
             .order_by("-started_at")

bgtask/static/bgtask/js/bgtask.js

@@ -14,6 +14,33 @@ const TIME_TO_REDUCED_REFRESH_PERIOD_S = 300;
 const REDUCED_REFRESH_PERIOD_S = 60;
 const PROGRESS_REFRESH_MS = 100;
 
+function millisecondsToTimeAgoString(ms) {
+  const seconds = Math.floor(ms / 1000);
+  const minutes = Math.floor(seconds / 60);
+  const hours = Math.floor(minutes / 60);
+  const days = Math.floor(hours / 24);
+
+  if (days > 0) {
+    if (days === 1) {
+      return `${days} day, ${hours % 24} hours ago`;
+    }
+    return `${days} days ago`;
+  }
+  if (hours > 0) {
+    if (hours === 1) {
+      return `${hours} hour, ${minutes % 60} minutes ago`;
+    }
+    return `${hours} hours ago`;
+  }
+  if (minutes > 0) {
+    if (minutes === 1) {
+      return `${minutes} minute, ${seconds % 60} seconds ago`;
+    }
+    return `${minutes} minutes ago`;
+  }
+  return `${seconds} seconds ago`;
+}
+
 // -------------------------------------------------------------------------------------------------
 // Generic live-looking progress bar manager.
 // -------------------------------------------------------------------------------------------------
@@ -140,7 +167,6 @@ class TaskProgressDiv {
   }
 
   updateFromTask(task) {
-    // console.log(`TaskProgressDiv.updateFromTask`, task);
     this.div.title = "";
 
     switch (task.state) {
@@ -152,7 +178,14 @@ class TaskProgressDiv {
       case "failed":
         this._hideProgress();
         this._showState();
-        this._addTitle("Task failed");
+        let title = "Task failed";
+        for (const error of task.errors) {
+          if (error.traceback) {
+            title = `${title}\n${error.traceback}\n\n${error.error_message}`;
+            break;
+          }
+        }
+        this._addTitle(title);
         break;
       case "success":
         this._hideProgress();
@@ -205,6 +238,26 @@ class TaskProgressDiv {
   }
 }
 
+// -------------------------------------------------------------------------------------------------
+// Manage an element which just shows one field from the task
+// -------------------------------------------------------------------------------------------------
+class TaskFieldElement {
+  constructor (element, fieldName, task) {
+    this.taskId = task.id;
+    this.element = element;
+    this.fieldName = fieldName;
+    this.updateFromTask(task);
+  }
+
+  attachToPoller(poller) {
+    poller.monitorTask(this.taskId, task => this.updateFromTask(task));
+  }
+
+  updateFromTask(task) {
+    this.element.innerHTML = task[this.fieldName];
+  }
+}
+
 // -------------------------------------------------------------------------------------------------
 // Manage a task detail div
 // -------------------------------------------------------------------------------------------------
@@ -299,6 +352,14 @@ class BGTaskPoller {
 
   static normalizeTask(task) {
     task.updated = new Date(task.updated);
+
+    if (task.completed_at !== null) {
+      task.completed_at = new Date(task.completed_at);
+      const completedTimeAgoMS = Date.now() - task.completed_at.getTime();
+      task.completed_at_time_ago = millisecondsToTimeAgoString(completedTimeAgoMS);
+    } else {
+      task.completed_at_time_ago = "–";
+    }
   }
 
   static sharedInstance(baseURL) {
@@ -357,7 +418,9 @@ class BGTaskPoller {
     const req = new XMLHttpRequest();
     const self = this;
     req.addEventListener("load", function () { self._receivePoll(this); });
-    const url = `${this.baseURL}?tasks=${Object.keys(this.taskCallbacks).join(",")}`;
+    const taskIds = Object.keys(this.taskCallbacks).join(",");
+    const url = `${this.baseURL}?tasks=${taskIds}`;
+    console.log(`Poll for tasks ${taskIds}`);
     req.open("GET", url);
     req.setRequestHeader('Accept', 'application/json');
     req.send();

bgtask/templates/bgtask/admin/change_list.html

@@ -8,7 +8,7 @@
 
 {% block object-tools %}
 {{ block.super }}
-<div>
+<div style="margin-bottom: 10px;">
   <h3>Action background tasks</h3>
   {% if not admin_bg_tasks|length %}
   <p class="help" style="font-style: italic;">No recent background tasks</p>
@@ -20,7 +20,7 @@
     <tbody>
       {% for bgt in admin_bg_tasks %}
       {% with bgtask=bgt.task_dict %}
-      <tr class="bgtask-row"><td><a href="{% url 'admin:bgtask_backgroundtask_change' bgtask.id %}">{{ bgt.admin_description }}</a></td><td>{{ bgt.started_at|timesince }} ago</td><td>{% if bgt.completed_at %}{{ bgt.completed_at|timesince }} ago{% else %}–{% endif %}</td><td>
+      <tr class="bgtask-row"><td><a href="{% url 'admin:bgtask_backgroundtask_change' bgtask.id %}">{{ bgt.admin_description }}</a></td><td>{{ bgt.started_at|timesince }} ago</td><td>{% include "bgtask/bg_completed_column.html" %}</td><td>
       {% include "bgtask/bg_changelist_status_column.html" %}</td></tr>
       {% endwith %}
       {% endfor %}

bgtask/templates/bgtask/bg_changelist_status_column.html

@@ -6,11 +6,7 @@
 {{ bgtask|json_script:initialTasksJsonId }}
 {% endwith %}
 <script src="{% static 'bgtask/js/bgtask-once.js' %}"></script>
-{% if bgtask.acted_on_object_id %}
-<a href="{% url 'bgtask:tasks' %}?object_id={{bgtask.acted_on_object_id}}">
-{% else %}
 <a href="{% url 'admin:bgtask_backgroundtask_change' bgtask.id %}">
-{% endif %}
     <div id="bgtask-column-{{ bgtask.id }}">
         {% include 'bgtask/progress.html' %}
     </div>

bgtask/templates/bgtask/bg_completed_column.html

@@ -0,0 +1,24 @@
+{% load static %}
+{% if not bgtask %}
+–
+{% else %}
+{% with initialTasksJsonId="initialTasksJson-completedcolumn-"|add:bgtask.id %}
+{{ bgtask|json_script:initialTasksJsonId }}
+{% endwith %}
+<script src="{% static 'bgtask/js/bgtask-once.js' %}"></script>
+<span></span>
+<script>
+(() => {
+    const completedSpan = document.currentScript.previousElementSibling;
+
+    const originalTask = JSON.parse(
+        document.getElementById("initialTasksJson-completedcolumn-{{ bgtask.id }}").textContent
+    );
+    BGTaskPoller.normalizeTask(originalTask);
+
+    const statusCol = new TaskFieldElement(completedSpan, "completed_at_time_ago", originalTask);
+    const poller = BGTaskPoller.sharedInstance("{% url 'bgtask:tasks' %}");
+    statusCol.attachToPoller(poller);
+})();
+</script>
+{% endif %}

{django_admin_background_task-0.2.0.dist-info → django_admin_background_task-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-admin-background-task
-Version: 0.2.0
+Version: 0.3.0
 Summary: A set of tools for django apps to persist and monitor the status of background tasks
 Home-page: https://github.com/daphtdazz/django-bgtask
 License: Apache-2.0

{django_admin_background_task-0.2.0.dist-info → django_admin_background_task-0.3.0.dist-info}/RECORD

@@ -7,13 +7,14 @@ bgtask/decorators.py,sha256=UsiCBhqaWm1Yn4n4hmFbauGQvNezoeDm7Iqpjbvxbjk,1119
 bgtask/migrations/0001_initial.py,sha256=YDTEy_hiXHru-1ZdIH-n3po82r5vyzbkyYIrF9_yiI8,2013
 bgtask/migrations/0002_backgroundtask_namespace_alter_backgroundtask_name.py,sha256=XIG_gIvdA73tz0NRbNkGU_ieKde3ZtQltPeLdslD5Xo,968
 bgtask/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bgtask/model_admin.py,sha256=tx90pz3MGMkN2DsZ8jEwEXY875QVXySGg_9rGcID50k,2925
+bgtask/model_admin.py,sha256=mwRoljw1CwMA-c3zbssmrRsV9l8xsgYJOBrkgVNilRs,4402
 bgtask/models.py,sha256=VulETkcBXJkyUgmExSx9fuOvpp1HnY2sLQLD5BUvbHk,9184
 bgtask/stack_contexts.py,sha256=2Hun6O5Mm949uAioC2E3i54Rg-_H8P7LsHdhJYc0w6E,231
 bgtask/static/bgtask/js/bgtask-once.js,sha256=DBerpxPLP21IH--HpBxVFk3kN52FtBiuyJsDCjg7mCA,746
-bgtask/static/bgtask/js/bgtask.js,sha256=VQ853jxRxs53E5E4XKLLeRByVcFuyqHEsniBIb8wq4Q,13151
-bgtask/templates/bgtask/admin/change_list.html,sha256=_Vjm1SLGj_Z2lX1H5y-jfLNWFiRfmvKLgh2YG0QfCpc,1062
-bgtask/templates/bgtask/bg_changelist_status_column.html,sha256=4ATqCKTx8yVINvSUfRKckjrs8JteX45R4fnAAHm9c9g,978
+bgtask/static/bgtask/js/bgtask.js,sha256=aYG02BY2F0aBlyRT_19turcC6RNLmyhJZUiuABUszoo,15026
+bgtask/templates/bgtask/admin/change_list.html,sha256=nR7LQ9yPv1LgD97g9oU4d8P6wxq9HrEJ4nneA14msNM,1053
+bgtask/templates/bgtask/bg_changelist_status_column.html,sha256=SA7EAL50oBrcRPGstXROXkFJ541vYA8RlHidM3srTPU,844
+bgtask/templates/bgtask/bg_completed_column.html,sha256=Q4l3EQaOXHYVOzv0jpx3ww3Wa2Bb33RyZp3RSWI1zog,788
 bgtask/templates/bgtask/bgtask_templates.html,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 bgtask/templates/bgtask/bgtask_view.html,sha256=AeubqERFF14PHWjInBD39wGImnRx4taS42iOMOPDUlM,1845
 bgtask/templates/bgtask/progress.html,sha256=VVYbrunGasOshRlz8Nm8hFPxqcVtrpaW1e2QxzAMcMM,166
@@ -22,7 +23,7 @@ bgtask/templatetags/bgtask.py,sha256=WUEWlQX5AarbXJBV6EGp-khT1zegVwGNigSKRF5mCTE
 bgtask/tests/test_bgtask.py,sha256=9AsBppPphaIdphuB-pHuQjgLhYNp8ySWtmcWF2uQl3Q,1090
 bgtask/urls.py,sha256=DXXjgj18LTSCq2-xZf23-pkgO4RmZglv67mgrp-fDyk,161
 bgtask/views.py,sha256=tPblidBq8mIv9cbyAfzJfNlM-I-Z11XVoobdA3DZBVM,1765
-django_admin_background_task-0.2.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-django_admin_background_task-0.2.0.dist-info/METADATA,sha256=zM0kMweJLTim6uyYHhQIsFKR-RpYW-wwcNlwcZAyk6M,1416
-django_admin_background_task-0.2.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-django_admin_background_task-0.2.0.dist-info/RECORD,,
+django_admin_background_task-0.3.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+django_admin_background_task-0.3.0.dist-info/METADATA,sha256=mDJm-1PabE_GFKUSxv9chBhiBhmtbBAaiyvlKxr0MQ4,1416
+django_admin_background_task-0.3.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+django_admin_background_task-0.3.0.dist-info/RECORD,,