Create alternate route for users that cannot force assign.

shipment.py

 #This file is part of Tryton.  The COPYRIGHT file at the top level of
 #this repository contains the full copyright notices and license terms.
 "Shipment"
 from trytond.model import ModelWorkflow, ModelView, ModelSQL, fields
 from trytond.modules.company import CompanyReport
 from trytond.wizard import Wizard
 from trytond.backend import TableHandler
 from trytond.pyson import Eval, Not, Equal, If, Or, And, Bool, In
 
 STATES = {
@@ skipping 1271 matching lines @@
             vals = self._get_inventory_moves(cursor, user, incoming_move,
                     context=context)
             if vals:
                 self.write(cursor, user, shipment.id, {
                     'inventory_moves': [('create', vals)]
                     }, context=context)
 
 ShipmentOutReturn()
 
 
-class AssignShipmentOutAskForce(ModelView):
-    'Assign Shipment Out Ask Force'
-    _name = 'stock.shipment.out.assign.ask_force'
+class AssignShipmentOutAssignFailed(ModelView):
+    'Assign Shipment Out Assign Failed'
+    _name = 'stock.shipment.out.assign.assign_failed'
     _description = __doc__
 
     inventory_moves = fields.Many2Many('stock.move', None, None,
             'Inventory Moves', readonly=True)
 
-AssignShipmentOutAskForce()
+AssignShipmentOutAssignFailed()
 
 
 class AssignShipmentOut(Wizard):
     'Assign Shipment Out'
     _name = 'stock.shipment.out.assign'
     states = {
         'init': {
             'result': {
                 'type': 'choice',
                 'next_state': '_choice',
             },
         },
+        'assign_failed': {
+            'actions': ['_moves'],
+            'result': {
+                'type': 'form',
+                'object': 'stock.shipment.out.assign.assign_failed',
+                'state': [
+                    ('end', 'Ok', 'tryton-ok', True),
+                ],
+            },
+        },
         'ask_force': {
             'actions': ['_moves'],
             'result': {
                 'type': 'form',
-                'object': 'stock.shipment.out.assign.ask_force',
+                'object': 'stock.shipment.out.assign.assign_failed',
                 'state': [
                     ('force', 'Force Assign', 'tryton-go-next'),
                     ('end', 'Ok', 'tryton-ok', True),
                 ],
             },
         },
         'force': {
             'result': {
                 'type': 'action',
                 'action': '_force',
                 'state': 'end',
             },
         },
     }
 
     def _choice(self, cursor, user, data, context=None):
         shipment_out_obj = self.pool.get('stock.shipment.out')
+        user_group_obj = self.pool.get('res.user-res.group')
+        model_data_obj = self.pool.get('ir.model.data')
+        transition_obj = self.pool.get('workflow.transition')
 
         shipment_out_obj.workflow_trigger_validate(cursor, user, data['id'],
                 'assign', context=context)
         shipment = shipment_out_obj.browse(cursor, user, data['id'],
                 context=context)
         if not [x.id for x in shipment.inventory_moves if x.state == 'draft']:
             return 'end'
         else:
-            return 'ask_force'
+            trans_id = model_data_obj.get_id(cursor, user, 'stock',
+                    'shipmentout_trans_waiting_assigned_force', context=context)
+            trans = transition_obj.read(cursor, user, trans_id, context=context)
+            user_in_group = user_group_obj.search(cursor, user, [
+                    ('uid', '=', user),
+                    ('gid', '=', trans['group']),
+                    ], limit=1, context=context)
+            if user_in_group:
+                return 'ask_force'
+            return 'assign_failed'
 
     def _moves(self, cursor, user, data, context=None):
         shipment_out_obj = self.pool.get('stock.shipment.out')
         shipment = shipment_out_obj.browse(cursor, user, data['id'],
                 context=context)
         return {'inventory_moves': [x.id for x in shipment.inventory_moves
             if x.state == 'draft']}
 
     def _force(self, cursor, user, data, context=None):
         shipment_out_obj = self.pool.get('stock.shipment.out')
@@ skipping 166 matching lines @@
 ShipmentInternal()
 
 
 class Address(ModelSQL, ModelView):
     _name = 'party.address'
     delivery = fields.Boolean('Delivery')
 
 Address()
 
 
-class AssignShipmentInternalAskForce(ModelView):
-    'Assign Shipment Internal Ask Force'
-    _name = 'stock.shipment.internal.assign.ask_force'
+class AssignShipmentInternalAssignFailed(ModelView):
+    'Assign Shipment Internal Assign Failed'
+    _name = 'stock.shipment.internal.assign.assign_failed'
     _description = __doc__
 
     moves = fields.Many2Many('stock.move', None, None, 'Moves',
             readonly=True)
 
-AssignShipmentInternalAskForce()
+AssignShipmentInternalAssignFailed()
 
 
 class AssignShipmentInternal(Wizard):
     'Assign Shipment Internal'
     _name = 'stock.shipment.internal.assign'
     states = {
         'init': {
             'result': {
                 'type': 'choice',
                 'next_state': '_choice',
             },
         },
+        'assign_failed': {
+            'actions': ['_moves'],
+            'result': {
+                'type': 'form',
+                'object': 'stock.shipment.internal.assign.assign_failed',
+                'state': [
+                    ('end', 'Ok', 'tryton-ok', True),
+                ],
+            },
+        },
         'ask_force': {
             'actions': ['_moves'],
             'result': {
                 'type': 'form',
-                'object': 'stock.shipment.internal.assign.ask_force',
+                'object': 'stock.shipment.internal.assign.assign_failed',
                 'state': [
                     ('force', 'Force Assign', 'tryton-go-next'),
                     ('end', 'Ok', 'tryton-ok', True),
                 ],
             },
         },
         'force': {
             'result': {
                 'type': 'action',
                 'action': '_force',
                 'state': 'end',
             },
         },
     }
 
     def _choice(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.internal')
+        user_group_obj = self.pool.get('res.user-res.group')
+        model_data_obj = self.pool.get('ir.model.data')
+        transition_obj = self.pool.get('workflow.transition')
 
         shipment_internal_obj.workflow_trigger_validate(cursor, user,
                 data['id'], 'assign', context=context)
         shipment = shipment_internal_obj.browse(cursor, user, data['id'],
                 context=context)
         if not [x.id for x in shipment.moves if x.state == 'draft']:
             return 'end'
         else:
-            return 'ask_force'
+            trans_id = model_data_obj.get_id(cursor, user, 'stock',
+                    'shipmentinternal_trans_waiting_assigned_force',
+                    context=context)
+            trans = transition_obj.read(cursor, user, trans_id,
+                    context=context)
+            user_in_group = user_group_obj.search(cursor, user, [
+                    ('uid', '=', user),
+                    ('gid', '=', trans['group']),
+                    ], limit=1, context=context)
+            if user_in_group:
+                return 'ask_force'
+            return 'assign_failed'
 
     def _moves(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.internal')
         shipment = shipment_internal_obj.browse(cursor, user, data['id'],
                 context=context)
         return {'moves': [x.id for x in shipment.moves if x.state == 'draft']}
 
     def _force(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.internal')
 
         shipment_internal_obj.workflow_trigger_validate(cursor, user,
                 data['id'], 'force_assign', context=context)
         return {}
 
 AssignShipmentInternal()
 
 
-class AssignShipmentInReturnAskForce(ModelView):
-    'Assign Supplier Return Shipment Ask Force'
-    _name = 'stock.shipment.in.return.assign.ask_force'
+class AssignShipmentInReturnAssignFailed(ModelView):
+    'Assign Supplier Return Shipment Assign Failed'
+    _name = 'stock.shipment.in.return.assign.assign_failed'
     _description = __doc__
 
     moves = fields.Many2Many('stock.move', None, None, 'Moves',
             readonly=True)
 
-AssignShipmentInReturnAskForce()
+AssignShipmentInReturnAssignFailed()
 
 
 class AssignShipmentInReturn(Wizard):
     'Assign Supplier Return Shipment'
     _name = 'stock.shipment.in.return.assign'
     states = {
         'init': {
             'result': {
                 'type': 'choice',
                 'next_state': '_choice',
             },
         },
+        'assign_failed': {
+            'actions': ['_moves'],
+            'result': {
+                'type': 'form',
+                'object': 'stock.shipment.in.return.assign.assign_failed',
+                'state': [
+                    ('end', 'Ok', 'tryton-ok', True),
+                ],
+            },
+        },
         'ask_force': {
             'actions': ['_moves'],
             'result': {
                 'type': 'form',
-                'object': 'stock.shipment.in.return.assign.ask_force',
+                'object': 'stock.shipment.in.return.assign.assign_failed',
                 'state': [
                     ('force', 'Force Assign', 'tryton-go-next'),
                     ('end', 'Ok', 'tryton-ok', True),
                 ],
             },
         },
         'force': {
             'result': {
                 'type': 'action',
                 'action': '_force',
                 'state': 'end',
             },
         },
     }
 
     def _choice(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.in.return')
+        user_group_obj = self.pool.get('res.user-res.group')
+        model_data_obj = self.pool.get('ir.model.data')
+        transition_obj = self.pool.get('workflow.transition')
 
         shipment_internal_obj.workflow_trigger_validate(cursor, user,
                 data['id'], 'assign', context=context)
         shipment = shipment_internal_obj.browse(cursor, user, data['id'],
                 context=context)
         if not [x.id for x in shipment.moves if x.state == 'draft']:
             return 'end'
         else:
-            return 'ask_force'
+            trans_id = model_data_obj.get_id(cursor, user, 'stock',
+                    'shipment_in_return_trans_waiting_assigned_force',
+                    context=context)
+            trans = transition_obj.read(cursor, user, trans_id, context=context)
+            user_in_group = user_group_obj.search(cursor, user, [
+                    ('uid', '=', user),
+                    ('gid', '=', trans['group']),
+                    ], limit=1, context=context)
+            if user_in_group:
+                return 'ask_force'
+            return 'assign_failed'
 
     def _moves(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.in.return')
         shipment = shipment_internal_obj.browse(cursor, user, data['id'],
                 context=context)
         return {'moves': [x.id for x in shipment.moves if x.state == 'draft']}
 
     def _force(self, cursor, user, data, context=None):
         shipment_internal_obj = self.pool.get('stock.shipment.in.return')
 
@@ skipping 289 matching lines @@
                 'to_location_ids' : to_location_ids}
 
 
     def get_compare_key(self, move, compare_context):
         from_location_ids = compare_context['from_location_ids']
         to_location_ids = compare_context['to_location_ids']
         return [from_location_ids.index(move.from_location.id),
                 to_location_ids.index(move.to_location.id)]
 
 InteralShipmentReport()