From 438d620eb6102026bea5a703420c77b9a30abd9b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adri=C3=A0=20Gil=20Sorribes?= Date: Tue, 24 Dec 2019 09:09:12 +0100 Subject: [PATCH] [IMP] account_move_line_purchase_info: black, isort --- .../__manifest__.py | 10 +- .../migrations/12.0.2.0.0/pre-migration.py | 10 +- .../models/__init__.py | 1 - .../models/account_invoice.py | 14 +- .../models/account_move.py | 18 +- .../models/purchase_order_line.py | 5 +- .../models/stock_move.py | 10 +- .../test_account_move_line_purchase_info.py | 256 +++++++++--------- 8 files changed, 169 insertions(+), 155 deletions(-) diff --git a/account_move_line_purchase_info/__manifest__.py b/account_move_line_purchase_info/__manifest__.py index 5025f018..295bbbcf 100644 --- a/account_move_line_purchase_info/__manifest__.py +++ b/account_move_line_purchase_info/__manifest__.py @@ -6,15 +6,11 @@ "name": "Account Move Line Purchase Info", "summary": "Introduces the purchase order line to the journal items", "version": "12.0.2.0.0", - "author": "Eficent, " - "Odoo Community Association (OCA)", + "author": "Eficent, " "Odoo Community Association (OCA)", "website": "https://www.github.com/OCA/account-financial-tools", "category": "Generic", "depends": ["purchase_stock"], "license": "AGPL-3", - "data": [ - "security/account_security.xml", - "views/account_move_view.xml", - ], - 'installable': True, + "data": ["security/account_security.xml", "views/account_move_view.xml"], + "installable": True, } diff --git a/account_move_line_purchase_info/migrations/12.0.2.0.0/pre-migration.py b/account_move_line_purchase_info/migrations/12.0.2.0.0/pre-migration.py index 77f8331c..b3f80199 100644 --- a/account_move_line_purchase_info/migrations/12.0.2.0.0/pre-migration.py +++ b/account_move_line_purchase_info/migrations/12.0.2.0.0/pre-migration.py @@ -4,23 +4,25 @@ import logging - _logger = logging.getLogger(__name__) __name__ = "Upgrade to 12.0.2.0.0" def update_purchase_id_column(cr): - cr.execute("""SELECT column_name + cr.execute( + """SELECT column_name FROM information_schema.columns WHERE table_name='account_move_line' AND - column_name='purchase_id'""") + column_name='purchase_id'""" + ) if not cr.fetchone(): _logger.info("""Add column purchase_id to account_move_line""") cr.execute( """ ALTER TABLE account_move_line ADD COLUMN purchase_id integer; - """) + """ + ) _logger.info("""Updating values for purchase_id in account_move_line""") cr.execute( """ diff --git a/account_move_line_purchase_info/models/__init__.py b/account_move_line_purchase_info/models/__init__.py index eedcf983..a8322b5f 100644 --- a/account_move_line_purchase_info/models/__init__.py +++ b/account_move_line_purchase_info/models/__init__.py @@ -2,4 +2,3 @@ from . import account_move from . import account_invoice from . import purchase_order_line from . import stock_move - diff --git a/account_move_line_purchase_info/models/account_invoice.py b/account_move_line_purchase_info/models/account_invoice.py index c1f842c6..5fd96ac0 100644 --- a/account_move_line_purchase_info/models/account_invoice.py +++ b/account_move_line_purchase_info/models/account_invoice.py @@ -7,23 +7,23 @@ from odoo import api, models class AccountInvoice(models.Model): - _inherit = 'account.invoice' + _inherit = "account.invoice" @api.model def invoice_line_move_line_get(self): res = super(AccountInvoice, self).invoice_line_move_line_get() - invoice_line_model = self.env['account.invoice.line'] + invoice_line_model = self.env["account.invoice.line"] for move_line_dict in res: - if 'invl_id' in move_line_dict: - line = invoice_line_model.browse(move_line_dict['invl_id']) - move_line_dict['purchase_line_id'] = line.purchase_line_id.id + if "invl_id" in move_line_dict: + line = invoice_line_model.browse(move_line_dict["invl_id"]) + move_line_dict["purchase_line_id"] = line.purchase_line_id.id return res @api.model def line_get_convert(self, line, part): res = super(AccountInvoice, self).line_get_convert(line, part) - if line.get('purchase_line_id', False): - res['purchase_line_id'] = line.get('purchase_line_id') + if line.get("purchase_line_id", False): + res["purchase_line_id"] = line.get("purchase_line_id") return res diff --git a/account_move_line_purchase_info/models/account_move.py b/account_move_line_purchase_info/models/account_move.py index da1cfb61..282b1108 100644 --- a/account_move_line_purchase_info/models/account_move.py +++ b/account_move_line_purchase_info/models/account_move.py @@ -7,17 +7,19 @@ from odoo import fields, models class AccountMoveLine(models.Model): - _inherit = 'account.move.line' + _inherit = "account.move.line" purchase_line_id = fields.Many2one( - comodel_name='purchase.order.line', - string='Purchase Order Line', - ondelete='set null', index=True, + comodel_name="purchase.order.line", + string="Purchase Order Line", + ondelete="set null", + index=True, ) purchase_id = fields.Many2one( - comodel_name='purchase.order', - related='purchase_line_id.order_id', - string='Purchase Order', - store=True, index=True, + comodel_name="purchase.order", + related="purchase_line_id.order_id", + string="Purchase Order", + store=True, + index=True, ) diff --git a/account_move_line_purchase_info/models/purchase_order_line.py b/account_move_line_purchase_info/models/purchase_order_line.py index 6e6b5581..6019faef 100644 --- a/account_move_line_purchase_info/models/purchase_order_line.py +++ b/account_move_line_purchase_info/models/purchase_order_line.py @@ -13,8 +13,7 @@ class PurchaseOrderLine(models.Model): orig_name = dict(super(PurchaseOrderLine, self).name_get()) for line in self: name = orig_name[line.id] - if self.env.context.get('po_line_info', False): - name = "[%s] %s (%s)" % (line.order_id.name, name, - line.order_id.state) + if self.env.context.get("po_line_info", False): + name = "[{}] {} ({})".format(line.order_id.name, name, line.order_id.state) result.append((line.id, name)) return result diff --git a/account_move_line_purchase_info/models/stock_move.py b/account_move_line_purchase_info/models/stock_move.py index c9047910..f306cedc 100644 --- a/account_move_line_purchase_info/models/stock_move.py +++ b/account_move_line_purchase_info/models/stock_move.py @@ -9,10 +9,12 @@ class StockMove(models.Model): _inherit = "stock.move" @api.model - def _prepare_account_move_line(self, qty, cost, - credit_account_id, debit_account_id): + def _prepare_account_move_line( + self, qty, cost, credit_account_id, debit_account_id + ): res = super(StockMove, self)._prepare_account_move_line( - qty, cost, credit_account_id, debit_account_id) + qty, cost, credit_account_id, debit_account_id + ) for line in res: - line[2]['purchase_line_id'] = self.purchase_line_id.id + line[2]["purchase_line_id"] = self.purchase_line_id.id return res diff --git a/account_move_line_purchase_info/tests/test_account_move_line_purchase_info.py b/account_move_line_purchase_info/tests/test_account_move_line_purchase_info.py index c09df15b..d56ab2c7 100644 --- a/account_move_line_purchase_info/tests/test_account_move_line_purchase_info.py +++ b/account_move_line_purchase_info/tests/test_account_move_line_purchase_info.py @@ -2,118 +2,118 @@ # (www.eficent.com) # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html). -from odoo.tests import common from odoo import fields +from odoo.tests import common class TestAccountMoveLinePurchaseInfo(common.TransactionCase): - def setUp(self): super(TestAccountMoveLinePurchaseInfo, self).setUp() - self.purchase_model = self.env['purchase.order'] - self.purchase_line_model = self.env['purchase.order.line'] - self.invoice_model = self.env['account.invoice'] - self.invoice_line_model = self.env['account.invoice.line'] - self.product_model = self.env['product.product'] - self.product_ctg_model = self.env['product.category'] - self.acc_type_model = self.env['account.account.type'] - self.account_model = self.env['account.account'] - self.aml_model = self.env['account.move.line'] - self.res_users_model = self.env['res.users'] + self.purchase_model = self.env["purchase.order"] + self.purchase_line_model = self.env["purchase.order.line"] + self.invoice_model = self.env["account.invoice"] + self.invoice_line_model = self.env["account.invoice.line"] + self.product_model = self.env["product.product"] + self.product_ctg_model = self.env["product.category"] + self.acc_type_model = self.env["account.account.type"] + self.account_model = self.env["account.account"] + self.aml_model = self.env["account.move.line"] + self.res_users_model = self.env["res.users"] - self.partner1 = self.env.ref('base.res_partner_1') - self.location_stock = self.env.ref('stock.stock_location_stock') - self.company = self.env.ref('base.main_company') - self.group_purchase_user = self.env.ref('purchase.group_purchase_user') - self.group_account_invoice = self.env.ref( - 'account.group_account_invoice') - self.group_account_manager = self.env.ref( - 'account.group_account_manager') + self.partner1 = self.env.ref("base.res_partner_1") + self.location_stock = self.env.ref("stock.stock_location_stock") + self.company = self.env.ref("base.main_company") + self.group_purchase_user = self.env.ref("purchase.group_purchase_user") + self.group_account_invoice = self.env.ref("account.group_account_invoice") + self.group_account_manager = self.env.ref("account.group_account_manager") # Create account for Goods Received Not Invoiced - acc_type = self._create_account_type('equity', 'other') - name = 'Goods Received Not Invoiced' - code = 'grni' - self.account_grni = self._create_account(acc_type, name, code, - self.company) + acc_type = self._create_account_type("equity", "other") + name = "Goods Received Not Invoiced" + code = "grni" + self.account_grni = self._create_account(acc_type, name, code, self.company) # Create account for Cost of Goods Sold - acc_type = self._create_account_type('expense', 'other') - name = 'Cost of Goods Sold' - code = 'cogs' - self.account_cogs = self._create_account(acc_type, name, code, - self.company) + acc_type = self._create_account_type("expense", "other") + name = "Cost of Goods Sold" + code = "cogs" + self.account_cogs = self._create_account(acc_type, name, code, self.company) # Create account for Inventory - acc_type = self._create_account_type('asset', 'other') - name = 'Inventory' - code = 'inventory' - self.account_inventory = self._create_account(acc_type, name, code, - self.company) + acc_type = self._create_account_type("asset", "other") + name = "Inventory" + code = "inventory" + self.account_inventory = self._create_account( + acc_type, name, code, self.company + ) # Create Product self.product = self._create_product() # Create users - self.purchase_user = self._create_user('purchase_user', - [self.group_purchase_user, - self.group_account_invoice], - self.company) - self.account_invoice = self._create_user('account_invoice', - [self.group_account_invoice], - self.company) - self.account_manager = self._create_user('account_manager', - [self.group_account_manager], - self.company) + self.purchase_user = self._create_user( + "purchase_user", + [self.group_purchase_user, self.group_account_invoice], + self.company, + ) + self.account_invoice = self._create_user( + "account_invoice", [self.group_account_invoice], self.company + ) + self.account_manager = self._create_user( + "account_manager", [self.group_account_manager], self.company + ) def _create_user(self, login, groups, company): """ Create a user.""" group_ids = [group.id for group in groups] - user = \ - self.res_users_model.with_context( - {'no_reset_password': True}).create({ - 'name': 'Test User', - 'login': login, - 'password': 'demo', - 'email': 'test@yourcompany.com', - 'company_id': company.id, - 'company_ids': [(4, company.id)], - 'groups_id': [(6, 0, group_ids)] - }) + user = self.res_users_model.with_context({"no_reset_password": True}).create( + { + "name": "Test User", + "login": login, + "password": "demo", + "email": "test@yourcompany.com", + "company_id": company.id, + "company_ids": [(4, company.id)], + "groups_id": [(6, 0, group_ids)], + } + ) return user.id def _create_account_type(self, name, type): - acc_type = self.acc_type_model.create({ - 'name': name, - 'type': type - }) + acc_type = self.acc_type_model.create({"name": name, "type": type}) return acc_type def _create_account(self, acc_type, name, code, company): """Create an account.""" - account = self.account_model.create({ - 'name': name, - 'code': code, - 'user_type_id': acc_type.id, - 'company_id': company.id - }) + account = self.account_model.create( + { + "name": name, + "code": code, + "user_type_id": acc_type.id, + "company_id": company.id, + } + ) return account def _create_product(self): """Create a Product.""" # group_ids = [group.id for group in groups] - product_ctg = self.product_ctg_model.create({ - 'name': 'test_product_ctg', - 'property_stock_valuation_account_id': self.account_inventory.id, - 'property_valuation': 'real_time', - 'property_stock_account_input_categ_id': self.account_grni.id, - 'property_stock_account_output_categ_id': self.account_cogs.id, - }) - product = self.product_model.create({ - 'name': 'test_product', - 'categ_id': product_ctg.id, - 'type': 'product', - 'standard_price': 1.0, - 'list_price': 1.0, - }) + product_ctg = self.product_ctg_model.create( + { + "name": "test_product_ctg", + "property_stock_valuation_account_id": self.account_inventory.id, + "property_valuation": "real_time", + "property_stock_account_input_categ_id": self.account_grni.id, + "property_stock_account_output_categ_id": self.account_cogs.id, + } + ) + product = self.product_model.create( + { + "name": "test_product", + "categ_id": product_ctg.id, + "type": "product", + "standard_price": 1.0, + "list_price": 1.0, + } + ) return product def _create_purchase(self, line_products): @@ -124,47 +124,48 @@ class TestAccountMoveLinePurchaseInfo(common.TransactionCase): lines = [] for product, qty in line_products: line_values = { - 'name': product.name, - 'product_id': product.id, - 'product_qty': qty, - 'product_uom': product.uom_id.id, - 'price_unit': 500, - 'date_planned': fields.datetime.now() + "name": product.name, + "product_id": product.id, + "product_qty": qty, + "product_uom": product.uom_id.id, + "price_unit": 500, + "date_planned": fields.datetime.now(), } - lines.append( - (0, 0, line_values) - ) - return self.purchase_model.create({ - 'partner_id': self.partner1.id, - 'order_line': lines, - }) + lines.append((0, 0, line_values)) + return self.purchase_model.create( + {"partner_id": self.partner1.id, "order_line": lines} + ) def _get_balance(self, domain): """ Call read_group method and return the balance of particular account. """ - aml_rec = self.aml_model.read_group(domain, - ['debit', 'credit', 'account_id'], - ['account_id']) + aml_rec = self.aml_model.read_group( + domain, ["debit", "credit", "account_id"], ["account_id"] + ) if aml_rec: - return aml_rec[0].get('debit', 0) - aml_rec[0].get('credit', 0) + return aml_rec[0].get("debit", 0) - aml_rec[0].get("credit", 0) else: return 0.0 - def _check_account_balance(self, account_id, purchase_line=None, - expected_balance=0.0): + def _check_account_balance( + self, account_id, purchase_line=None, expected_balance=0.0 + ): """ Check the balance of the account """ - domain = [('account_id', '=', account_id)] + domain = [("account_id", "=", account_id)] if purchase_line: - domain.extend([('purchase_line_id', '=', purchase_line.id)]) + domain.extend([("purchase_line_id", "=", purchase_line.id)]) balance = self._get_balance(domain) if purchase_line: - self.assertEqual(balance, expected_balance, - 'Balance is not %s for Purchase Line %s.' - % (str(expected_balance), purchase_line.name)) + self.assertEqual( + balance, + expected_balance, + "Balance is not %s for Purchase Line %s." + % (str(expected_balance), purchase_line.name), + ) def test_purchase_invoice(self): """Test that the po line moves from the purchase order to the @@ -178,35 +179,48 @@ class TestAccountMoveLinePurchaseInfo(common.TransactionCase): purchase.button_confirm() picking = purchase.picking_ids[0] picking.action_confirm() - picking.move_lines.write({'quantity_done': 1.0}) + picking.move_lines.write({"quantity_done": 1.0}) picking.button_validate() expected_balance = 1.0 - self._check_account_balance(self.account_inventory.id, - purchase_line=po_line, - expected_balance=expected_balance) + self._check_account_balance( + self.account_inventory.id, + purchase_line=po_line, + expected_balance=expected_balance, + ) - invoice = self.invoice_model.create({ - 'partner_id': self.partner1.id, - 'purchase_id': purchase.id, - 'account_id': purchase.partner_id.property_account_payable_id.id, - }) + invoice = self.invoice_model.create( + { + "partner_id": self.partner1.id, + "purchase_id": purchase.id, + "account_id": purchase.partner_id.property_account_payable_id.id, + } + ) invoice.purchase_order_change() invoice.action_invoice_open() for aml in invoice.move_id.line_ids: if aml.product_id == po_line.product_id and aml.invoice_id: - self.assertEqual(aml.purchase_line_id, po_line, - 'Purchase Order line has not been copied ' - 'from the invoice to the account move line.') + self.assertEqual( + aml.purchase_line_id, + po_line, + "Purchase Order line has not been copied " + "from the invoice to the account move line.", + ) def test_name_get(self): purchase = self._create_purchase([(self.product, 1)]) po_line = purchase.order_line[0] - name_get = po_line.with_context({'po_line_info': True}).name_get() - self.assertEqual(name_get, [(po_line.id, "[%s] %s (%s)" % ( - po_line.order_id.name, po_line.name, - po_line.order_id.state, - ))]) + name_get = po_line.with_context({"po_line_info": True}).name_get() + self.assertEqual( + name_get, + [ + ( + po_line.id, + "[%s] %s (%s)" + % (po_line.order_id.name, po_line.name, po_line.order_id.state), + ) + ], + ) name_get_no_ctx = po_line.name_get() self.assertEqual(name_get_no_ctx, [(po_line.id, po_line.name)])