# HG changeset patch
# User Sergi Almacellas Abellana <sergi@koolpi.com>
Add company domain on analytic entries


issue5104

review17991002

diff -r cb4e222d5c45 sale.py
--- a/trytond/trytond/modules/analytic_sale/sale.py	Mon Nov 02 15:34:18 2015 +0100
+++ b/trytond/trytond/modules/analytic_sale/sale.py	Wed Sep 28 17:06:51 2016 +0200
@@ -1,6 +1,8 @@
 # This file is part of Tryton.  The COPYRIGHT file at the top level of
 # this repository contains the full copyright notices and license terms.
+from trytond.model import fields
 from trytond.pool import Pool, PoolMeta
+from trytond.pyson import Eval, If

 from trytond.modules.analytic_account import AnalyticMixin

@@ -11,6 +13,15 @@
 class SaleLine(AnalyticMixin):
     __name__ = 'sale.line'

+    @classmethod
+    def __setup__(cls):
+        super(SaleLine, cls).__setup__()
+        cls.analytic_accounts.domain = [
+            ('company', '=', If(~Eval('_parent_sale', {}),
+                    Eval('context', {}).get('company', -1),
+                    Eval('_parent_sale', {}).get('company', -1))),
+            ]
+
     def get_invoice_line(self, invoice_type):
         pool = Pool()
         AnalyticAccountEntry = pool.get('analytic.account.entry')
@@ -32,3 +43,21 @@
     def _get_origin(cls):
         origins = super(AnalyticAccountEntry, cls)._get_origin()
         return origins + ['sale.line']
+
+    @fields.depends('origin')
+    def on_change_with_company(self, name=None):
+        pool = Pool()
+        SaleLine = pool.get('sale.line')
+        company = super(AnalyticAccountEntry, self).on_change_with_company(
+            name)
+        if isinstance(self.origin, SaleLine) and self.origin.sale:
+            company = self.origin.sale.company.id
+        return company
+
+    @classmethod
+    def search_company(cls, name, clause):
+        domain = super(AnalyticAccountEntry, cls).search_company(name, clause),
+        return ['OR',
+            domain,
+            ('origin.sale.company',) + tuple(clause[1:]) + ('sale.line',),
+            ]