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