Admin task status actions

Author: kmmbvnr

viewflow/admin.py

@@ -46,7 +46,34 @@ class TaskAdmin(admin.ModelAdmin):
                     'started', 'finished']
     list_display_links = ['pk', 'created', 'process']
     list_filter = ['status']
-    readonly_fields = ['process', 'flow_task', 'started', 'finished', 'previous', 'token']
+    readonly_fields = ['process', 'status', 'flow_task', 'started', 'finished', 'previous', 'token']
+
+    @property
+    def change_form_template(self):
+        opts = self.model._meta
+
+        return [
+            "admin/%s/%s/change_form.html" % (opts.app_label, opts.model_name),
+            "admin/%s/change_form.html" % opts.app_label,
+            'admin/viewflow/task/change_form.html'
+        ]
+
+    def save_model(self, request, obj, form, change):
+        result = super(TaskAdmin, self).save_model(request, obj, form, change)
+
+        status_action = next((action[len('_change_status_'):]
+                              for action in request.POST.keys()
+                              if action.startswith('_change_status_')), None)
+        if status_action:
+            activation = obj.activate()
+            activation_cls = activation.__class__
+            transition = next((transition for transition in activation_cls.status.get_available_transtions(activation)
+                               if transition.name == status_action), None)
+            if transition:
+                transition(activation)
+                request.POST['_continue'] = True
+
+        return result
 
 
 admin.site.register(Process, ProcessAdmin)

viewflow/flow/task_view.py

@@ -83,7 +83,9 @@ def get_management_form_cls(self):
     @Activation.status.super()
     def prepare(self, data=None, user=None):
         super(ManagedViewActivation, self).prepare.original()
-        self.task.owner = user
+
+        if user:
+            self.task.owner = user
 
         management_form_cls = self.get_management_form_cls()
         self.management_form = management_form_cls(data=data, instance=self.task)

viewflow/templates/admin/viewflow/task/change_form.html

@@ -0,0 +1,6 @@
+{% extends "admin/change_form.html" %}
+{% load admin_modify viewflow %}
+
+{% block submit_buttons_top %}{% viewflow_task_submit_row %}{% endblock %}
+
+{% block submit_buttons_bottom %}{% viewflow_task_submit_row %}{% endblock %}

viewflow/templates/admin/viewflow/task/submit_line.html

@@ -0,0 +1,14 @@
+{% load i18n admin_urls %}
+<div class="submit-row">
+{% if show_save %}<input type="submit" value="{% trans 'Save' %}" class="default" name="_save" />{% endif %}
+{% if show_delete_link %}
+    {% url opts|admin_urlname:'delete' original.pk|admin_urlquote as delete_url %}
+    <p class="deletelink-box"><a href="{% add_preserved_filters delete_url %}" class="deletelink">{% trans "Delete" %}</a></p>
+{% endif %}
+{% if show_save_as_new %}<input type="submit" value="{% trans 'Save as new' %}" name="_saveasnew" />{%endif%}
+{% if show_save_and_add_another %}<input type="submit" value="{% trans 'Save and add another' %}" name="_addanother" />{% endif %}
+{% if show_save_and_continue %}<input type="submit" value="{% trans 'Save and continue editing' %}" name="_continue" />{% endif %}
+{% for transition_name, transition in transitions %}
+<input type="submit" value="{{ transition_name }}" class="default" name="_change_status_{{ transition }}"/>
+{% endfor %}
+</div>

viewflow/templatetags/viewflow.py

@@ -1,6 +1,7 @@
 from inspect import getargspec
 
 from django import template
+from django.contrib.admin.templatetags.admin_modify import submit_row
 from django.core.urlresolvers import reverse
 from django.template.base import TemplateSyntaxError, TagHelperNode, parse_bits
 from django.template.loader import select_template
@@ -124,3 +125,17 @@ def include_process_data(context, process):
         return template.render(context)
     finally:
         context.pop()
+
+
+@register.inclusion_tag('admin/viewflow/task/submit_line.html', takes_context=True)
+def viewflow_task_submit_row(context):
+    task = context.get('original', None)
+    activation = task.activate()
+    activation_cls = activation.__class__
+
+    transitions = [(transition.name.replace('_', ' ').capitalize(), transition.name)
+                   for transition in activation_cls.status.get_available_transtions(activation)]
+
+    row_context = submit_row(context)
+    row_context['transitions'] = transitions
+    return row_context