feat: ability to call tasks directly w/o enqueue-ing them

kolona/task.py

@@ -81,6 +81,9 @@ async def enqueue(
         await q.put(task_item)
         return task_item.id
 
+    async def __call__(self, *args, **kwargs):
+        return await self.func(*args, **kwargs)
+
 
 class Task(GlobalTask):
     """
@@ -161,14 +164,22 @@ def task(
     """
 
     def wrapper(func):
-        task = GlobalTask(
+        task_obj = GlobalTask(
             func,
             queue=queue,
             max_retries=max_retries,
             retry_intervals=retry_intervals,
             delay=delay,
         )
-        update_wrapper(task, func)
-        return task
-
-    return wrapper
+
+        async def callable_task(*args, **kwargs):
+            if kwargs.pop('enqueue', False):
+                return await task_obj.enqueue(*args, **kwargs)
+            else:
+                return await task_obj(*args, **kwargs)
+
+        update_wrapper(callable_task, func)
+        callable_task.enqueue = task_obj.enqueue
+        return callable_task
+
+    return wrapper

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = kolona
-version = 0.2.0
+version = 0.2.1
 description = A minimalistic in-memory async Python Task queue
 long_description = file: README.md
 long_description_content_type = text/markdown

tests/test_integration.py

@@ -319,3 +319,14 @@ async def runner():
 
     assert log_spy.warning.call_count == 1
     assert queue.qsize() == 0
+
+
+async def test_calling_task_without_enqueing():
+
+    @task()
+    async def random_task():
+        return True
+
+    result = await random_task()
+    assert result == True
+