push wms

1 files changed, 5 insertions, 3 deletions

diff --git a/indoteknik_custom/models/stock_immediate_transfer.py b/indoteknik_custom/models/stock_immediate_transfer.py
index 4be0dff2..ec00df7b 100644
--- a/indoteknik_custom/models/stock_immediate_transfer.py
+++ b/indoteknik_custom/models/stock_immediate_transfer.py
@@ -5,25 +5,25 @@ class StockImmediateTransfer(models.TransientModel):
     _inherit = 'stock.immediate.transfer'
 
     def process(self):
-        """Override process method to add send_mail_bills logic."""
         pickings_to_do = self.env['stock.picking']
         pickings_not_to_do = self.env['stock.picking']
 
         for line in self.immediate_transfer_line_ids:
             if line.to_immediate is True:
+                line.picking_id.send_mail_bills()
+                line.picking_id.send_koli_to_so()
                 pickings_to_do |= line.picking_id
             else:
                 pickings_not_to_do |= line.picking_id
 
         for picking in pickings_to_do:
-            picking.send_mail_bills()
-            # If still in draft => confirm and assign
             if picking.state == 'draft':
                 picking.action_confirm()
                 if picking.state != 'assigned':
                     picking.action_assign()
                     if picking.state != 'assigned':
                         raise UserError(_("Could not reserve all requested products. Please use the 'Mark as Todo' button to handle the reservation manually."))
+            
             for move in picking.move_lines.filtered(lambda m: m.state not in ['done', 'cancel']):
                 for move_line in move.move_line_ids:
                     move_line.qty_done = move_line.product_uom_qty
@@ -33,4 +33,6 @@ class StockImmediateTransfer(models.TransientModel):
             pickings_to_validate = self.env['stock.picking'].browse(pickings_to_validate)
             pickings_to_validate = pickings_to_validate - pickings_not_to_do
             return pickings_to_validate.with_context(skip_immediate=True).button_validate()
+        
         return True
+