489 lines
16 KiB
Python
489 lines
16 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import json
|
|
import logging
|
|
|
|
from graphviz import Digraph
|
|
|
|
from django.conf import settings
|
|
from django.core.exceptions import PermissionDenied, ValidationError
|
|
from django.db import IntegrityError, models
|
|
from django.db.models import F, Max, Q
|
|
from django.urls import reverse
|
|
from django.utils.encoding import force_text, python_2_unicode_compatible
|
|
from django.utils.module_loading import import_string
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from acls.models import AccessControlList
|
|
from common.validators import validate_internal_name
|
|
from documents.models import Document, DocumentType
|
|
from events.models import StoredEventType
|
|
from permissions import Permission
|
|
|
|
from .error_logs import error_log_state_actions
|
|
from .literals import (
|
|
WORKFLOW_ACTION_WHEN_CHOICES, WORKFLOW_ACTION_ON_ENTRY,
|
|
WORKFLOW_ACTION_ON_EXIT
|
|
)
|
|
from .managers import WorkflowManager
|
|
from .permissions import permission_workflow_transition
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Workflow(models.Model):
|
|
"""
|
|
Fields:
|
|
* label - Identifier. A name/label to call the workflow
|
|
"""
|
|
internal_name = models.CharField(
|
|
db_index=True, help_text=_(
|
|
'This value will be used by other apps to reference this '
|
|
'workflow. Can only contain letters, numbers, and underscores.'
|
|
), max_length=255, unique=True, validators=[validate_internal_name],
|
|
verbose_name=_('Internal name')
|
|
)
|
|
label = models.CharField(
|
|
max_length=255, unique=True, verbose_name=_('Label')
|
|
)
|
|
document_types = models.ManyToManyField(
|
|
DocumentType, related_name='workflows',
|
|
verbose_name=_('Document types')
|
|
)
|
|
|
|
objects = WorkflowManager()
|
|
|
|
def __str__(self):
|
|
return self.label
|
|
|
|
def get_document_types_not_in_workflow(self):
|
|
return DocumentType.objects.exclude(pk__in=self.document_types.all())
|
|
|
|
def get_initial_state(self):
|
|
try:
|
|
return self.states.get(initial=True)
|
|
except self.states.model.DoesNotExist:
|
|
return None
|
|
|
|
def launch_for(self, document):
|
|
try:
|
|
logger.info(
|
|
'Launching workflow %s for document %s', self, document
|
|
)
|
|
self.instances.create(document=document)
|
|
except IntegrityError:
|
|
logger.info(
|
|
'Workflow %s already launched for document %s', self, document
|
|
)
|
|
else:
|
|
logger.info(
|
|
'Workflow %s launched for document %s', self, document
|
|
)
|
|
|
|
def render(self):
|
|
diagram = Digraph(
|
|
name='finite_state_machine', graph_attr={
|
|
'rankdir': 'LR', 'size': '8,5'
|
|
}, format='png'
|
|
)
|
|
|
|
state_cache = {}
|
|
transition_cache = []
|
|
|
|
for state in self.states.all():
|
|
state_cache['s{}'.format(state.pk)] = {
|
|
'name': 's{}'.format(state.pk),
|
|
'label': state.label,
|
|
'initial': state.initial,
|
|
'connections': {'origin': 0, 'destination': 0}
|
|
}
|
|
|
|
for transition in self.transitions.all():
|
|
transition_cache.append(
|
|
{
|
|
'tail_name': 's{}'.format(transition.origin_state.pk),
|
|
'head_name': 's{}'.format(transition.destination_state.pk),
|
|
'label': transition.label
|
|
}
|
|
)
|
|
state_cache['s{}'.format(transition.origin_state.pk)]['connections']['origin'] = state_cache['s{}'.format(transition.origin_state.pk)]['connections']['origin'] + 1
|
|
state_cache['s{}'.format(transition.destination_state.pk)]['connections']['destination'] += 1
|
|
|
|
for key, value in state_cache.items():
|
|
kwargs = {
|
|
'name': value['name'],
|
|
'label': value['label'],
|
|
'shape': 'doublecircle' if value['connections']['origin'] == 0 or value['connections']['destination'] == 0 or value['initial'] else 'circle',
|
|
}
|
|
diagram.node(**kwargs)
|
|
|
|
for transition in transition_cache:
|
|
diagram.edge(**transition)
|
|
|
|
return diagram.pipe()
|
|
|
|
class Meta:
|
|
ordering = ('label',)
|
|
verbose_name = _('Workflow')
|
|
verbose_name_plural = _('Workflows')
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowState(models.Model):
|
|
"""
|
|
Fields:
|
|
* completion - Completion Amount - A user defined numerical value to help
|
|
determine if the workflow of the document is nearing completion (100%).
|
|
The Completion Amount will be determined by the completion value of the
|
|
Actual State. Example: If the workflow has 3 states: registered, approved,
|
|
archived; the admin could give the follow completion values to the
|
|
states: 33%, 66%, 100%. If the Actual State of the document if approved,
|
|
the Completion Amount will show 66%.
|
|
"""
|
|
workflow = models.ForeignKey(
|
|
Workflow, on_delete=models.CASCADE, related_name='states',
|
|
verbose_name=_('Workflow')
|
|
)
|
|
label = models.CharField(max_length=255, verbose_name=_('Label'))
|
|
initial = models.BooleanField(
|
|
default=False,
|
|
help_text=_(
|
|
'Select if this will be the state with which you want the '
|
|
'workflow to start in. Only one state can be the initial state.'
|
|
), verbose_name=_('Initial')
|
|
)
|
|
completion = models.IntegerField(
|
|
blank=True, default=0, help_text=_(
|
|
'Enter the percent of completion that this state represents in '
|
|
'relation to the workflow. Use numbers without the percent sign.'
|
|
), verbose_name=_('Completion')
|
|
)
|
|
|
|
class Meta:
|
|
ordering = ('label',)
|
|
unique_together = ('workflow', 'label')
|
|
verbose_name = _('Workflow state')
|
|
verbose_name_plural = _('Workflow states')
|
|
|
|
def __str__(self):
|
|
return self.label
|
|
|
|
def save(self, *args, **kwargs):
|
|
if self.initial:
|
|
self.workflow.states.all().update(initial=False)
|
|
return super(WorkflowState, self).save(*args, **kwargs)
|
|
|
|
@property
|
|
def entry_actions(self):
|
|
return self.actions.filter(when=WORKFLOW_ACTION_ON_ENTRY)
|
|
|
|
@property
|
|
def exit_actions(self):
|
|
return self.actions.filter(when=WORKFLOW_ACTION_ON_EXIT)
|
|
|
|
def get_documents(self):
|
|
latest_entries = WorkflowInstanceLogEntry.objects.annotate(
|
|
max_datetime=Max(
|
|
'workflow_instance__log_entries__datetime'
|
|
)
|
|
).filter(
|
|
datetime=F('max_datetime')
|
|
)
|
|
|
|
state_latest_entries = latest_entries.filter(
|
|
transition__destination_state=self
|
|
)
|
|
|
|
return Document.objects.filter(
|
|
Q(
|
|
workflows__pk__in=state_latest_entries.values_list(
|
|
'workflow_instance', flat=True
|
|
)
|
|
) | Q(
|
|
workflows__log_entries__isnull=True,
|
|
workflows__workflow__states=self,
|
|
workflows__workflow__states__initial=True
|
|
)
|
|
).distinct()
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowStateAction(models.Model):
|
|
state = models.ForeignKey(
|
|
WorkflowState, on_delete=models.CASCADE,
|
|
related_name='actions', verbose_name=_('Workflow state')
|
|
)
|
|
label = models.CharField(max_length=255, verbose_name=_('Label'))
|
|
enabled = models.BooleanField(default=True, verbose_name=_('Enabled'))
|
|
when = models.PositiveIntegerField(
|
|
choices=WORKFLOW_ACTION_WHEN_CHOICES,
|
|
default=WORKFLOW_ACTION_ON_ENTRY, help_text=_(
|
|
'At which moment of the state this action will execute'
|
|
), verbose_name=_('When')
|
|
)
|
|
action_path = models.CharField(
|
|
max_length=128, help_text=_(
|
|
'The dotted Python path to the workflow action class to execute.'
|
|
), verbose_name=_('Entry action path')
|
|
)
|
|
action_data = models.TextField(
|
|
blank=True, verbose_name=_('Entry action data')
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.label
|
|
|
|
class Meta:
|
|
ordering = ('label',)
|
|
unique_together = ('state', 'label')
|
|
verbose_name = _('Workflow state action')
|
|
verbose_name_plural = _('Workflow state actions')
|
|
|
|
def dumps(self, data):
|
|
self.action_data = json.dumps(data)
|
|
self.save()
|
|
|
|
def execute(self, context):
|
|
try:
|
|
self.get_class_instance().execute(context=context)
|
|
except Exception as exception:
|
|
error_log_state_actions.create(
|
|
obj=self, result='{}; {}'.format(
|
|
exception.__class__.__name__, exception
|
|
)
|
|
)
|
|
|
|
if settings.DEBUG:
|
|
raise
|
|
|
|
def get_class(self):
|
|
return import_string(self.action_path)
|
|
|
|
def get_class_label(self):
|
|
return self.get_class().label
|
|
|
|
def get_class_instance(self):
|
|
return self.get_class()(form_data=self.loads())
|
|
|
|
def loads(self):
|
|
return json.loads(self.action_data)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowTransition(models.Model):
|
|
workflow = models.ForeignKey(
|
|
Workflow, on_delete=models.CASCADE, related_name='transitions',
|
|
verbose_name=_('Workflow')
|
|
)
|
|
label = models.CharField(max_length=255, verbose_name=_('Label'))
|
|
origin_state = models.ForeignKey(
|
|
WorkflowState, on_delete=models.CASCADE,
|
|
related_name='origin_transitions', verbose_name=_('Origin state')
|
|
)
|
|
destination_state = models.ForeignKey(
|
|
WorkflowState, on_delete=models.CASCADE,
|
|
related_name='destination_transitions',
|
|
verbose_name=_('Destination state')
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.label
|
|
|
|
class Meta:
|
|
ordering = ('label',)
|
|
unique_together = (
|
|
'workflow', 'label', 'origin_state', 'destination_state'
|
|
)
|
|
verbose_name = _('Workflow transition')
|
|
verbose_name_plural = _('Workflow transitions')
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowTransitionTriggerEvent(models.Model):
|
|
transition = models.ForeignKey(
|
|
WorkflowTransition, on_delete=models.CASCADE,
|
|
related_name='trigger_events', verbose_name=_('Transition')
|
|
)
|
|
event_type = models.ForeignKey(
|
|
StoredEventType, on_delete=models.CASCADE,
|
|
verbose_name=_('Event type')
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _('Workflow transition trigger event')
|
|
verbose_name_plural = _('Workflow transitions trigger events')
|
|
|
|
def __str__(self):
|
|
return force_text(self.transition)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowInstance(models.Model):
|
|
workflow = models.ForeignKey(
|
|
Workflow, on_delete=models.CASCADE, related_name='instances',
|
|
verbose_name=_('Workflow')
|
|
)
|
|
document = models.ForeignKey(
|
|
Document, on_delete=models.CASCADE, related_name='workflows',
|
|
verbose_name=_('Document')
|
|
)
|
|
|
|
def __str__(self):
|
|
return force_text(self.workflow)
|
|
|
|
def get_absolute_url(self):
|
|
return reverse(
|
|
'document_states:workflow_instance_detail', args=(str(self.pk),)
|
|
)
|
|
|
|
def do_transition(self, transition, user=None, comment=None):
|
|
try:
|
|
if transition in self.get_current_state().origin_transitions.all():
|
|
self.log_entries.create(
|
|
comment=comment or '', transition=transition, user=user
|
|
)
|
|
except AttributeError:
|
|
# No initial state has been set for this workflow
|
|
pass
|
|
|
|
def get_current_state(self):
|
|
"""
|
|
Actual State - The current state of the workflow. If there are
|
|
multiple states available, for example: registered, approved,
|
|
archived; this field will tell at the current state where the
|
|
document is right now.
|
|
"""
|
|
try:
|
|
return self.get_last_transition().destination_state
|
|
except AttributeError:
|
|
return self.workflow.get_initial_state()
|
|
|
|
def get_last_log_entry(self):
|
|
try:
|
|
return self.log_entries.order_by('datetime').last()
|
|
except AttributeError:
|
|
return None
|
|
|
|
def get_last_transition(self):
|
|
"""
|
|
Last Transition - The last transition used by the last user to put
|
|
the document in the actual state.
|
|
"""
|
|
try:
|
|
return self.get_last_log_entry().transition
|
|
except AttributeError:
|
|
return None
|
|
|
|
def get_transition_choices(self, _user=None):
|
|
current_state = self.get_current_state()
|
|
|
|
if current_state:
|
|
queryset = current_state.origin_transitions.all()
|
|
|
|
if _user:
|
|
try:
|
|
Permission.check_permissions(
|
|
requester=_user, permissions=(
|
|
permission_workflow_transition,
|
|
)
|
|
)
|
|
except PermissionDenied:
|
|
try:
|
|
"""
|
|
Check for ACL access to the workflow, if true, allow
|
|
all transition options.
|
|
"""
|
|
|
|
AccessControlList.objects.check_access(
|
|
permissions=permission_workflow_transition,
|
|
user=_user, obj=self.workflow
|
|
)
|
|
except PermissionDenied:
|
|
"""
|
|
If not ACL access to the workflow, filter transition
|
|
options by each transition ACL access
|
|
"""
|
|
|
|
queryset = AccessControlList.objects.filter_by_access(
|
|
permission=permission_workflow_transition,
|
|
user=_user, queryset=queryset
|
|
)
|
|
return queryset
|
|
else:
|
|
"""
|
|
This happens when a workflow has no initial state and a document
|
|
whose document type has this workflow is created. We return an
|
|
empty transition queryset.
|
|
"""
|
|
|
|
return WorkflowTransition.objects.none()
|
|
|
|
class Meta:
|
|
unique_together = ('document', 'workflow')
|
|
verbose_name = _('Workflow instance')
|
|
verbose_name_plural = _('Workflow instances')
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class WorkflowInstanceLogEntry(models.Model):
|
|
"""
|
|
Fields:
|
|
* user - The user who last transitioned the document from a state to the
|
|
Actual State.
|
|
* datetime - Date Time - The date and time when the last user transitioned
|
|
the document state to the Actual state.
|
|
"""
|
|
workflow_instance = models.ForeignKey(
|
|
WorkflowInstance, on_delete=models.CASCADE,
|
|
related_name='log_entries', verbose_name=_('Workflow instance')
|
|
)
|
|
datetime = models.DateTimeField(
|
|
auto_now_add=True, db_index=True, verbose_name=_('Datetime')
|
|
)
|
|
transition = models.ForeignKey(
|
|
WorkflowTransition, on_delete=models.CASCADE,
|
|
verbose_name=_('Transition')
|
|
)
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL, blank=True, null=True,
|
|
on_delete=models.CASCADE, verbose_name=_('User')
|
|
)
|
|
comment = models.TextField(blank=True, verbose_name=_('Comment'))
|
|
|
|
def __str__(self):
|
|
return force_text(self.transition)
|
|
|
|
class Meta:
|
|
verbose_name = _('Workflow instance log entry')
|
|
verbose_name_plural = _('Workflow instance log entries')
|
|
|
|
def clean(self):
|
|
if self.transition not in self.workflow_instance.get_transition_choices(_user=self.user):
|
|
raise ValidationError(_('Not a valid transition choice.'))
|
|
|
|
def save(self, *args, **kwargs):
|
|
result = super(WorkflowInstanceLogEntry, self).save(*args, **kwargs)
|
|
|
|
for action in self.transition.origin_state.exit_actions.filter(enabled=True):
|
|
action.execute(context={'action': action, 'entry_log': self})
|
|
|
|
for action in self.transition.destination_state.entry_actions.filter(enabled=True):
|
|
action.execute(context={'action': action, 'entry_log': self})
|
|
|
|
return result
|
|
|
|
|
|
class WorkflowRuntimeProxy(Workflow):
|
|
class Meta:
|
|
proxy = True
|
|
verbose_name = _('Workflow runtime proxy')
|
|
verbose_name_plural = _('Workflow runtime proxies')
|
|
|
|
|
|
class WorkflowStateRuntimeProxy(WorkflowState):
|
|
class Meta:
|
|
proxy = True
|
|
verbose_name = _('Workflow state runtime proxy')
|
|
verbose_name_plural = _('Workflow state runtime proxies')
|