From da5f23cde6610a9db15b089e7be3f7705e78a638 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20P=C3=A9ters?= <fpeters@entrouvert.com>
Date: Mon, 7 Mar 2016 21:28:20 +0100
Subject: [PATCH] wip: criticality levels

---
 wcs/admin/workflows.py              | 116 +++++++++++++++++++++++++++++++++++-
 wcs/backoffice/management.py        |  56 ++++++++++++++++-
 wcs/formdata.py                     |   2 +
 wcs/forms/backoffice.py             |  24 +++++++-
 wcs/forms/common.py                 |  14 -----
 wcs/qommon/static/css/dc2/admin.css |  31 ++++++++++
 wcs/qommon/static/js/biglist.js     |   1 +
 wcs/sql.py                          |  16 +++--
 wcs/wf/criticality.py               |  79 ++++++++++++++++++++++++
 wcs/workflows.py                    |  25 ++++++++
 10 files changed, 339 insertions(+), 25 deletions(-)
 create mode 100644 wcs/wf/criticality.py

diff --git a/wcs/admin/workflows.py b/wcs/admin/workflows.py
index 743b5c8..7c68467 100644
--- a/wcs/admin/workflows.py
+++ b/wcs/admin/workflows.py
@@ -949,6 +949,84 @@ class FunctionsDirectory(Directory):
         return redirect('..')
 
 
+class CriticalityLevelsDirectory(Directory):
+    _q_exports = ['', 'new']
+
+    def __init__(self, workflow):
+        self.workflow = workflow
+
+    def _q_traverse(self, path):
+        get_response().breadcrumb.append(
+                ('criticality-levels/', _('Criticality Levels')))
+        return Directory._q_traverse(self, path)
+
+    def new(self):
+        form = Form(enctype='multipart/form-data')
+        form.add(StringWidget, 'name', title=_('Name'), required=True, size=50)
+        form.add(ColourWidget, 'colour', title=_('Colour'), required=False)
+        form.add_submit('submit', _('Add'))
+        form.add_submit('cancel', _('Cancel'))
+        if form.get_widget('cancel').parse():
+            return redirect('..')
+
+        if form.is_submitted() and not form.has_errors():
+            if not self.workflow.criticality_levels:
+                self.workflow.criticality_levels = []
+            level = WorkflowCriticalityLevel()
+            level.name = form.get_widget('name').parse()
+            level.colour = form.get_widget('colour').parse()
+            self.workflow.criticality_levels.append(level)
+            self.workflow.store()
+            return redirect('..')
+
+        get_response().breadcrumb.append(('new', _('New Criticality Level')))
+        html_top('workflows', title=_('New Criticality level'))
+        r = TemplateIO(html=True)
+        r += htmltext('<h2>%s</h2>') % _('New Criticality Level')
+        r += form.render()
+        return r.getvalue()
+
+    def _q_lookup(self, component):
+        for level in (self.workflow.criticality_levels or []):
+            if level.id == component:
+                break
+        else:
+            raise errors.TraversalError()
+
+        form = Form(enctype='multipart/form-data')
+        form.add(StringWidget, 'name', title=_('Name'), required=True, size=50,
+                value=level.name)
+        form.add(ColourWidget, 'colour', title=_('Colour'), required=False,
+                value=level.colour)
+        form.add_submit('submit', _('Submit'))
+        form.add_submit('cancel', _('Cancel'))
+        form.add_submit('delete', _('Delete'))
+
+        if form.get_widget('cancel').parse():
+            return redirect('..')
+
+        if form.get_submit() == 'delete':
+            self.workflow.criticality_levels.remove(level)
+            self.workflow.store()
+            return redirect('..')
+
+        if form.is_submitted() and not form.has_errors():
+            level.name = form.get_widget('name').parse()
+            level.colour = form.get_widget('colour').parse()
+            self.workflow.store()
+            return redirect('..')
+
+        get_response().breadcrumb.append(('new', _('Edit Criticality Level')))
+        html_top('workflows', title=_('Edit Criticality Level'))
+        r = TemplateIO(html=True)
+        r += htmltext('<h2>%s</h2>') % _('Edit Criticality Level')
+        r += form.render()
+        return r.getvalue()
+
+    def _q_index(self):
+        return redirect('..')
+
+
 class GlobalActionPage(WorkflowStatusPage):
     _q_exports = ['', 'new', 'delete', 'newitem', ('items', 'items_dir'),
                   'edit', ('triggers', 'triggers_dir'),
@@ -1139,8 +1217,10 @@ class GlobalActionsDirectory(Directory):
 class WorkflowPage(Directory):
     _q_exports = ['', 'edit', 'delete', 'newstatus', ('status', 'status_dir'), 'update_order',
             'duplicate', 'export', 'svg', ('variables', 'variables_dir'),
-            'update_actions_order',
-            ('functions', 'functions_dir'), ('global-actions', 'global_actions_dir')]
+            'update_actions_order', 'update_criticality_levels_order',
+            ('functions', 'functions_dir'), ('global-actions', 'global_actions_dir'),
+            ('criticality-levels', 'criticality_levels_dir'),
+            ]
 
     def __init__(self, component, html_top):
         try:
