[ADD] repair_picking

This commit is contained in:
Joan Sisquella
2023-05-04 11:07:23 +02:00
committed by Lois Rilo
parent 1ee992a8e4
commit cf662b87cb
17 changed files with 942 additions and 0 deletions

View File

@@ -0,0 +1,3 @@
from . import stock_warehouse
from . import repair
from . import stock_rule

View File

@@ -0,0 +1,231 @@
# Copyright 2023 ForgeFlow S.L. (https://www.forgeflow.com)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo import api, fields, models
from odoo.exceptions import UserError
class RepairOrder(models.Model):
_inherit = "repair.order"
@api.model
def _get_default_location_id(self):
warehouse = self.env["stock.warehouse"].search(
[("company_id", "=", self.env.company.id)], limit=1
)
return (
warehouse.repair_location_id.id
if warehouse and warehouse.repair_location_id
else False
)
# Changing default value on existing field
location_id = fields.Many2one(
default=_get_default_location_id,
)
picking_ids = fields.Many2many(
comodel_name="stock.picking",
compute="_compute_picking_ids",
copy=False,
string="Pickings associated to this repair order",
)
picking_count = fields.Integer(
string="Transfers", copy=False, compute="_compute_picking_ids"
)
procurement_group_id = fields.Many2one(
"procurement.group", "Procurement Group", copy=False
)
def action_view_pickings(self):
self.ensure_one()
action = self.env["ir.actions.actions"]._for_xml_id(
"stock.action_picking_tree_all"
)
action["domain"] = [("id", "in", self.picking_ids.ids)]
return action
def _compute_picking_ids(self):
for order in self:
moves = self.env["stock.move"].search(
[("repair_line_id", "in", order.operations.ids)]
)
order.picking_ids = moves.mapped("picking_id")
order.picking_count = len(moves.mapped("picking_id"))
def action_repair_cancel(self):
res = super().action_repair_cancel()
for picking in self.picking_ids:
if picking.state not in ["cancel", "done"]:
picking.action_cancel()
return res
def _action_launch_stock_rule(self, repair_lines):
for line in repair_lines:
self._run_procurement_repair(line)
return True
def _run_procurement_repair(self, line):
procurements = []
errors = []
procurement = self._prepare_procurement_repair(line)
procurements.append(procurement)
try:
self.env["procurement.group"].run(procurements)
except UserError as error:
errors.append(error.args[0])
if errors:
raise UserError("\n".join(errors))
return True
@api.model
def _get_procurement_data_repair(self, line):
warehouse = self.location_id.get_warehouse()
if not self.procurement_group_id:
group_id = self.env["procurement.group"].create({"name": self.name})
self.procurement_group_id = group_id
procurement_data = {
"name": self.name,
"group_id": self.procurement_group_id,
"origin": self.name,
"date_planned": fields.Datetime.now(),
"product_id": line.product_id.id,
"product_qty": line.product_uom_qty,
"product_uom": line.product_uom.id,
"company_id": self.company_id,
"warehouse_id": warehouse,
"repair_line_id": line.id,
}
if line.lot_id:
procurement_data["lot_id"] = line.lot_id.id
if line.type == "remove":
procurement_data[
"source_repair_location_id"
] = line.repair_id.location_id.id
return procurement_data
@api.model
def _prepare_procurement_repair(self, line):
values = self._get_procurement_data_repair(line)
warehouse = self.location_id.get_warehouse()
location = (
self.location_id
if line.type == "add"
else warehouse.remove_c_type_id.default_location_dest_id
)
procurement = self.env["procurement.group"].Procurement(
line.product_id,
line.product_uom_qty,
line.product_uom,
location,
values.get("origin"),
values.get("origin"),
self.company_id,
values,
)
return procurement
def _update_stock_moves_and_picking_state(self):
for repair in self:
for picking in repair.picking_ids:
if picking.location_dest_id.id == self.location_id.id:
for move_line in picking.move_ids_without_package:
stock_moves = repair.stock_move_ids.filtered(
lambda m: m.product_id.id
== repair.operations.filtered(
lambda l: l.type == "add"
and l.product_id.id == m.product_id.id
).product_id.id
and m.location_id.id == self.location_id.id
)
if stock_moves:
stock_moves[0].write(
{
"move_orig_ids": [(4, move_line.id)],
"state": "waiting",
}
)
if picking.location_id.id == self.location_id.id:
for move_line in picking.move_ids_without_package:
stock_moves = repair.stock_move_ids.filtered(
lambda m: m.product_id.id
== repair.operations.filtered(
lambda l: l.type == "remove"
and l.product_id.id == m.product_id.id
).product_id.id
and m.location_dest_id.id == self.location_id.id
)
if stock_moves:
move_line.write(
{
"move_orig_ids": [(4, stock_moves[0].id)],
"state": "waiting",
}
)
# We are using write here because
# the repair_stock_move module does not use stock rules.
# As a result, we manually link the stock moves
# and then recompute the state of the picking.
picking._compute_state()
def action_repair_confirm(self):
res = super().action_repair_confirm()
for repair in self:
warehouse = repair.location_id.get_warehouse()
if warehouse.repair_steps in ["2_steps", "3_steps"]:
repair._action_launch_stock_rule(
repair.operations.filtered(lambda l: l.type == "add"),
)
if warehouse.repair_steps == "3_steps":
repair._action_launch_stock_rule(
repair.operations.filtered(lambda l: l.type == "remove"),
)
repair._update_stock_moves_and_picking_state()
return res
@api.onchange("location_id")
def _onchange_location_id(self):
warehouse = self.location_id.get_warehouse()
for line in self.operations:
if line.type == "add":
line.location_id = self.location_id
elif line.type == "remove" and warehouse.repair_steps == "3_steps":
line.location_dest_id = self.location_id
class RepairLine(models.Model):
_inherit = "repair.line"
@api.onchange("type", "product_id")
def onchange_operation_type(self):
super().onchange_operation_type()
production_location = self.env["stock.location"].search(
[("usage", "=", "production")], limit=1
)
warehouse = self.repair_id.location_id.get_warehouse()
if self.type == "add":
self.write(
{
"location_id": self.repair_id.location_id.id,
"location_dest_id": production_location.id,
}
)
elif self.type == "remove":
self.write({"location_id": production_location.id})
if warehouse.repair_steps in ["1_step", "2_steps"]:
scrap_location = self.env["stock.location"].search(
[
("scrap_location", "=", True),
("company_id", "=", warehouse.company_id.id),
],
limit=1,
)
self.write({"location_dest_id": scrap_location.id})
else:
self.write({"location_dest_id": self.repair_id.location_id.id})
def create(self, vals):
line = super().create(vals)
if line.repair_id.state in ["confirmed", "under_repair", "ready"]:
line.repair_id._action_launch_stock_rule(line)
return line

