<miqdad> return oke

1 files changed, 131 insertions, 111 deletions

diff --git a/indoteknik_custom/models/tukar_guling.py b/indoteknik_custom/models/tukar_guling.py
index 7bcf5e80..bdd2a2f5 100644
--- a/indoteknik_custom/models/tukar_guling.py
+++ b/indoteknik_custom/models/tukar_guling.py
@@ -11,6 +11,11 @@ class TukarGuling(models.Model):
     _order = 'date desc, id desc'
     _rec_name = 'name'
 
+    picking_ids = fields.One2many(
+        'stock.picking',
+        'tukar_guling_id',
+        string='Transfers')
+
     origin = fields.Char(string='Origin SO')
 
     real_shipping_id = fields.Many2one('res.partner', string='Shipping Address')
@@ -44,7 +49,16 @@ class TukarGuling(models.Model):
     def _onchange_operations(self):
         """Auto-populate lines ketika operations dipilih"""
         if self.operations:
-            # Clear existing lines
+            from_return_picking = self.env.context.get('from_return_picking', False) or \
+                                  self.env.context.get('default_line_ids', False)
+
+            if self.line_ids and from_return_picking:
+                # Hanya update origin, jangan ubah lines
+                if self.operations.origin:
+                    self.origin = self.operations.origin
+                return
+
+            # Clear existing lines hanya jika tidak dari return picking
             self.line_ids = [(5, 0, 0)]
 
             # Set origin dari operations
@@ -91,10 +105,14 @@ class TukarGuling(models.Model):
             else:
                 _logger.info("No lines created - no valid moves found")
         else:
-            # Clear lines jika operations dikosongkan
-            self.line_ids = [(5, 0, 0)]
-            self.origin = False
+            # Clear lines jika operations dikosongkan, kecuali dari return picking
+            from_return_picking = self.env.context.get('from_return_picking', False) or \
+                                  self.env.context.get('default_line_ids', False)
 
+            if not from_return_picking:
+                self.line_ids = [(5, 0, 0)]
+
+            self.origin = False
     def action_populate_lines(self):
         """Manual button untuk populate lines - sebagai alternatif"""
         self.ensure_one()
@@ -272,113 +290,115 @@ class TukarGuling(models.Model):
         self.state = 'cancel'
 
     def _create_pickings(self):
