diff --git a/inventory.py b/inventory.py index 74bae2dfba..d1c1817f5c 100644 --- a/inventory.py +++ b/inventory.py @@ -26,6 +26,10 @@ class Inventory(OSV): moves = fields.Many2Many( 'stock.move', 'inventory_move_rel', 'inventory', 'move', 'Generated moves') + company = fields.Many2One( + 'company.company', 'Company', required=True, states=STATES,) + + state = fields.Selection( [('open','Open'), ('done','Done'), @@ -47,6 +51,13 @@ class Inventory(OSV): def default_date(self, cursor, user, context=None): return datetime.datetime.today() + def default_company(self, cursor, user, context=None): + if context.get('company'): + return context.get('company') + user_obj = self.pool.get('res.user') + user = user_obj.browse(cursor, user, user, context=context) + return user.company.id + def set_state_cancel(self, cursor, user, ids, context=None): self.write(cursor, user, ids, { 'state': 'cancel', @@ -57,7 +68,6 @@ class Inventory(OSV): 'state': 'open', }, context=context) - def set_state_done(self, cursor, user, ids, context=None): self.write(cursor, user, ids, { 'state': 'done', @@ -226,6 +236,7 @@ class CompleteInventory(Wizard): {'product': product, 'uom': uom, 'quantity': qty, + 'company': inventory.company.id, 'inventory': inventory.id,}, context=context) return {} diff --git a/inventory.xml b/inventory.xml index 0442bbd430..b1abc1dcd2 100644 --- a/inventory.xml +++ b/inventory.xml @@ -11,21 +11,23 @@ form +