[STYLE] fix

This commit is contained in:
Stepan Savelyev 2019-12-04 16:54:42 +05:00
parent 50345c6c56
commit e88e4c757f

View File

@ -1,12 +1,8 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import base64 import base64
import io
import math import math
from datetime import datetime from datetime import datetime
from docxtpl import DocxTemplate
from odoo import api, fields, models from odoo import api, fields, models
from odoo.tools.config import config from odoo.tools.config import config
from pytils import numeral from pytils import numeral
@ -16,7 +12,6 @@ from ..utils.docxtpl import get_document_from_values_stream
class ContractWizard(models.TransientModel): class ContractWizard(models.TransientModel):
_name = 'res.partner.contract.wizard' _name = 'res.partner.contract.wizard'
# TODO: Move _get_default_+ method down
def _get_default_template(self): def _get_default_template(self):
_template = self.env['res.partner.contract.template'].search([ _template = self.env['res.partner.contract.template'].search([
('is_default', '=', True) ('is_default', '=', True)
@ -507,9 +502,7 @@ class ContractWizard(models.TransientModel):
} }
return context return context
def get_docx_contract(self): def get_docx_contract(self):
path_to_template = "{}/filestore/{}/{}".format( path_to_template = "{}/filestore/{}/{}".format(
config.get("data_dir"), config.get("data_dir"),
config.get("db_name"), config.get("db_name"),
@ -517,13 +510,14 @@ class ContractWizard(models.TransientModel):
) )
fields = self._generate_context() fields = self._generate_context()
binary_data = get_document_from_values_stream(path_to_template, fields).read() binary_data = get_document_from_values_stream(
path_to_template, fields).read()
encoded_data = base64.b64encode(binary_data) encoded_data = base64.b64encode(binary_data)
attachment = self.env['ir.attachment'].create({ attachment = self.env['ir.attachment'].create({
"name": "Contract-{}.doc".format(self.contract_id.name), "name": "Contract-{}.doc".format(self.contract_id.name),
"type": "binary", "type": "binary",
"datas": binary_data, "datas": encoded_data,
}) })
return attachment return attachment