[IMP] Ease extension if multicompany fiscal unit

account_tax_balance/models/account_tax.py

@@ -34,10 +34,11 @@ class AccountTax(models.Model):
 
     def get_context_values(self):
         context = self.env.context
+        actual_company_id = context.get("company_id", self.env.user.company_id.id)
         return (
             context.get('from_date', fields.Date.context_today(self)),
             context.get('to_date', fields.Date.context_today(self)),
-            context.get('company_id', self.env.user.company_id.id),
+            context.get('company_ids', [actual_company_id]),
             context.get('target_move', 'posted'),
         )
 
@@ -49,17 +50,19 @@ class AccountTax(models.Model):
         Caveat: this ignores record rules and ACL but it is good
         enough for filtering taxes with activity during the period.
         """
+        from_date, to_date, company_ids, _ = self.get_context_values()
+        company_ids = tuple(company_ids)
         req = """
             SELECT id
             FROM account_tax at
             WHERE
-            company_id = %s AND
+            company_id in %s AND
             EXISTS (
               SELECT 1 FROM account_move_Line aml
               WHERE
                 date >= %s AND
                 date <= %s AND
-                company_id = %s AND (
+                company_id in %s AND (
                   tax_line_id = at.id OR
                   EXISTS (
                     SELECT 1 FROM account_move_line_account_tax_rel
@@ -69,9 +72,8 @@ class AccountTax(models.Model):
                 )
             )
         """
-        from_date, to_date, company_id, target_move = self.get_context_values()
         self.env.cr.execute(
-            req, (company_id, from_date, to_date, company_id))
+            req, (company_ids, from_date, to_date, company_ids))
         return [r[0] for r in self.env.cr.fetchall()]
 
     @api.multi
@@ -121,11 +123,11 @@ class AccountTax(models.Model):
             state = []
         return state
 
-    def get_move_line_partial_domain(self, from_date, to_date, company_id):
+    def get_move_line_partial_domain(self, from_date, to_date, company_ids):
         return [
             ('date', '<=', to_date),
             ('date', '>=', from_date),
-            ('company_id', '=', company_id),
+            ('company_id', 'in', company_ids),
         ]
 
     def compute_balance(self, tax_or_base='tax', move_type=None):
@@ -161,11 +163,12 @@ class AccountTax(models.Model):
         return domain
 
     def get_move_lines_domain(self, tax_or_base='tax', move_type=None):
-        from_date, to_date, company_id, target_move = self.get_context_values()
+        from_date, to_date, company_ids, target_move = \
+            self.get_context_values()
         state_list = self.get_target_state_list(target_move)
         type_list = self.get_target_type_list(move_type)
         domain = self.get_move_line_partial_domain(
-            from_date, to_date, company_id)
+            from_date, to_date, company_ids)
         balance_domain = []
         if tax_or_base == 'tax':
             balance_domain = self.get_balance_domain(state_list, type_list)

account_tax_balance/wizard/open_tax_balances.py

@@ -54,6 +54,6 @@ class WizardOpenTaxBalances(models.TransientModel):
             'from_date': self.from_date,
             'to_date': self.to_date,
             'target_move': self.target_move,
-            'company_id': self.company_id.id,
+            'company_ids': [self.company_id.id],
         }
         return vals