@@ -1153,6 +1233,7 @@ class WorkflowPage(Directory):
         self.variables_dir = VariablesDirectory(self.workflow)
         self.functions_dir = FunctionsDirectory(self.workflow)
         self.global_actions_dir = GlobalActionsDirectory(self.workflow, html_top)
+        self.criticality_levels_dir = CriticalityLevelsDirectory(self.workflow)
         get_response().breadcrumb.append((component + '/', self.workflow.name))
 
     def _q_index(self):
@@ -1257,7 +1338,7 @@ class WorkflowPage(Directory):
                 r += htmltext('<ul id="status-list" class="biglist sortable" '
                               'data-order-function="update_actions_order">')
             else:
-                r += htmltext('<ul id="status-list" class="biglist">')
+                r += htmltext('<ul class="biglist">')
 
             for action in (self.workflow.global_actions or []):
                 r += htmltext('<li class="biglistitem" id="itemId_%s">' % action.id)
@@ -1270,6 +1351,27 @@ class WorkflowPage(Directory):
             r += htmltext('</ul>')
             r += htmltext('</div>')
 
+        if not str(self.workflow.id).startswith('_') and get_publisher().has_site_option('workflow-criticality-levels'):
+            r += htmltext('<div class="bo-block">')
+            r += htmltext('<h3>%s') % _('Criticality Levels')
+            if not str(self.workflow.id).startswith('_'):
+                r += htmltext(' <span class="change">'
+                              '(<a rel="popup" href="criticality-levels/new">'
+                              '%s</a>)</span>') % _('add criticality level')
+            r += htmltext('</h3>')
+            r += htmltext('<ul class="biglist sortable" '
+                              'data-order-function="update_criticality_levels_order">')
+            for level in (self.workflow.criticality_levels or []):
+                r += htmltext('<li class="biglistitem" id="itemId_%s">' % level.id)
+                if not str(self.workflow.id).startswith('_'):
+                    r += htmltext('<a rel="popup" href="criticality-levels/%s">%s</a>') % (
+                            level.id, level.name)
+                else:
+                    r += htmltext('<a>%s</a>') % level.name
+                r += htmltext('</li>')
+            r += htmltext('</ul>')
+            r += htmltext('</div>')
+
         r += htmltext('</div>') # .splitcontent-right
 
         r += htmltext('<br style="clear:both;"/>')
@@ -1355,6 +1457,14 @@ class WorkflowPage(Directory):
         self.workflow.store()
         return 'ok'
 
+    def update_criticality_levels_order(self):
+        request = get_request()
+        new_order = request.form['order'].strip(';').split(';')
+        self.workflow.criticality_levels = [
+                [x for x in self.workflow.criticality_levels if x.id == y][0] for y in new_order]
+        self.workflow.store()
+        return 'ok'
+
     def newstatus(self):
         form = Form(enctype='multipart/form-data', action = 'newstatus')
         form.add(StringWidget, 'name', title = _('Name'), size = 50)
diff --git a/wcs/backoffice/management.py b/wcs/backoffice/management.py
index c700247..8ba5c7e 100644
--- a/wcs/backoffice/management.py
+++ b/wcs/backoffice/management.py
@@ -688,11 +688,14 @@ class ManagementDirectory(Directory):
                 order_by=order_by, limit=limit, offset=offset)
         include_submission_channel = bool(
                 get_publisher().get_site_option('welco_url', 'variables'))
+        include_criticality_level = get_publisher().has_site_option('workflow-criticality-levels')
 
         r = TemplateIO(html=True)
         r += htmltext('<table id="listing" class="main">')
         r += htmltext('<thead>')
         r += htmltext('<th></th>') # lock
