214 lines
8 KiB
Python
214 lines
8 KiB
Python
from odoo import fields, models, api, Command, _
|
|
from odoo.exceptions import ValidationError, UserError
|
|
from odoo.osv import expression
|
|
from collections import defaultdict, namedtuple
|
|
from odoo.addons.project.models.project_task import CLOSED_STATES
|
|
import re
|
|
|
|
|
|
class Task(models.Model):
|
|
_inherit = "project.task"
|
|
|
|
equipment_ids = fields.Many2many(
|
|
comodel_name="bemade_fsm.equipment",
|
|
relation="bemade_fsm_task_equipment_rel",
|
|
column1="task_id",
|
|
column2="equipment_id",
|
|
string="Equipment to Service",
|
|
tracking=True,
|
|
)
|
|
|
|
work_order_contacts = fields.Many2many(
|
|
comodel_name="res.partner",
|
|
relation="task_work_order_contact_rel",
|
|
column1="task_id",
|
|
column2="res_partner_id",
|
|
compute="_compute_contacts",
|
|
inverse="_inverse_contacts",
|
|
store=True
|
|
)
|
|
|
|
site_contacts = fields.Many2many(
|
|
comodel_name="res.partner",
|
|
relation="task_site_contact_rel",
|
|
column1="task_id",
|
|
column2="res_partner_id",
|
|
compute="_compute_contacts",
|
|
inverse="_inverse_contacts",
|
|
store=True
|
|
)
|
|
|
|
# Override related field to make it return false if this is an FSM subtask
|
|
allow_billable = fields.Boolean(
|
|
string="Can be billed",
|
|
related=False,
|
|
compute="_compute_allow_billable",
|
|
store=True
|
|
)
|
|
|
|
visit_id = fields.Many2one(
|
|
comodel_name='bemade_fsm.visit',
|
|
string="Visit",
|
|
)
|
|
|
|
relevant_order_lines = fields.Many2many(
|
|
comodel_name='sale.order.line',
|
|
store=False,
|
|
compute='_compute_relevant_order_lines',
|
|
)
|
|
|
|
work_order_number = fields.Char(readonly=True)
|
|
|
|
propagate_assignment = fields.Boolean(
|
|
string='Propagate Assignment',
|
|
help='Propagate assignment of this task to all subtasks.',
|
|
default=False,
|
|
)
|
|
|
|
is_closed = fields.Boolean(
|
|
compute="_compute_is_closed",
|
|
)
|
|
|
|
def _compute_is_closed(self):
|
|
for rec in self:
|
|
rec.is_closed = rec.state in CLOSED_STATES
|
|
|
|
@api.model_create_multi
|
|
def create(self, vals):
|
|
res = super().create(vals)
|
|
for rec in res:
|
|
if rec.sale_order_id:
|
|
seq = 1
|
|
prev_seqs = self.sale_order_id.tasks_ids and \
|
|
self.sale_order_id.tasks_ids.mapped('work_order_number')
|
|
if prev_seqs:
|
|
pattern = re.compile(r"(\d+)$")
|
|
matches = map(lambda n: pattern.search(n), prev_seqs)
|
|
seq += max(map(lambda n: int(n.group(1)) if n else 0, matches))
|
|
rec.work_order_number = rec.sale_order_id.name.replace('SO', 'SVR', 1) \
|
|
+ f"-{seq}"
|
|
return res
|
|
|
|
def write(self, vals):
|
|
res = super().write(vals)
|
|
if not self: # End recursion on empty RecordSet
|
|
return res
|
|
if 'propagate_assignment' in vals:
|
|
# When a user sets propagate assignment, it should propagate that setting all the way down the chain
|
|
self.child_ids.write({'propagate_assignment': vals['propagate_assignment']})
|
|
if 'user_ids' in vals:
|
|
to_propagate = self.filtered(lambda task: task.propagate_assignment)
|
|
# Here we use child_ids instead of _get_all_subtasks() so as to allow for setting propagate_assignment
|
|
# to false on a child task.
|
|
to_propagate.child_ids.write({'user_ids': vals['user_ids']})
|
|
return res
|
|
|
|
@api.depends('sale_order_id')
|
|
def _compute_relevant_order_lines(self):
|
|
for rec in self:
|
|
rec.relevant_order_lines = (
|
|
rec.sale_order_id and rec.sale_order_id.get_relevant_order_lines(
|
|
rec) or False)
|
|
|
|
def _get_closed_stage_by_project(self):
|
|
""" Gets the stage representing completed tasks for each project in
|
|
self.project_id. Copied from industry_fsm/.../project.py:217-221
|
|
for consistency.
|
|
|
|
:returns: Dict of project.project -> project.task.type"""
|
|
return {
|
|
project:
|
|
project.type_ids.filtered(lambda stage: stage.is_closed)[:1]
|
|
or project.type_ids[-1:]
|
|
for project in self.project_id
|
|
}
|
|
|
|
@api.depends('sale_line_id.order_id.site_contacts',
|
|
'sale_line_id.order_id.work_order_contacts')
|
|
def _compute_contacts(self):
|
|
""" The work order contacts and site contacts for a given task are taken from the sale order if the task
|
|
is related to one, and from the task's customer if there is no sale order related to the task."""
|
|
for rec in self:
|
|
site_contacts = self.sale_line_id and self.sale_line_id.order_id.site_contacts or \
|
|
self.partner_id.site_contacts
|
|
work_order_contacts = self.sale_line_id and self.sale_line_id.order_id.work_order_contacts or \
|
|
self.partner_id.work_order_contacts
|
|
rec.write({
|
|
'site_contacts': [Command.set(site_contacts.ids)],
|
|
'work_order_contacts': [Command.set(work_order_contacts.ids)]
|
|
})
|
|
|
|
def _inverse_contacts(self):
|
|
""" If the task is linked to a sales order, the sales order should have its contacts updated to match."""
|
|
for rec in self:
|
|
if rec.sale_line_id:
|
|
rec.sale_line_id.order_id.write({
|
|
'work_order_contacts': [Command.set(rec.work_order_contacts.ids)],
|
|
'site_contacts': [Command.set(rec.site_contacts.ids)],
|
|
})
|
|
|
|
@api.depends('parent_id.visit_id', 'project_id.is_fsm', 'project_id.allow_billable')
|
|
def _compute_allow_billable(self):
|
|
for rec in self:
|
|
# If an FSM task has a parent that is linked to an SO line, then the parent is the billable one
|
|
if rec.parent_id and not rec.parent_id.visit_id and rec.project_id and rec.project_id.is_fsm:
|
|
rec.allow_billable = False
|
|
else:
|
|
rec.allow_billable = rec.project_id.allow_billable
|
|
|
|
def _fsm_create_sale_order_line(self):
|
|
# Override to not generate new lines for tasks that are just linked to a visit item
|
|
self.ensure_one()
|
|
if self.visit_id:
|
|
return
|
|
else:
|
|
super()._fsm_create_sale_order_line()
|
|
|
|
def action_fsm_validate(self, stop_running_timers=False):
|
|
# Override to close out subtasks as well
|
|
all_tasks = self | self._get_all_subtasks()
|
|
return super(Task, all_tasks).action_fsm_validate(stop_running_timers)
|
|
|
|
def _get_full_hierarchy(self):
|
|
if self.child_ids:
|
|
return self | self.child_ids._get_full_hierarchy()
|
|
return self
|
|
|
|
def synchronize_name_fsm(self):
|
|
""" Applies naming to the entire task tree for tasks that are part of this
|
|
recordset. Root tasks are named:
|
|
|
|
Partner Shipping Name - Sale Line Name (Template Name)
|
|
|
|
Child tasks with sale_line_id are named by their template if set, sale line name
|
|
if not.
|
|
|
|
Child tasks not linked to sale lines are left with their original names."""
|
|
|
|
all_tasks = self | self._get_all_subtasks()
|
|
for rec in all_tasks:
|
|
assert rec.is_fsm, "This method should only be called on FSM tasks."
|
|
|
|
template = rec.sale_line_id and rec.sale_line_id.product_id.task_template_id
|
|
|
|
if template:
|
|
title = template.name
|
|
elif rec.sale_line_id:
|
|
name_parts = rec.sale_line_id.name.split('\n')
|
|
title = name_parts and name_parts[0] or rec.sale_line_id.product_id.name
|
|
elif rec.visit_id:
|
|
title = rec.visit_id.label
|
|
else:
|
|
rec.name = rec.name
|
|
return
|
|
|
|
client_name = rec.sale_order_id.partner_shipping_id.name
|
|
|
|
if not rec.parent_id:
|
|
rec.name = f"{rec.work_order_number} - {client_name} - " f"{title}"
|
|
else:
|
|
rec.name = title
|
|
|
|
@property
|
|
def root_ancestor(self):
|
|
return self.parent_id and self.parent_id.root_ancestor or self
|