-        if not self.operations:
-            raise UserError("BU/Out harus diisi terlebih dahulu.")
-
-        origin_so = self.operations.origin
-        if not origin_so:
-            raise UserError("BU/OUT tidak memiliki origin (SO), tidak bisa cari BU/PICK.")
-
-        # Cari DO dari SO
-        get_group_id = self.env['stock.picking'].search([
-            ('origin', '=', origin_so),
-        ], limit=1)
-
-        if not get_group_id:
-            raise UserError(f"Delivery Order dari SO {origin_so} tidak ditemukan.")
-
-        group_id = get_group_id.group_id.id if get_group_id.group_id else False
-
-        Picking = self.env['stock.picking']
-        srt_type = self.env['stock.picking.type'].search([
-            ('sequence_code', '=', 'SRT')
-        ], limit=1)
-
-        ort_type = self.env['stock.picking.type'].search([
-            ('sequence_code', '=', 'ORT')
-        ], limit=1)
-
-        # Lokasi
-        location_dest_id = srt_type.default_location_dest_id.id
-        location_dest_id_ort = ort_type.default_location_dest_id.id
-        location_customer = self.operations.location_dest_id
-
-        # 1. BU/SRT: retur dari operations
-        srt_picking = Picking.create({
-            'partner_id': self.operations.partner_id.id,
-            'real_shipping_id': self.operations.real_shipping_id.id,
-            'picking_type_id': srt_type.id,
-            'location_id': location_customer.id,
-            'location_dest_id': location_dest_id,
-            'origin': f"Retur {self.operations.name}",
-            'tukar_guling_id': self.id,
-            'group_id': group_id,
-            'move_ids_without_package': [
-                (0, 0, {
-                    'name': line.name or line.product_id.name,
-                    'product_id': line.product_id.id,
-                    'product_uom_qty': line.product_uom_qty,
-                    'product_uom': line.product_uom.id,
-                    'location_id': location_customer.id,
-                    'location_dest_id': location_dest_id,
-                    'group_id': group_id,
-                }) for line in self.line_ids
-            ]
-        })
-        srt_picking.action_confirm()
-
-        # 2. Cari BU/PICK dari SO yang sama
-        origin_so = self.operations.origin
-        if not origin_so:
-            raise UserError("BU/OUT tidak memiliki origin (SO), tidak bisa cari BU/PICK.")
-
-        pick = Picking.search([
-            ('origin', '=', origin_so),
-            ('picking_type_id.code', '=', 'internal')
-        ], limit=1)
-
-        if not pick:
-            raise UserError(f"BU/PICK dengan origin {origin_so} tidak ditemukan.")
-
-        # 3. BU/ORT: retur dari BU/PICK
-        ort_picking = Picking.create({
-            'partner_id': self.operations.partner_id.id,
-            'real_shipping_id': self.operations.real_shipping_id.id,
-            'picking_type_id': ort_type.id,
-            'location_id': location_dest_id,
-            'location_dest_id': location_dest_id_ort,
-            'origin': f"Retur {pick.name}",
-            'tukar_guling_id': self.id,
-            'group_id': group_id,
-            'move_ids_without_package': [
-                (0, 0, {
-                    'name': line.name or line.product_id.name,
-                    'product_id': line.product_id.id,
-                    'product_uom_qty': line.product_uom_qty,
-                    'product_uom': line.product_uom.id,
-                    'location_id': location_dest_id,
-                    'location_dest_id': location_dest_id_ort,
-                    'group_id': group_id,
-                }) for line in self.line_ids
-            ]
-        })
-        for line in self.line_ids:
-            move = ort_picking.move_ids_without_package.filtered(
-                lambda m: m.product_id == line.product_id
-            )[:1]
-
-            if move:
-                self.env['stock.move.line'].create({
-                    'move_id': move.id,
-                    'picking_id': ort_picking.id,
-                    'product_id': line.product_id.id,
-                    'product_uom_id': line.product_uom.id,
-                    'qty_done': line.product_uom_qty,  # Ambil dari return.picking.line.quantity
-                    'location_id': location_customer.id,
-                    'location_dest_id': location_dest_id,
+        for record in self:
+            if not record.operations:
+                raise UserError("BU/OUT dari field operations tidak ditemukan.")
+
+            operation_picking = record.operations
+
+            # 1. Cari semua picking DONE berdasarkan origin SO
+            related_pickings = self.env['stock.picking'].search([
+                ('origin', '=', record.origin),
+                ('state', '=', 'done'),
+            ])
+            if not related_pickings:
+                raise UserError("Tidak ditemukan BU/PICK atau BU/OUT dari SO: %s" % record.origin + "Atau masih belum Done")
+
+            # 2. Filter berdasarkan tipe picking
+            bu_pick_to_return = related_pickings.filtered(lambda p: p.picking_type_id.id == 30)  # BU/PICK
+            bu_out_to_return = related_pickings.filtered(lambda p: p.picking_type_id.id == 29)  # BU/OUT
+
+            if not bu_pick_to_return and not bu_out_to_return:
+                raise UserError("Tidak ada BU/PICK atau BU/OUT yang selesai untuk diretur.")
+
+            created_returns = []
+
+            # Lokasi default untuk retur
+            bu_out_type = self.env['stock.picking.type'].browse(73)
+            bu_stock_type = self.env['stock.picking.type'].browse(74)
+
+            bu_out = bu_out_type.default_location_src_id.id
+            bu_stock = bu_out_type.default_location_dest_id.id
+
+            if not bu_out or not bu_stock:
+                raise UserError("salahwoi")
+
+            partner_location = self.env['stock.location'].browse(2)
+            if not partner_location:
+                raise UserError("Lokasi partner (real_shipping_id) tidak ditemukan pada BU/OUT utama.")
+
+            # Fungsi membuat retur dari picking tertentu
+            def _create_return_from_picking(picking):
+                grup = self.env['stock.picking'].search([('origin', '=', self.operations.origin)])
+                # Tentukan lokasi berdasarkan jenis picking
+                if picking.picking_type_id.id == 29:  # BU/OUT → BU/SRT
+                    default_location_id = partner_location.id
+                    default_location_dest_id = bu_out
+                elif picking.picking_type_id.id == 30:  # BU/PICK → BU/ORT
+                    default_location_id = bu_out
+                    default_location_dest_id = bu_stock
+                else:
+                    return None
+
+                return_context = dict(self.env.context)
+                return_context.update({
+                    'active_id': picking.id,
+                    'default_location_id': default_location_id,
+                    'default_location_dest_id': default_location_dest_id,
+                    'from_ui': False,
                 })
-        ort_picking.action_confirm()
-        ort_picking.action_assign()
+
+                return_wizard = self.env['stock.return.picking'].with_context(return_context).create({
+                    'picking_id': picking.id,
+                    'location_id': default_location_id,
+                })
+
+                # Buat return lines
+                return_lines = []
+                for move in picking.move_lines:
+                    if move.quantity_done > 0:
+                        return_lines.append((0, 0, {
+                            'product_id': move.product_id.id,
+                            'quantity': move.quantity_done,
+                            'move_id': move.id,
+                        }))
+                if not return_lines:
+                    return None
+
+                return_wizard.product_return_moves = return_lines
+
+                _logger.info("Creating return for picking %s", picking.name)
+                _logger.info("Default location src: %s", default_location_id)
+                _logger.info("Default location dest: %s", default_location_dest_id)
+                _logger.info("Move lines: %s", picking.move_lines)
+                return_vals = return_wizard.create_returns()
+                return_id = return_vals.get('res_id')
+
+                if not return_id:
+                    raise UserError("Retur gagal dibuat. Hasil create_returns: %s" % str(return_vals))
+
+                picking_obj = self.env['stock.picking'].browse(return_id)
+                for p in picking_obj:
+                    p.group_id = self.operations.group_id.id
+                    p.origin_tukar_guling_id = record.id
+
+                return picking_obj.name
+
+            # Buat return dari BU/PICK
+            for picking in bu_pick_to_return:
+                name = _create_return_from_picking(picking)
+                if name:
+                    created_returns.append(name)
+
+            # Buat return dari BU/OUT
+            for picking in bu_out_to_return:
+                name = _create_return_from_picking(picking)
+                if name:
+                    created_returns.append(name)
+
+            if not created_returns:
+                raise UserError("wkwkwk")
+
 
 class TukarGulingLine(models.Model):
     _name = 'tukar.guling.line'
@@ -418,4 +438,4 @@ class TukarGulingLine(models.Model):
 class StockPicking(models.Model):
     _inherit = 'stock.picking'
 
-    origin_tukar_guling_id = fields.Many2one('tukar.guling', string='Tukar Guling Ref')
\ No newline at end of file
+    tukar_guling_id = fields.Many2one('tukar.guling', string='Tukar Guling Ref')
\ No newline at end of file