Added state list, create, edit, delete views

apps/workflows/views.py

@@ -18,11 +18,13 @@ from common.utils import encapsulate
 #from common.widgets import two_state_template
 #from acls.models import AccessEntry
 
-from .models import Workflow
-from .forms import WorkflowSetupForm
+from .models import Workflow, State, Transition
+from .forms import WorkflowSetupForm, StateSetupForm
 from .permissions import (PERMISSION_WORKFLOW_SETUP_VIEW,
     PERMISSION_WORKFLOW_SETUP_CREATE, PERMISSION_WORKFLOW_SETUP_EDIT,
-    PERMISSION_WORKFLOW_SETUP_DELETE)
+    PERMISSION_WORKFLOW_SETUP_DELETE, PERMISSION_STATE_SETUP_VIEW,
+    PERMISSION_STATE_SETUP_CREATE, PERMISSION_STATE_SETUP_EDIT,
+    PERMISSION_STATE_SETUP_DELETE)
 
 
 logger = logging.getLogger(__name__)
@@ -139,3 +141,114 @@ def setup_workflow_delete(request, workflow_pk=None, workflow_pk_list=None):
 
     return render_to_response('generic_confirm.html', context,
         context_instance=RequestContext(request))    
+
+# States
+def setup_state_list(request):
+    Permission.objects.check_permissions(request.user, [PERMISSION_STATE_SETUP_VIEW])
+
+    context = {
+        'object_list': State.objects.all(),
+        'title': _(u'states'),
+        'hide_link': True,
+        #'list_object_variable_name': 'source',
+        #'source_type': source_type,
+        #'extra_columns': [
+        #    {'name': _(u'Initial state'), 'attribute': encapsulate(lambda workflow: workflow.initial_state or _(u'None'))},
+        #],
+    }
+
+    return render_to_response('generic_list.html', context,
+        context_instance=RequestContext(request))
+
+
+def setup_state_create(request):
+    Permission.objects.check_permissions(request.user, [PERMISSION_STATE_SETUP_CREATE])
+    redirect_url = reverse('setup_state_list')
+
+    if request.method == 'POST':
+        form = StateSetupForm(request.POST)
+        if form.is_valid():
+            state = form.save()
+            messages.success(request, _(u'State created succesfully.'))
+            return HttpResponseRedirect(redirect_url)
+    else:
+        form = StateSetupForm()
+
+    return render_to_response('generic_form.html', {
+        'title': _(u'create state'),
+        'form': form,
+    },
+    context_instance=RequestContext(request))
+
+
+def setup_state_edit(request, state_pk):
+    Permission.objects.check_permissions(request.user, [PERMISSION_STATE_SETUP_EDIT])
+    state = get_object_or_404(State, pk=state_pk)
+
+    if request.method == 'POST':
+        form = StateSetupForm(instance=state, data=request.POST)
+        if form.is_valid():
+            form.save()
+            messages.success(request, _(u'State "%s" updated successfully.') % state)
+            return HttpResponseRedirect(reverse('setup_state_list'))
+    else:
+        form = StateSetupForm(instance=state)
+
+    return render_to_response('generic_form.html', {
+        'title': _(u'edit state: %s') % state,
+        'form': form,
+        'object': state,
+        'object_name': _(u'state'),
+    },
+    context_instance=RequestContext(request))
+    
+    
+def setup_state_delete(request, state_pk=None, state_pk_list=None):
+    post_action_redirect = None
+
+    if state_pk:
+        states = [get_object_or_404(State, pk=state_pk)]
+        post_action_redirect = reverse('setup_state_list')
+    elif state_pk_list:
+        states = [get_object_or_404(State, pk=state_pk) for state_pk in state_pk_list.split(',')]
+    else:
+        messages.error(request, _(u'Must provide at least one state.'))
+        return HttpResponseRedirect(request.META.get('HTTP_REFERER', '/'))
+
+    try:
+        Permission.objects.check_permissions(request.user, [PERMISSION_STATE_SETUP_DELETE])
+    except PermissionDenied:
+        states = AccessEntry.objects.filter_objects_by_access(PERMISSION_STATE_SETUP_DELETE, request.user, states)
+
+    previous = request.POST.get('previous', request.GET.get('previous', request.META.get('HTTP_REFERER', '/')))
+    next = request.POST.get('next', request.GET.get('next', post_action_redirect if post_action_redirect else request.META.get('HTTP_REFERER', '/')))
+
+    if request.method == 'POST':
+        for state in states:
+            try:
+                state.delete()
+                messages.success(request, _(u'States "%s" deleted successfully.') % state)
+            except Exception, e:
+                messages.error(request, _(u'Error deleting state "%(state)s": %(error)s') % {
+                    'state': state, 'error': e
+                })
+
+        return HttpResponseRedirect(next)
+
+    context = {
+        'object_name': _(u'state'),
+        'delete_view': True,
+        'previous': previous,
+        'next': next,
+        'form_icon': u'transmit_delete.png',
+    }
+    if len(states) == 1:
+        context['object'] = states[0]
+        context['title'] = _(u'Are you sure you wish to delete the state: %s?') % ', '.join([unicode(d) for d in states])
+        context['message'] = _('Will be removed from all documents.')
+    elif len(states) > 1:
+        context['title'] = _(u'Are you sure you wish to delete the states: %s?') % ', '.join([unicode(d) for d in states])
+        context['message'] = _('Will be removed from all documents.')
+
+    return render_to_response('generic_confirm.html', context,
+        context_instance=RequestContext(request))