From bbe5eace9c5c2f3461dccf0009451c225cdfeea2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Tue, 7 Jul 2020 23:56:10 +0200
Subject: [PATCH] Add model, record and records attribute on Wizard

issue9421
review309861012
---
 stock.py | 37 ++++++++++---------------------------
 1 file changed, 10 insertions(+), 27 deletions(-)

diff --git a/stock.py b/stock.py
index 8b4a5c0d45..eb6438521a 100644
--- a/stock.py
+++ b/stock.py
@@ -108,20 +108,15 @@ class SplitMove(Wizard):
     split = StateTransition()
 
     def default_start(self, fields):
-        pool = Pool()
-        Move = pool.get('stock.move')
-        default = {}
-        move = Move(Transaction().context['active_id'])
-        default['uom'] = move.uom.id
-        default['unit_digits'] = move.unit_digits
-        default['uom_category'] = move.uom.category.id
-        return default
+        return {
+            'uom': self.record.uom.id,
+            'unit_digits': self.record.unit_digits,
+            'uom_category': self.record.uom.category.id,
+            }
 
     def transition_split(self):
-        pool = Pool()
-        Move = pool.get('stock.move')
-        move = Move(Transaction().context['active_id'])
-        move.split(self.start.quantity, self.start.uom, self.start.count)
+        self.record.split(
+            self.start.quantity, self.start.uom, self.start.count)
         return 'end'
 
 
@@ -166,17 +161,6 @@ class SplitShipment(Wizard):
             ])
     split = StateTransition()
 
-    def get_shipment(self):
-        pool = Pool()
-        context = Transaction().context
-        if context['active_model'] in {
-                'stock.shipment.in.return',
-                'stock.shipment.out',
-                'stock.shipment.internal',
-                }:
-            Shipment = pool.get(context['active_model'])
-            return Shipment(context['active_id'])
-
     def get_moves(self, shipment):
         if shipment.__name__ == 'stock.shipment.out':
             return shipment.outgoing_moves
@@ -187,8 +171,7 @@ class SplitShipment(Wizard):
             return shipment.moves
 
     def default_start(self, fields):
-        shipment = self.get_shipment()
-        moves = self.get_moves(shipment)
+        moves = self.get_moves(self.record)
         move_ids = [m.id for m in moves if m.state == 'draft']
         return {
             'domain_moves': move_ids,
@@ -197,8 +180,8 @@ class SplitShipment(Wizard):
     def transition_split(self):
         pool = Pool()
         Move = pool.get('stock.move')
-        shipment = self.get_shipment()
-        Shipment = shipment.__class__
+        shipment = self.record
+        Shipment = self.model
         if shipment.state != 'draft':
             raise ValueError("Wrong shipment state")
         if not set(self.start.moves).issubset(self.start.domain_moves):