ltq 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

ltq/task.py

@@ -1,18 +1,20 @@
 from __future__ import annotations
 
-from typing import Any, Awaitable, Callable, Generic, ParamSpec
+from functools import update_wrapper
+from typing import Awaitable, Callable, Generic, ParamSpec, TypeVar
 
 from .message import Message
 from .q import Queue
 
 P = ParamSpec("P")
+R = TypeVar("R")
 
 
-class Task(Generic[P]):
+class Task(Generic[P, R]):
     def __init__(
         self,
         name: str,
-        fn: Callable[P, Awaitable[Any]],
+        fn: Callable[P, Awaitable[R]],
         queue: Queue,
         ttl: int | None = None,
     ) -> None:
@@ -36,3 +38,6 @@ class Task(Generic[P]):
     async def send_bulk(self, messages: list[Message]) -> list[str]:
         await self.queue.put(messages, ttl=self.ttl)
         return [message.id for message in messages]
+
+    async def __call__(self, *args: P.args, **kwargs: P.kwargs) -> R:
+        return await self.fn(*args, **kwargs)

ltq/worker.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import asyncio
 from functools import partial
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Awaitable, Callable, ParamSpec
+from typing import TYPE_CHECKING, Any, Awaitable, Callable, ParamSpec, TypeVar
 
 import redis.asyncio as redis
 
@@ -21,6 +21,7 @@ logger = get_logger()
 
 
 P = ParamSpec("P")
+R = TypeVar("R")
 
 
 class Worker:
@@ -41,8 +42,8 @@ class Worker:
         self,
         queue_name: str | None = None,
         ttl: int | None = None,
-    ) -> Callable[[Callable[P, Awaitable[Any]]], Task[P]]:
-        def decorator(fn: Callable[P, Awaitable[Any]]) -> Task[P]:
+    ) -> Callable[[Callable[P, Awaitable[R]]], Task[P, R]]:
+        def decorator(fn: Callable[P, Awaitable[R]]) -> Task[P, R]:
             filename = Path(fn.__code__.co_filename).stem
             task_name = f"{filename}:{fn.__qualname__}"
             queue = Queue(self.client, queue_name or task_name)
@@ -50,7 +51,7 @@ class Worker:
                 name=task_name,
                 fn=fn,
                 queue=queue,
-                ttl=ttl, 
+                ttl=ttl,
             )
             self.tasks.append(task)
             return task
@@ -89,5 +90,8 @@ class Worker:
             await task.queue.ack(messages)
 
     async def run(self) -> None:
-        workers = (self.worker(task) for task in self.tasks)
-        await asyncio.gather(*workers)
+        try:
+            workers = (self.worker(task) for task in self.tasks)
+            await asyncio.gather(*workers)
+        finally:
+            await self.client.aclose()

{ltq-0.1.1.dist-info → ltq-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ltq
-Version: 0.1.1
+Version: 0.1.2
 Summary: Add your description here
 Author: Tom Clesius
 Author-email: Tom Clesius <tomclesius@gmail.com>

{ltq-0.1.1.dist-info → ltq-0.1.2.dist-info}/RECORD

@@ -5,9 +5,9 @@ ltq/logger.py,sha256=HPClhDt3ecwZqE0Vq2oYF8Nr9jj-xrsSX9tM6enVgkA,1791
 ltq/message.py,sha256=C6gJR6KuRrnIb-l9Jna7e-XIu_aFHBhnZICUqE1_2MU,715
 ltq/middleware.py,sha256=NMRI7UiJiUTqAHOB18Gz686mQGszSu-6hJiyTDPCKdE,3631
 ltq/q.py,sha256=LddedNdb9uYD9qAjkzN0inBhvQ-mp6uJLgguEhv_TeE,2462
-ltq/task.py,sha256=anX9jKGGrNr8JDEhJ4L-QALbOUSDWkYHRwjyZDYCCDs,1008
-ltq/worker.py,sha256=wqWHzRcZwDP8TRX1TJ6DA7RlIsiVyF4oeZcQdDywZps,2927
-ltq-0.1.1.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
-ltq-0.1.1.dist-info/entry_points.txt,sha256=OogYaOJ_RORrWtrLlEL_gTN9Vx5tkgawl8BO7G9FKcg,38
-ltq-0.1.1.dist-info/METADATA,sha256=fW53GkOFYMynG5IjrdsJFn4kjOMhDesBTdGla5oYj2I,2316
-ltq-0.1.1.dist-info/RECORD,,
+ltq/task.py,sha256=JOeyulGJ-jSM_wzYyKPTPElRGx7Ncu6n1dRdn9NGwMI,1184
+ltq/worker.py,sha256=PAMW8hvTuWCUALw41TnVBmcCxXytJbgVsxALAY9qmFY,3031
+ltq-0.1.2.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
+ltq-0.1.2.dist-info/entry_points.txt,sha256=OogYaOJ_RORrWtrLlEL_gTN9Vx5tkgawl8BO7G9FKcg,38
+ltq-0.1.2.dist-info/METADATA,sha256=7Mvz05MwsRJ5ddbegQofl9iLDHgELxwY2JEsSnJGK5Q,2316
+ltq-0.1.2.dist-info/RECORD,,