From dc78e7b3a4331496d6917a8f3f9d2c25d577cded Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Mon, 3 Jan 2022 22:54:01 +0100
Subject: [PATCH] Setup forecast for all supplies

issue8702
review250041003
---
 CHANGELOG                       |  1 +
 __init__.py                     |  6 +++---
 purchase_request.py => stock.py | 10 +++++-----
 3 files changed, 9 insertions(+), 8 deletions(-)
 rename purchase_request.py => stock.py (74%)

diff --git a/CHANGELOG b/CHANGELOG
index bc233aa0e4..0e6fc1c09c 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -1,3 +1,4 @@
+* Setup forecast for all supplies
 * Add support for Python 3.10
 * Remove support for Python 3.6
 
diff --git a/__init__.py b/__init__.py
index 8e3087e8b7..78dd684cf9 100644
--- a/__init__.py
+++ b/__init__.py
@@ -3,10 +3,10 @@
 
 from trytond.pool import Pool
 
-from . import purchase_request
+from . import stock
 
 
 def register():
     Pool.register(
-        purchase_request.Request,
-        module='stock_supply_forecast', type_='model')
+        stock.Supply,
+        module='stock_supply_forecast', type_='wizard')
diff --git a/purchase_request.py b/stock.py
similarity index 74%
rename from purchase_request.py
rename to stock.py
index e7cada7434..cb16c71aeb 100644
--- a/purchase_request.py
+++ b/stock.py
@@ -3,11 +3,10 @@
 from trytond.pool import Pool, PoolMeta
 
 
-class Request(metaclass=PoolMeta):
-    __name__ = 'purchase.request'
+class Supply(metaclass=PoolMeta):
+    __name__ = 'stock.supply'
 
-    @classmethod
-    def generate_requests(cls, *args, **kwargs):
+    def transition_create_(self):
         pool = Pool()
         Forecast = pool.get('stock.forecast')
         Date = pool.get('ir.date')
@@ -19,5 +18,6 @@ class Request(metaclass=PoolMeta):
                 ('state', '=', 'done'),
                 ])
         Forecast.create_moves(forecasts)
-        super().generate_requests(*args, **kwargs)
+        transition = super().transition_create_()
         Forecast.delete_moves(forecasts)
+        return transition