From d7028b6d744991148bcacb6e2eb260c1733ffed0 Mon Sep 17 00:00:00 2001
From: eeintech <eeintech@eeinte.ch>
Date: Thu, 22 Jul 2021 16:34:35 -0400
Subject: [PATCH] Make it generic method instead

---
 InvenTree/InvenTree/tasks.py | 46 ++++++++++++++++++++++++++++++++++++
 InvenTree/InvenTree/views.py | 14 ++++-------
 2 files changed, 50 insertions(+), 10 deletions(-)

diff --git a/InvenTree/InvenTree/tasks.py b/InvenTree/InvenTree/tasks.py
index cab87ee64b..1c77febf79 100644
--- a/InvenTree/InvenTree/tasks.py
+++ b/InvenTree/InvenTree/tasks.py
@@ -69,6 +69,47 @@ def offload_task(taskname, *args, **kwargs):
     task.run()
 
 
+def run_task(taskname):
+    """
+        1. Check if task is implemented
+            - yes: proceed
+            - no: return
+        2. Check if worker cluster is running
+            - yes: add task to queue
+            - no: run it as blocking process
+    """
+
+    # Get task list
+    tasks = get_task_list()
+
+    # Check if task exists
+    if taskname not in tasks:
+        logger.warning(f'Task "{taskname}" is not implemented')
+        return
+
+    from InvenTree.status import is_worker_running
+
+    if is_worker_running():
+        # Append module path
+        taskname = 'InvenTree.tasks.' + taskname
+        # Running as task
+        offload_task(taskname)
+    else:
+        # Retrieve local method from task name
+        _func = eval(taskname)
+        # Run it as blocking process
+        _func()
+
+
+def get_task_list():
+    return [task for task in LOCAL_METHODS if task not in TASK_MANAGEMENT]
+
+
+# Keep TASK_MANAGEMENT before task methods
+TASK_MANAGEMENT = [key for key, value in locals().items() if callable(value) and value.__module__ == __name__]
+#
+
+
 def heartbeat():
     """
     Simple task which runs at 5 minute intervals,
@@ -217,3 +258,8 @@ def send_email(subject, body, recipients, from_email=None):
         from_email,
         recipients,
     )
+
+
+# Keep LOCAL_METHODS at the end of the file
+LOCAL_METHODS = [key for key, value in locals().items() if callable(value) and value.__module__ == __name__]
+#
diff --git a/InvenTree/InvenTree/views.py b/InvenTree/InvenTree/views.py
index 80d0a015ff..88199f39a0 100644
--- a/InvenTree/InvenTree/views.py
+++ b/InvenTree/InvenTree/views.py
@@ -803,17 +803,11 @@ class CurrencyRefreshView(RedirectView):
         On a POST request we will attempt to refresh the exchange rates
         """
 
-        # Define associated task
-        task_name = 'InvenTree.tasks.update_exchange_rates'
+        # Define associated task from InvenTree.tasks list of methods
+        taskname = 'update_exchange_rates'
 
-        if InvenTree.status.is_worker_running():
-            # Running as task
-            InvenTree.tasks.offload_task(task_name)
-        else:
-            # Retrieve function from task name
-            _func = eval(task_name)
-            # Run it: will block for a little bit
-            _func()
+        # Run it
+        InvenTree.tasks.run_task(taskname)
 
         return self.get(request, *args, **kwargs)