new workflow actions to add (or remove) a role (#1513)

wcs/wf/roles.py

@@ -0,0 +1,84 @@
+# w.c.s. - web application for online forms
+# Copyright (C) 2005-2012  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_request
+from qommon.form import *
+from wcs.workflows import WorkflowStatusItem, register_item_class
+from wcs.roles import get_user_roles
+from wcs.users import User
+
+class AddRoleWorkflowStatusItem(WorkflowStatusItem):
+    description = N_('Add Role')
+    key = 'add_role'
+
+    role_id = None
+
+    def get_parameters(self):
+        return ('role_id',)
+
+    def add_parameters_widgets(self, form, parameters, prefix='', formdef=None):
+        if 'role_id' in parameters:
+            form.add(SingleSelectWidget, '%srole_id' % prefix,
+                     title=_('Role to Add'), value=self.role_id,
+                     options=[(None, '----')] + get_user_roles())
+
+    def perform(self, formdata):
+        if not formdata.user_id:
+            # we can't work on anonymous or user_hash'ed forms
+            return
+        user = User.get(formdata.user_id)
+        if not user.roles:
+            user.roles = []
+        if not self.role_id in user.roles:
+            user.roles.append(self.role_id)
+        user.store()
+        if get_request().user and get_request().user.id == user.id:
+            # if we changed the currently logged in user, we update it with the
+            # changes.
+            get_request().user = user
+
+register_item_class(AddRoleWorkflowStatusItem)
+
+
+class RemoveRoleWorkflowStatusItem(WorkflowStatusItem):
+    description = N_('Remove Role')
+    key = 'remove_role'
+
+    role_id = None
+
+    def get_parameters(self):
+        return ('role_id',)
+
+    def add_parameters_widgets(self, form, parameters, prefix='', formdef=None):
+        if 'role_id' in parameters:
+            form.add(SingleSelectWidget, '%srole_id' % prefix,
+                     title=_('Role to Remove'), value=self.role_id,
+                     options=[(None, '----')] + get_user_roles())
+
+    def perform(self, formdata):
+        if not formdata.user_id:
+            # we can't work on anonymous or user_hash'ed forms
+            return
+        user = User.get(formdata.user_id)
+        if user.roles and self.role_id in user.roles:
+            user.roles.remove(self.role_id)
+            user.store()
+            if get_request().user and get_request().user.id == user.id:
+                # if we changed the currently logged in user, we update it
+                # with the changes.
+                get_request().user = user
+
+register_item_class(RemoveRoleWorkflowStatusItem)

wcs/workflows.py

@@ -1107,3 +1107,4 @@ def load_extra():
     import wf.jump
     import wf.attachment
     import wf.remove
+    import wf.roles