View File

@@ -0,0 +1,27 @@
# Copyright 2023 ForgeFlow S.L. (https://www.forgeflow.com)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo import api, models
class StockRule(models.Model):
_inherit = "stock.rule"
def _get_custom_move_fields(self):
fields = super(StockRule, self)._get_custom_move_fields()
# Fields is added on `repair_stock_move` module.
fields += ["repair_line_id"]
return fields
class ProcurementGroup(models.Model):
_inherit = "procurement.group"
@api.model
def _get_rule_domain(self, location, values):
domain = super(ProcurementGroup, self)._get_rule_domain(location, values)
if values.get("source_repair_location_id"):
domain.append(
("location_src_id", "=", values.get("source_repair_location_id"))
)
return domain

View File

@@ -0,0 +1,184 @@
# Copyright 2023 ForgeFlow S.L. (https://www.forgeflow.com)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo import fields, models
class StockWarehouse(models.Model):
_inherit = "stock.warehouse"
repair_steps = fields.Selection(
[
("1_step", "Repair"),
("2_steps", "Pick component, repair"),
("3_steps", "Pick component, repair, store removed component"),
],
string="Repair Steps",
default="1_step",
)
add_c_type_id = fields.Many2one(
"stock.picking.type", string="Add Component to Repair"
)
remove_c_type_id = fields.Many2one(
"stock.picking.type", string="Remove component from Repair"
)
repair_route_id = fields.Many2one("stock.location.route", string="Repair Route")
repair_location_id = fields.Many2one("stock.location", string="Repair Location")
def update_picking_types(self, repair_steps, repair_location_id):
if repair_steps in ["2_steps", "3_steps"]:
self.add_c_type_id.active = True
if repair_steps == "1_step":
self.add_c_type_id.active = False
if repair_steps == "3_steps":
self.remove_c_type_id.active = True
if repair_steps in ["1_step", "2_steps"]:
self.remove_c_type_id.active = False
if repair_location_id:
self.add_c_type_id.write({"default_location_dest_id": repair_location_id})
self.remove_c_type_id.write({"default_location_src_id": repair_location_id})
def update_repair_routes(self, repair_steps, repair_location_id):
if repair_steps == "2_steps" or repair_steps == "3_steps":
self.repair_route_id.active = True
existing_rule = (
self.env["stock.rule"]
.with_context(active_test=False)
.search(
[
("picking_type_id", "=", self.add_c_type_id.id),
("route_id", "=", self.repair_route_id.id),
],
limit=1,
)
)
existing_rule.active = True
if repair_steps == "1_step":
for rule in self.repair_route_id.rule_ids:
rule.active = False
self.repair_route_id.active = False
if repair_location_id:
self.repair_route_id.rule_ids.filtered(
lambda r: r.picking_type_id == self.add_c_type_id
).write({"location_id": repair_location_id})
self.repair_route_id.rule_ids.filtered(
lambda r: r.picking_type_id == self.remove_c_type_id
).write({"location_src_id": repair_location_id})
if repair_steps in ["1_step", "2_steps"]:
self.repair_route_id.rule_ids.filtered(
lambda r: r.picking_type_id == self.remove_c_type_id
).active = False
def write(self, vals):
res = super(StockWarehouse, self).write(vals)
for warehouse in self:
repair_steps = vals.get("repair_steps")
repair_location_id = vals.get("repair_location_id")
if repair_steps:
if repair_steps in ["3_steps", "2_steps"]:
warehouse._create_repair_picking_types()
warehouse._create_repair_route()
if repair_steps == "3_steps":
warehouse._create_remove_rule()
if repair_steps or repair_location_id:
warehouse.update_picking_types(repair_steps, repair_location_id)
warehouse.update_repair_routes(repair_steps, repair_location_id)
return res
def _create_repair_picking_types(self):
for warehouse in self:
repair_location_id = (
warehouse.repair_location_id.id or warehouse.lot_stock_id.id
)
if not warehouse.add_c_type_id:
pbr_type = self.env["stock.picking.type"].create(
{
"name": "Add Component to Repair",
"code": "internal",
"sequence_code": "ACR",
"warehouse_id": warehouse.id,
"default_location_src_id": warehouse.lot_stock_id.id,
"default_location_dest_id": repair_location_id,
"company_id": warehouse.company_id.id,
}
)
warehouse.add_c_type_id = pbr_type.id
else:
warehouse.add_c_type_id.write(
{"default_location_dest_id": repair_location_id}
)
if not warehouse.remove_c_type_id:
par_type = self.env["stock.picking.type"].create(
{
"name": "Remove component from Repair",
"code": "internal",
"sequence_code": "RCR",
"warehouse_id": warehouse.id,
"default_location_src_id": repair_location_id,
"default_location_dest_id": warehouse.view_location_id.id,
"company_id": warehouse.company_id.id,
}
)
warehouse.remove_c_type_id = par_type.id
else:
warehouse.remove_c_type_id.write(
{"default_location_src_id": repair_location_id}
)
def _create_repair_route(self):
for warehouse in self:
if not warehouse.repair_route_id:
route = self.env["stock.location.route"].create(
{
"name": "Repair Route for %s" % warehouse.name,
"warehouse_selectable": True,
"product_selectable": False,
"warehouse_ids": [(6, 0, warehouse.ids)],
"company_id": warehouse.company_id.id,
}
)
warehouse.repair_route_id = route.id
self.env["stock.rule"].create(
{
"name": "Add Component to Repair",
"picking_type_id": warehouse.add_c_type_id.id,
"route_id": route.id,
"location_src_id": warehouse.lot_stock_id.id,
"location_id": warehouse.repair_location_id.id
or warehouse.view_location_id.id,
"action": "pull",
"company_id": warehouse.company_id.id,
"warehouse_id": warehouse.id,
}
)
def _create_remove_rule(self):
for warehouse in self:
existing_rule = (
self.env["stock.rule"]
.with_context(active_test=False)
.search(
[
("picking_type_id", "=", warehouse.remove_c_type_id.id),
("route_id", "=", warehouse.repair_route_id.id),
],
limit=1,
)
)
if not existing_rule:
self.env["stock.rule"].create(
{
"name": "Remove component from Repair",
"picking_type_id": warehouse.remove_c_type_id.id,
"route_id": warehouse.repair_route_id.id,
"location_src_id": warehouse.repair_location_id.id
or warehouse.view_location_id.id,
"location_id": warehouse.view_location_id.id,
"action": "pull",
"company_id": warehouse.company_id.id,
"warehouse_id": warehouse.id,
"active": True,
}
)
else:
existing_rule.active = True