ERP5Workflow: add acquire_permission_list attribute, setup acquisition of...

ERP5Workflow: add acquire_permission_list attribute, setup acquisition of permission/roles correctly when converting, rename methods

product/ERP5Workflow/Document/PermissionRoles.py

@@ -117,7 +117,7 @@ class PermissionRoles(XMLObject):
 
     # update the state permission structure to take into account
     # the selection/non-selection of this cell
-    roles = state._getStatePermissionRolesDict()[cell_permission]
+    roles = state.getStatePermissionRolesDict()[cell_permission]
     acquired = isinstance(roles, tuple)
     roles = list(set(roles))
     if value and (cell_role not in roles):

product/ERP5Workflow/Document/State.py

@@ -66,6 +66,7 @@ class State(IdAsReferenceMixin("state_", "prefix"), XMLObject, CustomStorageMatr
   isRADContent = 1
   default_reference = ''
   state_type = ()
+  acquire_permission = []
 
   # Declarative security
   security = ClassSecurityInfo()
@@ -92,7 +93,7 @@ class State(IdAsReferenceMixin("state_", "prefix"), XMLObject, CustomStorageMatr
     # return possible transition id list:
     return self.getDestinationIdList()
 
-  def _getStatePermissionRolesDict(self):
+  def getStatePermissionRolesDict(self):
     role_dict = getattr(self, 'state_permission_roles', None)
     if role_dict is None:
       # PersistentMapping is required to have changes on the dict
@@ -100,15 +101,14 @@ class State(IdAsReferenceMixin("state_", "prefix"), XMLObject, CustomStorageMatr
       self.state_permission_roles = role_dict = PersistentMapping()
     return role_dict
   
-  def _setStatePermissionRoles(self, permission_roles):
+  def setStatePermissionRolesDict(self, permission_roles):
     self.state_permission_roles = PersistentMapping(permission_roles)
 
   def setPermission(self, permission, acquired, roles, REQUEST=None):
-      """Set a permission for this State."""
-      if acquired:
+    """
+    Set a permission for this State.
+    """
     self.state_permission_roles[permission] = list(roles)
-      else:
-          self.state_permission_roles[permission] = tuple(roles)
 
   def getDestinationReferenceList(self):
     ref_list = []
@@ -135,4 +135,4 @@ class State(IdAsReferenceMixin("state_", "prefix"), XMLObject, CustomStorageMatr
     """
     cell_permission = cell._getPermission()
     cell_role = cell._getRole()
-    cell.selected = cell_role in self._getStatePermissionRolesDict()[cell_permission]
+    cell.selected = cell_role in self.getStatePermissionRolesDict()[cell_permission]

product/ERP5Workflow/Document/Workflow.py

@@ -46,7 +46,6 @@ from Products.DCWorkflow.DCWorkflow import ValidationFailed
 from Products.DCWorkflow.Expression import StateChangeInfo
 from Products.DCWorkflow.utils import Message as _
 from Products.DCWorkflow.utils import modifyRolesForPermission
-from Products.DCWorkflow.DCWorkflow import DCWorkflowDefinition as DCWorkflow
 from Products.ERP5Type import Permissions, PropertySheet
 from Products.ERP5Type.Cache import CachingMethod
 from Products.ERP5Type.Globals import PersistentMapping
@@ -217,7 +216,13 @@ class Workflow(IdAsReferenceMixin("", "prefix"), XMLObject):
     changed = 0
     state = self._getWorkflowStateOf(document, id_only=False)
     if state is not None:
-      for permission, role_list in state.state_permission_roles.items():
+      for permission, role_list in state.getStatePermissionRolesDict().items():
+        # tuple means "don't acquire" in zope internal security and list
+        # is used when acquisition should be done
+        if permission in state.getAcquirePermissionList():
+          role_list = list(role_list)
+        else:
+          role_list = tuple(role_list)
         if modifyRolesForPermission(document, permission, role_list):
           changed = 1
     return changed
@@ -1169,10 +1174,13 @@ class Workflow(IdAsReferenceMixin("", "prefix"), XMLObject):
                          sorted(self.getManagedRoleList()),
                          base_id='cell')
       for permission in permission_list:
-        permission_roles_dict = state._getStatePermissionRolesDict()
+        permission_roles_dict = state.getStatePermissionRolesDict()
         if permission not in permission_roles_dict:
           state.state_permission_roles[permission] = []
       # remove permission from state_permission_roles dict
+      permission_to_delete = []
       for permission in permission_roles_dict:
         if permission not in permission_list:
+          permission_to_delete.append(permission)
+      for permission in permission_to_delete:
         del state.state_permission_roles[permission]

product/ERP5Workflow/Tool/WorkflowTool.py

@@ -370,10 +370,19 @@ class WorkflowTool(BaseTool, OriginalWorkflowTool):
             state.setTitle(sdef.title)
             state.setReference(sdef.id)
             state.setDescription(sdef.description)
-            permission_roles = sdef.permission_roles or {}
-            state.setStatePermissionRoles(permission_roles)
-            if permission_roles:
-              state.setCellRange(sorted(permission_roles.keys()),
+
+            acquire_permission_list = []
+            permission_roles_dict = {}
+            if sdef.permission_roles:
+              for (permission, roles) in sdef.permission_roles.items():
+                if isinstance(roles, list): # type 'list' means acquisition
+                  acquire_permission_list.append(permission)
+                permission_roles_dict[permission] = list(roles)
+
+            state.setAcquirePermission(acquire_permission_list)
+            state.setStatePermissionRolesDict(permission_roles_dict)
+            if permission_roles_dict:
+              state.setCellRange(sorted(permission_roles_dict.keys()),
                     sorted(workflow.getManagedRoleList()),
                     base_id='cell')