diff --git a/addons/om_account_asset/models/account_move.py b/addons/om_account_asset/models/account_move.py new file mode 100644 index 0000000..3c599f9 --- /dev/null +++ b/addons/om_account_asset/models/account_move.py @@ -0,0 +1,160 @@ +from dateutil.relativedelta import relativedelta +from odoo import api, fields, models, _ +from odoo.exceptions import UserError, ValidationError + + +class AccountMove(models.Model): + _inherit = 'account.move' + + asset_ids = fields.One2many( + 'account.asset.asset', 'invoice_id', string="Assets" + ) + + def button_draft(self): + res = super(AccountMove, self).button_draft() + for move in self: + if any(asset_id.state != 'draft' for asset_id in move.asset_ids): + raise ValidationError(_( + 'You cannot reset to draft for an entry having a posted asset')) + if move.asset_ids: + move.asset_ids.sudo().write({'active': False}) + for asset in move.asset_ids: + asset.sudo().message_post(body=_("Vendor bill cancelled.")) + return res + + @api.model + def _refund_cleanup_lines(self, lines): + result = super(AccountMove, self)._refund_cleanup_lines(lines) + for i, line in enumerate(lines): + for name, field in line._fields.items(): + if name == 'asset_category_id': + result[i][2][name] = False + break + return result + + def action_cancel(self): + res = super(AccountMove, self).action_cancel() + assets = self.env['account.asset.asset'].sudo().search( + [('invoice_id', 'in', self.ids)]) + if assets: + assets.sudo().write({'active': False}) + for asset in assets: + asset.sudo().message_post(body=_("Vendor bill cancelled.")) + return res + + def action_post(self): + result = super(AccountMove, self).action_post() + for inv in self: + context = dict(self.env.context) + context.pop('default_type', None) + for mv_line in inv.invoice_line_ids: + mv_line.with_context(context).asset_create() + return result + + +class AccountMoveLine(models.Model): + _inherit = 'account.move.line' + + asset_category_id = fields.Many2one( + 'account.asset.category', string='Asset Category' + ) + asset_start_date = fields.Date( + string='Asset Start Date', compute='_get_asset_date', + readonly=True, store=True + ) + asset_end_date = fields.Date( + string='Asset End Date', compute='_get_asset_date', + readonly=True, store=True + ) + asset_mrr = fields.Float( + string='Monthly Recurring Revenue', compute='_get_asset_date', + readonly=True, store=True + ) + + @api.model + def default_get(self, fields): + res = super(AccountMoveLine, self).default_get(fields) + if self.env.context.get('create_bill') and not self.asset_category_id: + if self.product_id and self.move_id.move_type == 'out_invoice' and \ + self.product_id.product_tmpl_id.deferred_revenue_category_id: + self.asset_category_id = self.product_id.product_tmpl_id.deferred_revenue_category_id.id + elif self.product_id and self.product_id.product_tmpl_id.asset_category_id and \ + self.move_id.move_type == 'in_invoice': + self.asset_category_id = self.product_id.product_tmpl_id.asset_category_id.id + self.onchange_asset_category_id() + return res + + @api.depends('asset_category_id', 'move_id.invoice_date') + def _get_asset_date(self): + for rec in self: + rec.asset_mrr = 0 + rec.asset_start_date = False + rec.asset_end_date = False + cat = rec.asset_category_id + if cat: + if cat.method_number == 0 or cat.method_period == 0: + raise UserError(_('The number of depreciations or the period length of ' + 'your asset category cannot be 0.')) + months = cat.method_number * cat.method_period + if rec.move_id.move_type in ['out_invoice', 'out_refund']: + price_subtotal = self.currency_id._convert( + self.price_subtotal, + self.company_currency_id, + self.company_id, + self.move_id.invoice_date or fields.Date.context_today( + self)) + + rec.asset_mrr = price_subtotal / months + if rec.move_id.invoice_date: + start_date = rec.move_id.invoice_date.replace(day=1) + end_date = (start_date + relativedelta(months=months, days=-1)) + rec.asset_start_date = start_date + rec.asset_end_date = end_date + + def asset_create(self): + if self.asset_category_id: + price_subtotal = self.currency_id._convert( + self.price_subtotal, + self.company_currency_id, + self.company_id, + self.move_id.invoice_date or fields.Date.context_today( + self)) + vals = { + 'name': self.name, + 'code': self.name or False, + 'category_id': self.asset_category_id.id, + 'value': price_subtotal, + 'partner_id': self.move_id.partner_id.id, + 'company_id': self.move_id.company_id.id, + 'currency_id': self.move_id.company_currency_id.id, + 'date': self.move_id.invoice_date or self.move_id.date, + 'invoice_id': self.move_id.id, + } + changed_vals = self.env['account.asset.asset'].onchange_category_id_values(vals['category_id']) + vals.update(changed_vals['value']) + asset = self.env['account.asset.asset'].create(vals) + if self.asset_category_id.open_asset: + if asset.date_first_depreciation == 'manual': + asset.first_depreciation_manual_date = asset.date + asset.validate() + return True + + @api.onchange('asset_category_id', 'product_uom_id') + def onchange_asset_category_id(self): + if self.move_id.move_type == 'out_invoice' and self.asset_category_id: + self.account_id = self.asset_category_id.account_asset_id.id + elif self.move_id.move_type == 'in_invoice' and self.asset_category_id: + self.account_id = self.asset_category_id.account_asset_id.id + + @api.onchange('product_id') + def _inverse_product_id(self): + res = super(AccountMoveLine, self)._inverse_product_id() + for rec in self: + if rec.product_id: + if rec.move_id.move_type == 'out_invoice': + rec.asset_category_id = rec.product_id.product_tmpl_id.deferred_revenue_category_id.id + elif rec.move_id.move_type == 'in_invoice': + rec.asset_category_id = rec.product_id.product_tmpl_id.asset_category_id.id + + def get_invoice_line_account(self, type, product, fpos, company): + return product.asset_category_id.account_asset_id or super(AccountMoveLine, self).get_invoice_line_account(type, product, fpos, company)