46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from odoo import fields, models
|
|
|
|
|
|
class EstimateWizard(models.Model):
|
|
_name = "estimate.wizard"
|
|
|
|
template_id = fields.Many2one(comodel_name='hg.templates', string='Шаблон сметы', required=True)
|
|
code_id = fields.Many2one(comodel_name='hg.index.code', string='Код показателей', required=True)
|
|
|
|
def confirm_action(self):
|
|
self.ensure_one()
|
|
task = self.env.context.get('active_id') and self.env['project.task'].browse(self.env.context['active_id'])
|
|
if not task:
|
|
return
|
|
for line in self.template_id.line_ids:
|
|
node = self._get_node(res_model='project.task', res_id=task.id, name=task.name)
|
|
index = self.env['hg.index'].create({
|
|
'node_id': node.id,
|
|
'name': line.index_id.name,
|
|
'internal_code_id': self.code_id.id,
|
|
})
|
|
self.env['hg.value'].create({
|
|
'index_id': index.id,
|
|
'type': 'alone',
|
|
'date_due': line.date_due,
|
|
'value_float_plan': line.value_float_plan,
|
|
})
|
|
return {'type': 'ir.actions.act_window_close'}
|
|
|
|
|
|
def _get_node(self, res_model, res_id, name):
|
|
Node = self.env['hg.node']
|
|
|
|
node = Node.search([
|
|
('res_model', '=', res_model),
|
|
('res_id', '=', res_id)
|
|
], limit=1)
|
|
|
|
if not node:
|
|
node = Node.create({
|
|
'name': name,
|
|
'res_model': 'project.task',
|
|
'res_id': res_id,
|
|
})
|
|
return node
|