+        if include_criticality_level:
+            r += htmltext('<th data-field-sort-key="criticality_level"><span></span></th>')
         if include_submission_channel:
             r += htmltext('<th data-field-sort-key="submission_channel"><span>%s</span></th>') % _('Channel')
         r += htmltext('<th data-field-sort-key="formdef_name"><span>%s</span></th>') % _('Form')
@@ -713,10 +716,22 @@ class ManagementDirectory(Directory):
             object_key = 'formdata-%s-%s' % (formdata.formdef.url_name, formdata.id)
             if object_key in visited_objects:
                 classes.append('advisory-lock')
-            r += htmltext('<tr class="%s" data-link="%s">' % (
+            style = ''
+            if include_criticality_level:
+                try:
+                    level = formdata.formdef.workflow.criticality_levels[formdata.criticality_level]
+                except (IndexError, TypeError):
+                    pass
+                else:
+                    classes.append('criticality-level')
+                    style = 'style="border-color: #%s;"' % level.colour
+            r += htmltext('<tr class="%s" data-link="%s" %s>' % (
                     ' '.join(classes),
-                    formdata.get_url(backoffice=True)))
+                    formdata.get_url(backoffice=True),
+                    style))
             r += htmltext('<td></td>') # lock
+            if include_criticality_level:
+                r += htmltext('<td></td>')
             if include_submission_channel:
                 r += htmltext('<td>%s</td>') % formdata.get_submission_channel_label()
             r += htmltext('<td>%s</td>') % formdata.formdef.name
@@ -1642,6 +1657,43 @@ class FormBackOfficeStatusPage(FormStatusPage):
         formdata = self.filled
 
         r = TemplateIO(html=True)
+
+        if formdata.receipt_time:
+            r += htmltext('<div class="extra-context">')
+            r += htmltext('<h3>%s</h3>') % _('General Information')
+            r += htmltext('<p>')
+            tm = misc.localstrftime(formdata.receipt_time)
+            r += _('The form has been recorded on %(date)s with the number %(number)s.') % {
+                    'date': tm, 'number': formdata.get_display_id()}
+            r += htmltext('</p>')
+            try:
+                status_colour = formdata.get_status().colour
+            except AttributeError:
+                status_colour = 'ffffff'
+            fg_colour = misc.get_foreground_colour(status_colour)
+
+            r += htmltext('<p class="current-status"><span class="item" style="background: #%s; color: %s;"></span>' %
+                    (status_colour, fg_colour))
+            r += htmltext('<span>%s %s</span></p>') % (_('Status:'), formdata.get_status_label())
+            if formdata.formdef.workflow.criticality_levels:
+                try:
+                    level = formdata.formdef.workflow.criticality_levels[formdata.criticality_level]
+                except IndexError:
+                    pass
+                else:
+                    fg_colour = misc.get_foreground_colour(level.colour)
+                    r += htmltext('<p class="current-level"><span class="item" style="background: #%s; color: %s;"></span>' %
+                            (level.colour, fg_colour))
+                    r += htmltext('<span>%s %s</span></p>') % (_('Criticality Level:'), level.name)
+
+            if formdata.anonymised:
+                r += htmltext('<div class="infonotice">')
+                r += htmltext(_('This form has been anonymised on %(date)s.')) % {
+                        'date': formdata.anonymised.strftime(misc.date_format())}
+                r += htmltext('</div>')
+
+            r += htmltext('</div>')
+
         if formdata.submission_context or formdata.submission_channel:
             extra_context = formdata.submission_context or {}
             r += htmltext('<div class="extra-context">')
diff --git a/wcs/formdata.py b/wcs/formdata.py
index 0d0e36c..5367e13 100644
--- a/wcs/formdata.py
+++ b/wcs/formdata.py
@@ -171,6 +171,7 @@ class FormData(StorableObject):
     backoffice_submission = False
     submission_context = None
     submission_channel = None
+    criticality_level = 0
 
     workflow_data = None
     workflow_roles = None
@@ -408,6 +409,7 @@ class FormData(StorableObject):
                 'form_url': self.get_url(),
                 'form_url_backoffice': self.get_url(backoffice=True),
                 'form_uri': '%s/%s/' % (self.formdef.url_name, self.id),
+                'form_criticality_level': self.criticality_level,
             })
 
         d['form_status'] = self.get_status_label()
diff --git a/wcs/forms/backoffice.py b/wcs/forms/backoffice.py
index af0b4c4..24978cc 100644
--- a/wcs/forms/backoffice.py
+++ b/wcs/forms/backoffice.py
@@ -75,8 +75,12 @@ class FormDefUI(object):
             r += htmltext('<col />')
         r += htmltext('</colgroup>')
 
