2
0
account-financial-tools/account_spread_cost_revenue/models/account_spread_template.py

201 lines
7.1 KiB
Python
Raw Normal View History

# Copyright 2018-2020 Onestein (<https://www.onestein.eu>)
2018-10-22 12:27:50 +02:00
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
2020-09-07 14:17:04 +02:00
from odoo import _, api, fields, models
from odoo.exceptions import UserError
2018-10-22 12:27:50 +02:00
class AccountSpreadTemplate(models.Model):
2020-09-07 14:17:04 +02:00
_name = "account.spread.template"
_description = "Account Spread Template"
2018-10-22 12:27:50 +02:00
name = fields.Char(required=True)
2020-09-07 14:17:04 +02:00
spread_type = fields.Selection(
[("sale", "Customer"), ("purchase", "Supplier")], default="sale", required=True
)
2018-10-22 12:27:50 +02:00
company_id = fields.Many2one(
"res.company", default=lambda self: self.env.company, required=True
2020-09-07 14:17:04 +02:00
)
2018-10-22 12:27:50 +02:00
spread_journal_id = fields.Many2one(
"account.journal",
string="Journal",
compute="_compute_spread_journal",
readonly=False,
store=True,
required=True,
2020-09-07 14:17:04 +02:00
)
use_invoice_line_account = fields.Boolean(
string="Invoice account as spread account",
help="Use invoice line's account as Balance sheet / spread account.\n"
2020-09-07 14:17:04 +02:00
"In this case, user need to select expense/revenue account too.",
)
2018-10-22 12:27:50 +02:00
spread_account_id = fields.Many2one(
"account.account",
string="Spread Balance Sheet Account",
compute="_compute_spread_account",
readonly=False,
store=True,
required=False,
2020-09-07 14:17:04 +02:00
)
exp_rev_account_id = fields.Many2one(
2020-09-07 14:17:04 +02:00
"account.account",
string="Expense/Revenue Account",
help="Optional account to overwrite the existing expense/revenue account",
2020-09-07 14:17:04 +02:00
)
2018-12-20 16:39:00 +01:00
period_number = fields.Integer(
2020-09-07 14:17:04 +02:00
string="Number of Repetitions", help="Define the number of spread lines"
)
period_type = fields.Selection(
[("month", "Month"), ("quarter", "Quarter"), ("year", "Year")],
help="Period length for the entries",
)
2018-12-20 16:39:00 +01:00
start_date = fields.Date()
auto_spread = fields.Boolean(
2020-09-07 14:17:04 +02:00
string="Auto assign template on invoice validate",
help="If checked, provide option to auto create spread during "
2020-09-07 14:17:04 +02:00
"invoice validation, based on product/account/analytic in invoice line.",
)
auto_spread_ids = fields.One2many(
2020-09-07 14:17:04 +02:00
comodel_name="account.spread.template.auto",
string="Auto Spread On",
inverse_name="template_id",
)
2018-10-22 12:27:50 +02:00
@api.model
def default_get(self, fields):
res = super().default_get(fields)
if not res.get("company_id"):
res["company_id"] = self.env.company.id
if "spread_journal_id" not in res:
default_journal = self.env["account.spread"].default_journal(
res["company_id"]
)
if default_journal:
res["spread_journal_id"] = default_journal.id
2018-10-22 12:27:50 +02:00
return res
2020-09-07 14:17:04 +02:00
@api.constrains("auto_spread", "auto_spread_ids")
def _check_product_account(self):
2020-09-07 14:17:04 +02:00
for rec in self.filtered("auto_spread"):
for line in rec.auto_spread_ids:
if not line.product_id and not line.account_id:
2020-09-07 14:17:04 +02:00
raise UserError(
_(
"Please select product and/or account "
"on auto spread options"
)
)
@api.depends("spread_type", "company_id")
def _compute_spread_journal(self):
for spread in self:
company = spread.company_id
if spread.spread_type == "sale":
journal = company.default_spread_revenue_journal_id
else:
journal = company.default_spread_expense_journal_id
if journal:
spread.spread_journal_id = journal
@api.depends("spread_type", "company_id")
def _compute_spread_account(self):
for spread in self:
company = spread.company_id
if spread.spread_type == "sale":
account = company.default_spread_revenue_account_id
else:
account = company.default_spread_expense_account_id
if account:
spread.spread_account_id = account
2018-10-22 12:27:50 +02:00
2020-09-07 14:17:04 +02:00
@api.onchange("use_invoice_line_account")
def _onchange_user_invoice_line_account(self):
self.exp_rev_account_id = False
def _prepare_spread_from_template(self, spread_account_id=False):
2018-10-22 12:27:50 +02:00
self.ensure_one()
company = self.company_id
spread_vals = {
2020-09-07 14:17:04 +02:00
"name": self.name,
"template_id": self.id,
"journal_id": self.spread_journal_id.id,
"use_invoice_line_account": self.use_invoice_line_account,
"company_id": company.id,
2018-10-22 12:27:50 +02:00
}
account_id = spread_account_id or self.spread_account_id.id
2020-09-07 14:17:04 +02:00
if self.spread_type == "sale":
invoice_type = "out_invoice"
spread_vals["debit_account_id"] = account_id
2018-10-22 12:27:50 +02:00
else:
2020-09-07 14:17:04 +02:00
invoice_type = "in_invoice"
spread_vals["credit_account_id"] = account_id
2018-10-22 12:27:50 +02:00
2018-12-20 16:39:00 +01:00
if self.period_number:
2020-09-07 14:17:04 +02:00
spread_vals["period_number"] = self.period_number
2018-12-20 16:39:00 +01:00
if self.period_type:
2020-09-07 14:17:04 +02:00
spread_vals["period_type"] = self.period_type
2018-12-20 16:39:00 +01:00
if self.start_date:
2020-09-07 14:17:04 +02:00
spread_vals["spread_date"] = self.start_date
2018-12-20 16:39:00 +01:00
2020-09-07 14:17:04 +02:00
spread_vals["invoice_type"] = invoice_type
2018-10-22 12:27:50 +02:00
return spread_vals
2020-09-07 14:17:04 +02:00
@api.constrains("auto_spread_ids", "auto_spread")
def _check_auto_spread_ids_unique(self):
query = """
select product_id, account_id, analytic_account_id
from (
select product_id, account_id, analytic_account_id, count(*)
from account_spread_template_auto a
join account_spread_template b on a.template_id = b.id
where b.auto_spread = true and b.id in %s
group by product_id, account_id, analytic_account_id
) x where x.count > 1 """
self._cr.execute(query, [self._ids])
results = []
for res in self._cr.fetchall():
2020-09-07 14:17:04 +02:00
product = self.env["product.product"].browse(res[0])
account = self.env["account.account"].browse(res[1])
analytic = self.env["account.analytic.account"].browse(res[2])
results.append(
"{} / {} / {}".format(product.name, account.name, analytic.name)
)
if results:
raise UserError(
2020-09-07 14:17:04 +02:00
_("Followings are duplicated combinations,\n\n%s" % "\n".join(results))
)
class AccountSpreadTemplateAuto(models.Model):
2020-09-07 14:17:04 +02:00
_name = "account.spread.template.auto"
_description = "Auto create spread, based on product/account/analytic"
template_id = fields.Many2one(
2020-09-07 14:17:04 +02:00
comodel_name="account.spread.template",
string="Spread Template",
required=True,
2020-09-07 14:17:04 +02:00
ondelete="cascade",
index=True,
)
company_id = fields.Many2one(
related="template_id.company_id",
store=True,
)
name = fields.Char(
required=True,
default="/",
)
product_id = fields.Many2one(
comodel_name="product.product",
string="Product",
)
account_id = fields.Many2one(
comodel_name="account.account",
string="Account",
)
analytic_account_id = fields.Many2one(
comodel_name="account.analytic.account",
string="Analytic",
)