Merge d8b967b into 90641e1

OCA · Jul 5, 2017 · 87c04d1 · 87c04d1
2 parents 90641e1 + d8b967b
commit 87c04d1
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/account_operating_unit/models/invoice.py b/account_operating_unit/models/invoice.py
@@ -13,7 +13,10 @@ class AccountInvoice(models.Model):
     operating_unit_id = fields.Many2one('operating.unit', 'Operating Unit',
                                         default=lambda self:
                                         self.env['res.users'].
-                                        operating_unit_default_get(self._uid))
+                                        operating_unit_default_get(self._uid),
+                                        readonly=True,
+                                        states={'draft': [('readonly',
+                                                           False)]})
 
     @api.multi
     def finalize_invoice_move_lines(self, move_lines):

diff --git a/purchase_operating_unit/models/purchase.py b/purchase_operating_unit/models/purchase.py
@@ -24,15 +24,24 @@ def _default_picking_type(self):
             res = types[:1].id
         return res
 
+    READONLY_STATES = {
+        'purchase': [('readonly', True)],
+        'done': [('readonly', True)],
+        'cancel': [('readonly', True)],
+    }
+
     operating_unit_id = fields.Many2one(
         comodel_name='operating.unit',
         string='Operating Unit',
+        states=READONLY_STATES,
         default=lambda self: (self.env['res.users'].
                               operating_unit_default_get(self.env.uid))
     )
+
     requesting_operating_unit_id = fields.Many2one(
         comodel_name='operating.unit',
         string='Requesting Operating Unit',
+        states=READONLY_STATES,
         default=lambda self: (self.env['res.users'].
                               operating_unit_default_get(self.env.uid))
     )

diff --git a/sale_operating_unit/models/sale.py b/sale_operating_unit/models/sale.py
@@ -21,7 +21,10 @@ def _default_operating_unit(self):
     operating_unit_id = fields.Many2one(
         comodel_name='operating.unit',
         string='Operating Unit',
-        default=_default_operating_unit
+        default=_default_operating_unit,
+        readonly=True,
+        states={'draft': [('readonly', False)],
+                'sent': [('readonly', False)]}
     )
 
     @api.onchange('team_id')