+        include_criticality_level = get_publisher().has_site_option('workflow-criticality-levels')
         r += htmltext('<thead><tr>')
-        r += htmltext('<td></td>') # lock
+        if include_criticality_level:
+            r += htmltext('<th style="width: 4ex;" data-field-sort-key="criticality_level"><span></span>')
+        else:
+            r += htmltext('<td></td>') # lock
         for f in fields:
             field_sort_key = None
             if getattr(f, 'fake', False):
@@ -209,6 +213,9 @@ class FormDefUI(object):
             url_action = ''
         root_url = get_publisher().get_root_url()
         visited_objects = get_publisher().get_visited_objects(exclude_user=get_session().user)
+        include_criticality_level = get_publisher().has_site_option('workflow-criticality-levels')
+        if include_criticality_level:
+            include_criticality_level = bool(self.formdef.workflow.criticality_levels)
         for i, filled in enumerate(items):
             classes = ['status-%s-%s' % (filled.formdef.workflow.id, filled.status)]
             if i%2:
@@ -220,12 +227,25 @@ class FormDefUI(object):
             if object_key in visited_objects:
                 classes.append('advisory-lock')
 
+            style = ''
+            if include_criticality_level:
+                try:
+                    level = filled.formdef.workflow.criticality_levels[filled.criticality_level]
+                except (IndexError, TypeError):
+                    style = ''
+                else:
+                    classes.append('criticality-level')
+                    style = ' style="border-color: #%s;"' % level.colour
+
             link = str(filled.id) + '/'
             data = ' data-link="%s"' % link
             if filled.anonymised:
                 data += ' data-anonymised="true"'
             r += htmltext('<tr class="%s"%s>' % (' '.join(classes), data))
-            r += htmltext('<td></td>') # lock
+            if include_criticality_level:
+                r += htmltext('<td %s></td>' % style) # criticality_level
+            else:
+                r += htmltext('<td></td>') # lock
             for i, f in enumerate(fields):
                 if f.type == 'id':
                     r += htmltext('<td class="cell-id"><a href="%s%s">%s</a></td>') % (link, url_action,
diff --git a/wcs/forms/common.py b/wcs/forms/common.py
index 7280ea1..f166e80 100644
--- a/wcs/forms/common.py
+++ b/wcs/forms/common.py
@@ -497,20 +497,6 @@ class FormStatusPage(Directory):
         get_logger().info('form %s - id: %s - view status' % (self.formdef.name, self.filled.id))
         self.html_top('%s - %s' % (self.formdef.name, self.filled.id))
         r = TemplateIO(html=True)
-        if self.filled.receipt_time is not None:
-            tm = misc.localstrftime(self.filled.receipt_time)
-        else:
-            tm = '???'
-        r += htmltext("<p>")
-        r += _('The form has been recorded on %(date)s with the number %(number)s.') % {
-                'date': tm, 'number': self.filled.get_display_id()}
-        r += htmltext("</p>")
-
-        if self.filled.anonymised:
-            r += htmltext('<div class="infonotice">')
-            r += htmltext(_('This form has been anonymised on %(date)s.')) % {
-                    'date': self.filled.anonymised.strftime(misc.date_format())}
-            r += htmltext('</div>')
 
         r += htmltext(self.workflow_messages())
 
diff --git a/wcs/qommon/static/css/dc2/admin.css b/wcs/qommon/static/css/dc2/admin.css
index 82d77c0..fbd3ba3 100644
--- a/wcs/qommon/static/css/dc2/admin.css
+++ b/wcs/qommon/static/css/dc2/admin.css
@@ -1251,3 +1251,34 @@ form div.page > div {
 .c-360-user-view li > span {
 	padding-left: 1ex;
 }
+
+p.current-level span,
+p.current-status span {
+	line-height: 40px;
+}
+
+p.current-level .item,
+p.current-status .item {
+	float: left;
+	display: inline-block;
+	width: 40px;
+	height: 40px;
+	border-radius: 20px;
+	border: 1px solid #aaa;
+	margin-right: 1ex;
+	margin-bottom: 1ex;
+}
+
+p.current-level .item {
+	border-radius: 0;
+}
+
+tr.criticality-level {
+	border-left-width: 5px;
+	border-left-style: solid;
+}
+
+table#listing tr.criticality-level td:first-child {
+	border-left-width: 5px;
+	border-left-style: solid;
+}
diff --git a/wcs/qommon/static/js/biglist.js b/wcs/qommon/static/js/biglist.js
index b982cdd..8b2a464 100644
--- a/wcs/qommon/static/js/biglist.js
+++ b/wcs/qommon/static/js/biglist.js
@@ -9,6 +9,7 @@ $(document).ready(
             {
                 accept: 'biglistitem',
                 scroll: true,
+                helper: 'clone',
                 helperclass: 'sortHelper',
                 activeclass :     'sortableactive',
                 hoverclass :     'sortablehover',
diff --git a/wcs/sql.py b/wcs/sql.py
index 6414ec1..dbe8e8f 100644
--- a/wcs/sql.py
+++ b/wcs/sql.py
@@ -325,7 +325,8 @@ def do_formdef_tables(formdef, conn=None, cur=None, rebuild_views=False, rebuild
         'anonymised', 'workflow_roles', 'workflow_roles_array',
         'concerned_roles_array', 'tracking_code',
         'actions_roles_array', 'backoffice_submission',
-        'submission_context', 'submission_channel'])
+        'submission_context', 'submission_channel',
+        'criticality_level'])
 
     # migrations
     if not 'fts' in existing_fields:
