diff --git a/invoice.py b/invoice.py index 55d463c102..2b236ad777 100644 --- a/invoice.py +++ b/invoice.py @@ -118,6 +118,16 @@ class InvoiceLine(metaclass=PoolMeta): category = self.origin.unit.category.id return category + def get_warehouse(self, name): + pool = Pool() + PurchaseLine = pool.get('purchase.line') + warehouse = super().get_warehouse(name) + if (not warehouse + and isinstance(self.origin, PurchaseLine) + and self.origin.purchase.warehouse): + warehouse = self.origin.purchase.warehouse.id + return warehouse + @property def origin_name(self): pool = Pool()