From ac997803b42679c9077f3b294876c4c531e74088 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?St=C3=A9phane=20Bidoul=20=28ACSONE=29?= Date: Sun, 26 Mar 2017 09:46:10 +0200 Subject: [PATCH] [IMP] account_tax_balance: faster search of taxes with moves --- account_tax_balance/__manifest__.py | 4 +- account_tax_balance/models/account_tax.py | 101 ++++++++++-------- .../views/account_tax_view.xml | 5 +- 3 files changed, 61 insertions(+), 49 deletions(-) diff --git a/account_tax_balance/__manifest__.py b/account_tax_balance/__manifest__.py index 18e5790a..4ae50fd4 100644 --- a/account_tax_balance/__manifest__.py +++ b/account_tax_balance/__manifest__.py @@ -5,10 +5,10 @@ { "name": "Tax Balance", "summary": "Compute tax balances based on date range", - "version": "10.0.1.0.1", + "version": "10.0.1.1.0", "category": "Accounting & Finance", "website": "https://www.agilebg.com/", - "author": "Agile Business Group, Therp BV, Tecnativa, " + "author": "Agile Business Group, Therp BV, Tecnativa, ACSONE SA/NV, " "Odoo Community Association (OCA)", "license": "AGPL-3", "application": False, diff --git a/account_tax_balance/models/account_tax.py b/account_tax_balance/models/account_tax.py index 6fdbf58b..9af72d21 100644 --- a/account_tax_balance/models/account_tax.py +++ b/account_tax_balance/models/account_tax.py @@ -3,8 +3,7 @@ # © 2016 Antonio Espinosa # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). -from openerp import models, fields, api -from openerp.tools.safe_eval import safe_eval +from openerp import models, fields, api, _ class AccountTax(models.Model): @@ -12,57 +11,27 @@ class AccountTax(models.Model): balance = fields.Float( string="Total Balance", compute="_compute_balance", - search='_search_balance') + ) base_balance = fields.Float( string="Total Base Balance", compute="_compute_balance", - search='_search_base_balance') + ) balance_regular = fields.Float( string="Balance", compute="_compute_balance", - search='_search_balance_regular') + ) base_balance_regular = fields.Float( string="Base Balance", compute="_compute_balance", - search='_search_base_balance_regular') + ) balance_refund = fields.Float( string="Balance Refund", compute="_compute_balance", - search='_search_balance_refund') + ) base_balance_refund = fields.Float( string="Base Balance Refund", compute="_compute_balance", - search='_search_base_balance_refund') - - def _search_balance_field(self, field, operator, value): - operators = {'>', '<', '>=', '<=', '!=', '=', '<>'} - fields = { - 'balance', 'base_balance', 'balance_regular', - 'base_balance_regular', 'balance_refund', 'base_balance_refund', - } - domain = [] - if operator in operators and field in fields: - value = float(value) if value else 0 - taxes = self.search([]).filtered( - lambda x: safe_eval( - '%.2f %s %.2f' % (x[field], operator, value))) - domain.append(('id', 'in', taxes.ids)) - return domain - - def _search_balance(self, operator, value): - return self._search_balance_field('balance', operator, value) - - def _search_base_balance(self, operator, value): - return self._search_balance_field('base_balance', operator, value) - - def _search_balance_regular(self, operator, value): - return self._search_balance_field('balance_regular', operator, value) - - def _search_base_balance_regular(self, operator, value): - return self._search_balance_field( - 'base_balance_regular', operator, value) - - def _search_balance_refund(self, operator, value): - return self._search_balance_field('balance_refund', operator, value) - - def _search_base_balance_refund(self, operator, value): - return self._search_balance_field( - 'base_balance_refund', operator, value) + ) + has_moves = fields.Boolean( + string="Has balance in period", + compute="_compute_has_moves", + search="_search_has_moves", + ) def get_context_values(self): context = self.env.context @@ -73,6 +42,52 @@ class AccountTax(models.Model): context.get('target_move', 'posted') ) + def _account_tax_ids_with_moves(self): + """ Return all account.tax ids for which there is at least + one account.move.line in the context period + for the user company. + + Caveat: this ignores record rules and ACL but it is good + enough for filtering taxes with activity during the period. + """ + req = """ + SELECT id + FROM account_tax at + WHERE + company_id = %s AND + EXISTS ( + SELECT 1 FROM account_move_Line aml + WHERE + date >= %s AND + date <= %s AND + company_id = %s AND ( + tax_line_id = at.id OR + EXISTS ( + SELECT 1 FROM account_move_line_account_tax_rel + WHERE account_move_line_id = aml.id AND + account_tax_id = at.id + ) + ) + ) + """ + 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)) + return [r[0] for r in self.env.cr.fetchall()] + + @api.multi + def _compute_has_moves(self): + ids_with_moves = set(self._account_tax_ids_with_moves()) + for tax in self: + tax.has_moves = tax.id in ids_with_moves + + @api.model + def _search_has_moves(self, operator, value): + if operator != '=' or not value: + raise ValueError(_("Unsupported search operator")) + ids_with_moves = self._account_tax_ids_with_moves() + return [('id', 'in', ids_with_moves)] + def _compute_balance(self): for tax in self: tax.balance_regular = tax.compute_balance( diff --git a/account_tax_balance/views/account_tax_view.xml b/account_tax_balance/views/account_tax_view.xml index fa1895c8..6c5ce910 100644 --- a/account_tax_balance/views/account_tax_view.xml +++ b/account_tax_balance/views/account_tax_view.xml @@ -60,10 +60,7 @@ account.tax form tree - [ - '|', '|', ('base_balance', '!=', 0), ('base_balance_regular', '!=', 0), - '|', '|', ('base_balance_refund', '!=', 0), ('balance', '!=', 0), - '|', ('balance_regular', '!=', 0), ('balance_refund', '!=', 0)] + [('has_moves', '=', True)]