manufacture/mrp_subcontracting_purchase/models/res_company.py

from odoo import api, models


class ResCompany(models.Model):
    _inherit = "res.company"

    def _create_subcontracting_dropshipping_rules(self):
        """Adds new dropshipping stock rules for subcontracting"""
        route = self.env.ref(
            "mrp_subcontracting_purchase.route_subcontracting_dropshipping",
            raise_if_not_found=False,
        )
        supplier_location = self.env.ref(
            "stock.stock_location_suppliers", raise_if_not_found=False
        )
        vals = self._prepared_subcontracting_dropshipping_rules(
            route, supplier_location
        )
        if vals:
            self.env["stock.rule"].create(vals)

    def _prepared_subcontracting_dropshipping_rules(self, route, supplier_location):
        vals = []
        for company in self:
            subcontracting_location = company.subcontracting_location_id
            dropship_picking_type = self.env["stock.picking.type"].search(
                [
                    ("company_id", "=", company.id),
                    ("default_location_src_id.usage", "=", "supplier"),
                    ("default_location_dest_id.usage", "=", "customer"),
                ],
                limit=1,
                order="sequence",
            )
            if dropship_picking_type:
                vals.append(
                    {
                        "name": "%s → %s"
                        % (supplier_location.name, subcontracting_location.name),
                        "action": "buy",
                        "location_id": subcontracting_location.id,
                        "location_src_id": supplier_location.id,
                        "procure_method": "make_to_stock",
                        "route_id": route.id,
                        "picking_type_id": dropship_picking_type.id,
                        "company_id": company.id,
                    }
                )
        return vals

    @api.model
    def create_missing_subcontracting_dropshipping_rules(self):
        """Adds new stock rules for missing subcontracting dropshipping"""
        route = self.env.ref(
            "mrp_subcontracting_purchase.route_subcontracting_dropshipping",
            raise_if_not_found=False,
        )
        companies = self.env["res.company"].search([])
        company_has_rules = (
            self.env["stock.rule"]
            .search([("route_id", "=", route.id)])
            .mapped("company_id")
        )
        company_todo_rules = companies - company_has_rules
        company_todo_rules._create_subcontracting_dropshipping_rules()

    def _create_per_company_rules(self):
        res = super(ResCompany, self)._create_per_company_rules()
        self.create_missing_subcontracting_dropshipping_rules()
        return res