@@ -360,6 +361,9 @@ def do_formdef_tables(formdef, conn=None, cur=None, rebuild_views=False, rebuild
     if not 'submission_channel' in existing_fields:
         cur.execute('''ALTER TABLE %s ADD COLUMN submission_channel varchar''' % table_name)
 
+    if not 'criticality_level' in existing_fields:
+        cur.execute('''ALTER TABLE %s ADD COLUMN criticality_level integer NOT NULL DEFAULT(0)''' % table_name)
+
     # add new fields
     for field in formdef.fields:
         assert field.id is not None
@@ -565,7 +569,8 @@ def get_view_fields(formdef):
     view_fields.append(("int '%s'" % (formdef.category_id or 0), 'category_id'))
     view_fields.append(("int '%s'" % (formdef.id or 0), 'formdef_id'))
     for field in ('id', 'user_id', 'user_hash', 'receipt_time', 'status',
-            'id_display', 'submission_channel', 'backoffice_submission'):
+            'id_display', 'submission_channel', 'backoffice_submission',
+            'criticality_level'):
         view_fields.append((field, field))
     return view_fields
 
@@ -994,6 +999,7 @@ class SqlFormData(SqlMixin, wcs.formdata.FormData):
         ('backoffice_submission', 'boolean'),
         ('submission_context', 'bytea'),
         ('submission_channel', 'varchar'),
+        ('criticality_level', 'int'),
     ]
 
     def __init__(self, id=None):
@@ -1088,6 +1094,7 @@ class SqlFormData(SqlMixin, wcs.formdata.FormData):
                 'backoffice_submission': self.backoffice_submission,
                 'submission_context': self.submission_context,
                 'submission_channel': self.submission_channel,
+                'criticality_level': self.criticality_level,
         }
         if self.receipt_time:
             sql_dict['receipt_time'] = datetime.datetime.fromtimestamp(time.mktime(self.receipt_time)),
@@ -1786,7 +1793,7 @@ def get_yearly_totals(period_start=None, period_end=None, criterias=None):
     return result
 
 
