86 lines
2.2 KiB
Python
86 lines
2.2 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from odoo import api, fields, models
|
||
|
|
|
||
|
|
|
||
|
|
class PartnerContract(models.Model):
|
||
|
|
|
||
|
|
_name = 'res.partner.contract'
|
||
|
|
|
||
|
|
name = fields.Char(
|
||
|
|
string='Contract number',
|
||
|
|
help='Number of contract, letters and digits',
|
||
|
|
)
|
||
|
|
date = fields.Date(
|
||
|
|
string='Date of conclusion',
|
||
|
|
help='Date, when contract was concluded',
|
||
|
|
default=datetime.now().date(),
|
||
|
|
required=True
|
||
|
|
)
|
||
|
|
order_ids = fields.One2many(
|
||
|
|
'sale.order',
|
||
|
|
'contract_id',
|
||
|
|
string='Annexes',
|
||
|
|
help='Annexes to this contract'
|
||
|
|
)
|
||
|
|
partner_id = fields.Many2one(
|
||
|
|
'res.partner',
|
||
|
|
string='Contract Partner',
|
||
|
|
help='Contract partner',
|
||
|
|
default=lambda self: self.env.context['active_id'],
|
||
|
|
required=True
|
||
|
|
)
|
||
|
|
|
||
|
|
@api.onchange('date')
|
||
|
|
def _change_contract_name(self):
|
||
|
|
"""
|
||
|
|
Procedure for forming contract name
|
||
|
|
:return: contract name in format "DDMM-YY-№"
|
||
|
|
"""
|
||
|
|
self.name = self._calculate_contract_name(self.date)
|
||
|
|
|
||
|
|
@api.model
|
||
|
|
def create(self, vals):
|
||
|
|
|
||
|
|
datetime_now = datetime.now().strftime("%Y-%m-%d")
|
||
|
|
vals['name'] = self._calculate_contract_name(datetime_now)
|
||
|
|
|
||
|
|
return super(PartnerContract, self).create(vals)
|
||
|
|
|
||
|
|
def _calculate_contract_name(self, _date):
|
||
|
|
|
||
|
|
contract_date = datetime.strptime(_date, '%Y-%m-%d')
|
||
|
|
date_part = contract_date.strftime('%d%m-%y')
|
||
|
|
|
||
|
|
today_contracts = self.search([
|
||
|
|
('date', '=', contract_date.date()),
|
||
|
|
])
|
||
|
|
if len(today_contracts) > 0:
|
||
|
|
last_contract_number = int(
|
||
|
|
today_contracts[-1].name.split('-')[2]) + 1
|
||
|
|
else:
|
||
|
|
last_contract_number = 1
|
||
|
|
|
||
|
|
return '{}-{}'.format(date_part, last_contract_number)
|
||
|
|
|
||
|
|
|
||
|
|
class AnnexType(models.Model):
|
||
|
|
_name = 'res.partner.contract.annex.type'
|
||
|
|
|
||
|
|
name = fields.Char(
|
||
|
|
string='Annex template name'
|
||
|
|
)
|
||
|
|
description = fields.Text(
|
||
|
|
string='Annex template description'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class ContractTemplate(models.Model):
|
||
|
|
_name = 'res.partner.contract.template'
|
||
|
|
_inherit = 'ir.attachment'
|
||
|
|
|
||
|
|
is_contract_template = fields.Boolean(
|
||
|
|
srting='Is this document contract template?',
|
||
|
|
default=True
|
||
|
|
) # TODO: srting -> string
|