Yaltik/docx_report_generation
2
0
Fork 0
Code Pull Requests Actions Packages Releases Activity

[IMP] add order_line getter, better code

Browse Source
This commit is contained in:
Stepan Savelyev 2020-01-27 15:33:03 +05:00
parent 06864fe013
commit f84391ac2e
1 changed files with 42 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
wizard/res_partner_contract_wizard.py
View File

@ -18,6 +18,21 @@ class ContractWizard(models.TransientModel):
def _default_document_template(self):
return self.env["res.partner.document.template"].search(self._get_template_domain(), limit=1)
def _get_template_domain(self):
template_type = {
"res.partner.contract": "contract",
"res.partner.contract.annex": "annex",
}.get(self.active_model, False)
company_type = (
self.partner_id.company_form if self.partner_id.is_company else "person"
)
document_template_domain = [
("template_type", "=", template_type),
("company_type", "=", company_type),
]
return document_template_domain
target = fields.Reference(
selection=[
("res.partner.contract", "Contract"),
@ -152,6 +167,7 @@ class ContractWizard(models.TransientModel):
return self.afterload(document_as_attachment)
def payload(self):
# Collect fields into a key-value structure
fields = {
transient_field.technical_name: transient_field.value
for transient_field in (
@ -159,6 +175,7 @@ class ContractWizard(models.TransientModel):
)
if transient_field.technical_name and transient_field.value
}
# Extend with special case
if self.target._name == "res.partner.contract.annex":
fields.update(
{
@ -166,6 +183,29 @@ class ContractWizard(models.TransientModel):
"specification_name": self.target.specification_name,
}
)
# Extend with order product lines
if hasattr(self.target, "order_id") and self.target.order_id.order_line:
def number_generator(n=1):
while (True):
yield n
n += 1
counter = number_generator()
fields.update(
{
"order_products": [
{
"number": next(counter),
"label": item.product_id.name,
"count": item.product_uom_qty,
"unit": item.product_uom.name,
"cost": item.price_unit,
"amount": item.price_subtotal,
} for item in self.target.order_id.order_line or []
]
}
)
return fields
def afterload(self, result):
@ -173,7 +213,7 @@ class ContractWizard(models.TransientModel):
if hasattr(self.target, "contract_id"):
res_id = self.target.contract_id.id
return self.env["mail.message"].create(
self.env["mail.message"].create(
{
"model": "res.partner.contract",
"res_id": res_id,
@ -182,20 +222,7 @@ class ContractWizard(models.TransientModel):
}
)
def _get_template_domain(self):
template_type = {
"res.partner.contract": "contract",
"res.partner.contract.annex": "annex",
}.get(self.active_model, False)
company_type = (
self.partner_id.company_form if self.partner_id.is_company else "person"
)
document_template_domain = [
("template_type", "=", template_type),
("company_type", "=", company_type),
]
return document_template_domain
return result
@property
def active_model(self):