# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. from ast import literal_eval from odoo import api, fields, models, _, _lt from odoo.exceptions import ValidationError, UserError from odoo.osv import expression from odoo.osv.query import Query class Project(models.Model): _inherit = 'project.project' sale_line_id = fields.Many2one( 'sale.order.line', 'Sales Order Item', copy=False, compute="_compute_sale_line_id", store=True, readonly=False, index=True, domain="[('is_service', '=', True), ('is_expense', '=', False), ('state', 'in', ['sale', 'done']), ('order_partner_id', '=?', partner_id), '|', ('company_id', '=', False), ('company_id', '=', company_id)]", help="Sales order item to which the project is linked. Link the timesheet entry to the sales order item defined on the project. " "Only applies on tasks without sale order item defined, and if the employee is not in the 'Employee/Sales Order Item Mapping' of the project.") sale_order_id = fields.Many2one(string='Sales Order', related='sale_line_id.order_id', help="Sales order to which the project is linked.") has_any_so_to_invoice = fields.Boolean('Has SO to Invoice', compute='_compute_has_any_so_to_invoice') @api.model def _map_tasks_default_valeus(self, task, project): defaults = super()._map_tasks_default_valeus(task, project) defaults['sale_line_id'] = False return defaults @api.depends('partner_id') def _compute_sale_line_id(self): self.filtered( lambda p: p.sale_line_id and ( not p.partner_id or p.sale_line_id.order_partner_id.commercial_partner_id != p.partner_id.commercial_partner_id ) ).update({'sale_line_id': False}) @api.depends('sale_order_id.invoice_status', 'tasks.sale_order_id.invoice_status') def _compute_has_any_so_to_invoice(self): """Has any Sale Order whose invoice_status is set as To Invoice""" if not self.ids: self.has_any_so_to_invoice = False return self.env.cr.execute(""" SELECT id FROM project_project pp WHERE pp.active = true AND ( EXISTS(SELECT 1 FROM sale_order so JOIN project_task pt ON pt.sale_order_id = so.id WHERE pt.project_id = pp.id AND pt.active = true AND so.invoice_status = 'to invoice') OR EXISTS(SELECT 1 FROM sale_order so JOIN sale_order_line sol ON sol.order_id = so.id WHERE sol.id = pp.sale_line_id AND so.invoice_status = 'to invoice')) AND id in %s""", (tuple(self.ids),)) project_to_invoice = self.env['project.project'].browse([x[0] for x in self.env.cr.fetchall()]) project_to_invoice.has_any_so_to_invoice = True (self - project_to_invoice).has_any_so_to_invoice = False def action_view_so(self): self.ensure_one() action_window = { "type": "ir.actions.act_window", "res_model": "sale.order", "name": "Sales Order", "views": [[False, "form"]], "context": {"create": False, "show_sale": True}, "res_id": self.sale_order_id.id } return action_window def action_create_invoice(self): if not self.has_any_so_to_invoice: raise UserError(_("There is nothing to invoice in this project.")) action = self.env["ir.actions.actions"]._for_xml_id("sale.action_view_sale_advance_payment_inv") so_ids = (self.sale_order_id | self.task_ids.sale_order_id).filtered(lambda so: so.invoice_status == 'to invoice').ids action['context'] = { 'active_id': so_ids[0] if len(so_ids) == 1 else False, 'active_ids': so_ids } return action # ---------------------------- # Project Updates # ---------------------------- def _get_sale_order_stat_button(self): self.ensure_one() return { 'icon': 'dollar', 'text': _lt('Sales Order'), 'action_type': 'object', 'action': 'action_view_so', 'show': bool(self.sale_order_id), 'sequence': 1, } def _fetch_sale_order_items(self, domain_per_model=None, limit=None, offset=None): return self.env['sale.order.line'].browse(self._fetch_sale_order_item_ids(domain_per_model, limit, offset)) def _fetch_sale_order_item_ids(self, domain_per_model=None, limit=None, offset=None): if not self: return [] query = self._get_sale_order_items_query(domain_per_model) query.limit = limit query.offset = offset query_str, params = query.select('DISTINCT sale_line_id') self._cr.execute(query_str, params) return [row[0] for row in self._cr.fetchall()] def _get_sale_orders(self): return self._get_sale_order_items().order_id def _get_sale_order_items(self): return self._fetch_sale_order_items() def _get_sale_order_items_query(self, domain_per_model=None): if domain_per_model is None: domain_per_model = {} project_domain = [('id', 'in', self.ids), ('sale_line_id', '!=', False)] if 'project.project' in domain_per_model: project_domain = expression.AND([project_domain, domain_per_model['project.project']]) project_query = self.env['project.project']._where_calc(project_domain) self._apply_ir_rules(project_query, 'read') project_query_str, project_params = project_query.select('id', 'sale_line_id') Task = self.env['project.task'] task_domain = [('project_id', 'in', self.ids), ('sale_line_id', '!=', False)] if Task._name in domain_per_model: task_domain = expression.AND([task_domain, domain_per_model[Task._name]]) task_query = Task._where_calc(task_domain) Task._apply_ir_rules(task_query, 'read') task_query_str, task_params = task_query.select(f'{Task._table}.project_id AS id', f'{Task._table}.sale_line_id') query = Query(self._cr, 'project_sale_order_item', ' UNION '.join([project_query_str, task_query_str])) query._where_params = project_params + task_params return query def _get_stat_buttons(self): buttons = super(Project, self)._get_stat_buttons() if self.user_has_groups('sales_team.group_sale_salesman_all_leads'): buttons.append(self._get_sale_order_stat_button()) return buttons class ProjectTask(models.Model): _inherit = "project.task" sale_order_id = fields.Many2one('sale.order', 'Sales Order', compute='_compute_sale_order_id', store=True, help="Sales order to which the task is linked.") sale_line_id = fields.Many2one( 'sale.order.line', 'Sales Order Item', domain="[('company_id', '=', company_id), ('is_service', '=', True), ('order_partner_id', 'child_of', commercial_partner_id), ('is_expense', '=', False), ('state', 'in', ['sale', 'done'])]", compute='_compute_sale_line', recursive=True, store=True, readonly=False, copy=False, tracking=True, index=True, help="Sales Order Item to which the time spent on this task will be added, in order to be invoiced to your customer.") project_sale_order_id = fields.Many2one('sale.order', string="Project's sale order", related='project_id.sale_order_id') invoice_count = fields.Integer("Number of invoices", related='sale_order_id.invoice_count') task_to_invoice = fields.Boolean("To invoice", compute='_compute_task_to_invoice', search='_search_task_to_invoice', groups='sales_team.group_sale_salesman_all_leads') @property def SELF_READABLE_FIELDS(self): return super().SELF_READABLE_FIELDS | {'sale_order_id', 'sale_line_id'} @api.depends('sale_line_id', 'project_id', 'commercial_partner_id') def _compute_sale_order_id(self): for task in self: sale_order_id = task.sale_order_id or self.env["sale.order"] if task.sale_line_id: sale_order_id = task.sale_line_id.sudo().order_id elif task.project_id.sale_order_id: sale_order_id = task.project_id.sale_order_id if task.commercial_partner_id != sale_order_id.partner_id.commercial_partner_id: sale_order_id = False if sale_order_id and not task.partner_id: task.partner_id = sale_order_id.partner_id task.sale_order_id = sale_order_id @api.depends('commercial_partner_id', 'sale_line_id.order_partner_id.commercial_partner_id', 'parent_id.sale_line_id', 'project_id.sale_line_id') def _compute_sale_line(self): for task in self: if not task.sale_line_id: # if the display_project_id is set then it means the task is classic task or a subtask with another project than its parent. task.sale_line_id = task.display_project_id.sale_line_id or task.parent_id.sale_line_id or task.project_id.sale_line_id # check sale_line_id and customer are coherent if task.sale_line_id.order_partner_id.commercial_partner_id != task.partner_id.commercial_partner_id: task.sale_line_id = False @api.constrains('sale_line_id') def _check_sale_line_type(self): for task in self.sudo(): if task.sale_line_id: if not task.sale_line_id.is_service or task.sale_line_id.is_expense: raise ValidationError(_( 'You cannot link the order item %(order_id)s - %(product_id)s to this task because it is a re-invoiced expense.', order_id=task.sale_line_id.order_id.name, product_id=task.sale_line_id.product_id.display_name, )) @api.ondelete(at_uninstall=False) def _unlink_except_linked_so(self): if any(task.sale_line_id for task in self): raise ValidationError(_('You have to unlink the task from the sale order item in order to delete it.')) # --------------------------------------------------- # Actions # --------------------------------------------------- def _get_action_view_so_ids(self): return self.sale_order_id.ids def action_view_so(self): self.ensure_one() so_ids = self._get_action_view_so_ids() action_window = { "type": "ir.actions.act_window", "res_model": "sale.order", "name": "Sales Order", "views": [[False, "tree"], [False, "form"]], "context": {"create": False, "show_sale": True}, "domain": [["id", "in", so_ids]], } if len(so_ids) == 1: action_window["views"] = [[False, "form"]] action_window["res_id"] = so_ids[0] return action_window def rating_get_partner_id(self): partner = self.partner_id or self.sale_line_id.order_id.partner_id if partner: return partner return super().rating_get_partner_id() @api.depends('sale_order_id.invoice_status', 'sale_order_id.order_line') def _compute_task_to_invoice(self): for task in self: if task.sale_order_id: task.task_to_invoice = bool(task.sale_order_id.invoice_status not in ('no', 'invoiced')) else: task.task_to_invoice = False @api.model def _search_task_to_invoice(self, operator, value): query = """ SELECT so.id FROM sale_order so WHERE so.invoice_status != 'invoiced' AND so.invoice_status != 'no' """ operator_new = 'inselect' if(bool(operator == '=') ^ bool(value)): operator_new = 'not inselect' return [('sale_order_id', operator_new, (query, ()))] class ProjectTaskRecurrence(models.Model): _inherit = 'project.task.recurrence' def _new_task_values(self, task): values = super(ProjectTaskRecurrence, self)._new_task_values(task) task = self.sudo().task_ids[0] values['sale_line_id'] = self._get_sale_line_id(task) return values def _get_sale_line_id(self, task): return task.sale_line_id.id