from __future__ import absolute_import, unicode_literals import logging 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.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 EventType from permissions import Permission 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 ) 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) 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 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( EventType, 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.')) 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')