-SQL_LEVEL = 13
+SQL_LEVEL = 14
 
 def migrate_global_views(conn, cur):
     cur.execute('''SELECT COUNT(*) FROM information_schema.tables
@@ -1820,7 +1827,7 @@ def migrate():
         raise RuntimeError()
     if sql_level < 1: # 1: introduction of tracking_code table
         do_tracking_code_table()
-    if sql_level < 13:
+    if sql_level < 14:
         # 2: introduction of formdef_id in views
         # 5: add concerned_roles_array, is_at_endpoint and fts to views
         # 7: add backoffice_submission to tables
@@ -1829,6 +1836,7 @@ def migrate():
         # 10: add submission_channel to tables
         # 11: add formdef_name and user_name to views
         # 13: add backoffice_submission to views
+        # 14: add criticality_level to tables & views
         migrate_views(conn, cur)
     if sql_level < 12:
         # 3: introduction of _structured for user fields
diff --git a/wcs/wf/criticality.py b/wcs/wf/criticality.py
new file mode 100644
index 0000000..c1eeb79
--- /dev/null
+++ b/wcs/wf/criticality.py
@@ -0,0 +1,79 @@
+# w.c.s. - web application for online forms
+# Copyright (C) 2005-2016  Entr'ouvert
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, see <http://www.gnu.org/licenses/>.
+
+from quixote import get_publisher
+
+from wcs.qommon.form import SingleSelectWidget
+from wcs.workflows import WorkflowStatusItem, register_item_class
+
+MODE_INC = '1'
+MODE_DEC = '2'
+MODE_SET = '3'
+
+class ModifyCriticalityWorkflowStatusItem(WorkflowStatusItem):
+    description = N_('Modify Criticality')
+    key = 'modify_criticality'
+
+    mode = None
+    absolute_value = None
+
+    def get_parameters(self):
+        return ('mode', 'absolute_value')
+
+    @classmethod
+    def is_available(cls):
+        # TODO: if we had per-workflow availability we could show this action
+        # only on the workflow where criticality levels are set.
+        return get_publisher().has_site_option('workflow-criticality-levels')
+
+    def add_parameters_widgets(self, form, parameters, prefix='', formdef=None):
+        if 'mode' in parameters:
+            form.add(SingleSelectWidget, '%smode' % prefix,
+                    title=_('Modification Mode'), value=self.mode,
+                    required=True,
+                    options=[
+                        (MODE_INC, _('Increase Level')),
+                        (MODE_DEC, _('Decrease Level')),
+                        (MODE_SET, _('Set Level'))],
+                    attrs={'data-dynamic-display-parent': 'true'})
+        if 'absolute_value' in parameters:
+            form.add(SingleSelectWidget, '%sabsolute_value' % prefix,
+                     title=_('Value'), value=self.absolute_value,
+                     options=[(str(i), x.name) for i, x in enumerate(self.parent.parent.criticality_levels)],
+                     attrs={
+                         'data-dynamic-display-child-of': '%smode' % prefix,
+                         'data-dynamic-display-value': _('Set Level'),
+                         }
+                     )
+
+    def perform(self, formdata):
+        levels = formdata.formdef.workflow.criticality_levels
+        current_level = formdata.criticality_level or 0
+        if self.mode == MODE_INC:
+            new_level = current_level + 1
+        elif self.mode == MODE_DEC:
+            new_level = current_level - 1
+        elif self.mode == MODE_SET:
+            new_level = int(self.absolute_value)
+        if new_level < 0:
+            new_level = 0
+        if new_level >= len(levels):
+            new_level = len(levels) - 1
+        if new_level != formdata.criticality_level:
+            formdata.criticality_level = new_level
+            formdata.store()
+
+register_item_class(ModifyCriticalityWorkflowStatusItem)
diff --git a/wcs/workflows.py b/wcs/workflows.py
index 49f4338..f7d707a 100644
--- a/wcs/workflows.py
+++ b/wcs/workflows.py
@@ -251,6 +251,7 @@ class Workflow(StorableObject):
     roles = None
     variables_formdef = None
     global_actions = None
+    criticality_levels = None
 
     last_modification_time = None
     last_modification_user_id = None
@@ -261,6 +262,7 @@ class Workflow(StorableObject):
         self.possible_status = []
         self.roles = {'_receiver': _('Recipient')}
         self.global_actions = []
+        self.criticality_levels = []
 
     def migrate(self):
         changed = False
@@ -941,6 +943,28 @@ class WorkflowGlobalAction(object):
             trigger_o.init_with_xml(trigger, charset, include_id=include_id)
 
 
+class WorkflowCriticalityLevel(object):
+    id = None
+    name = None
+    colour = None
+
+    def __init__(self, name=None):
+        self.name = name
+        self.id = str(random.randint(0, 100000))
+
+    def export_to_xml(self, charset, include_id=False):
+        level = ET.Element('criticality-level')
+        ET.SubElement(level, 'id').text = unicode(self.id, charset)
+        ET.SubElement(level, 'name').text = unicode(self.name, charset)
+        ET.SubElement(level, 'colour').text = unicode(self.colour, charset)
+        return level
+
+    def init_with_xml(self, elem, charset, include_id=False):
+        self.id = elem.find('id').text.encode(charset)
+        self.name = elem.find('name').text.encode(charset)
+        self.colour = elem.find('colour').text.encode(charset)
+
+
 class WorkflowStatus(object):
     id = None
     name = None
@@ -1961,5 +1985,6 @@ def load_extra():
     import wf.anonymise
     import wf.export_to_model
     import wf.resubmit
+    import wf.criticality
 
 from wf.export_to_model import ExportToModel
-- 
2.7.0