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

[IMP] less code

Browse Source
This commit is contained in:
Stepan Savelyev 2020-01-27 13:56:09 +05:00
parent 2360093e24
commit 94c50928ec
1 changed files with 23 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
wizard/res_partner_contract_wizard.py
View File

@ -132,9 +132,27 @@ class ContractWizard(models.TransientModel):
@api.multi @api.multi
def get_docx_contract(self): def get_docx_contract(self):
template = self.document_template.attachment_id template = self.document_template.attachment_id
template_path = template._full_path(template.store_fname)
path_to_template = template._full_path(template.store_fname) payload = self.payload()
binary_data = get_document_from_values_stream(template_path, payload).read()
encoded_data = base64.b64encode(binary_data)
get_fn = self.target.get_filename_by_document_template
attachment_name = "{}.docx".format(get_fn(self.document_template or "Unknown"))
document_as_attachment = self.env["ir.attachment"].create(
{
"name": attachment_name,
"datas_fname": attachment_name,
"type": "binary",
"datas": encoded_data,
}
)
return self.afterload(document_as_attachment)
def payload(self):
fields = { fields = {
transient_field.technical_name: transient_field.value transient_field.technical_name: transient_field.value
for transient_field in ( for transient_field in (
@ -149,41 +167,22 @@ class ContractWizard(models.TransientModel):
"specification_name": self.target.specification_name, "specification_name": self.target.specification_name,
} }
) )
return fields
binary_data = get_document_from_values_stream(path_to_template, fields).read() def afterload(self, result):
encoded_data = base64.b64encode(binary_data)
attachment_name = (
self.target.get_filename_by_document_template(self.document_template)
or "Unknown"
)
attachment_name = "{}.docx".format(attachment_name)
document_as_attachment = self.env["ir.attachment"].create(
{
"name": attachment_name,
"datas_fname": attachment_name,
"type": "binary",
"datas": encoded_data,
}
)
# Send message with attachment to a mail.thread of the company
res_id = self.target.id res_id = self.target.id
if hasattr(self.target, "contract_id"): if hasattr(self.target, "contract_id"):
res_id = self.target.contract_id.id res_id = self.target.contract_id.id
self.env["mail.message"].create( return self.env["mail.message"].create(
{ {
"model": "res.partner.contract", "model": "res.partner.contract",
"res_id": res_id, "res_id": res_id,
"message_type": "comment", "message_type": "comment",
"attachment_ids": [(4, document_as_attachment.id, False)], "attachment_ids": [(4, result.id, False)],
} }
) )
return document_as_attachment
def _get_template_domain(self): def _get_template_domain(self):
template_type = { template_type = {
"res.partner.contract": "contract", "res.partner.contract": "contract",