ltq 0.1.0__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,9 @@ 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
 
 from .errors import RetryMessage
 from .task import Task
@@ -19,17 +21,18 @@ logger = get_logger()
 
 
 P = ParamSpec("P")
+R = TypeVar("R")
 
 
 class Worker:
     def __init__(
         self,
-        client: AsyncRedis,
+        url: str = "redis://localhost:6379",
         middlewares: list[Middleware] | None = None,
         concurrency: int = 250,
         poll_sleep: float = 0.1,
     ) -> None:
-        self.client: AsyncRedis = client
+        self.client: AsyncRedis = redis.from_url(url)
         self.tasks: list[Task] = []
         self.middlewares: list[Middleware] = middlewares or []
         self.concurrency: int = concurrency
@@ -39,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)
@@ -48,7 +51,7 @@ class Worker:
                 name=task_name,
                 fn=fn,
                 queue=queue,
-                ttl=ttl, 
+                ttl=ttl,
             )
             self.tasks.append(task)
             return task
@@ -87,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.0.dist-info → ltq-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ltq
-Version: 0.1.0
+Version: 0.1.2
 Summary: Add your description here
 Author: Tom Clesius
 Author-email: Tom Clesius <tomclesius@gmail.com>
@@ -11,7 +11,7 @@ Provides-Extra: sentry
 Description-Content-Type: text/markdown
 
 <p align="center">
-  <img src="assets/logo.png" alt="LTQ" width="400">
+  <img src="https://raw.githubusercontent.com/tclesius/ltq/refs/heads/main/assets/logo.png" alt="LTQ" width="400">
 </p>
 
 <p align="center">
@@ -30,11 +30,9 @@ uv add ltq
 
 ```python
 import asyncio
-import redis.asyncio as redis
 import ltq
 
-client = redis.from_url("redis://localhost:6379")
-worker = ltq.Worker(client=client)
+worker = ltq.Worker(url="redis://localhost:6379")
 
 @worker.task()
 async def send_email(to: str, subject: str, body: str) -> None:
@@ -83,7 +81,7 @@ Add middleware to handle cross-cutting concerns:
 from ltq.middleware import Retry, RateLimit, Timeout
 
 worker = ltq.Worker(
-    client=client,
+    url="redis://localhost:6379",
     middlewares=[
         Retry(max_retries=3, min_delay=1.0),
         RateLimit(requests_per_second=10),

{ltq-0.1.0.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=dMYiX4oz_PCq0yytI0WalPPamBdVNXg1QVH92lRZkX0,2866
-ltq-0.1.0.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
-ltq-0.1.0.dist-info/entry_points.txt,sha256=OogYaOJ_RORrWtrLlEL_gTN9Vx5tkgawl8BO7G9FKcg,38
-ltq-0.1.0.dist-info/METADATA,sha256=rfFU1puSrxGuiwDHTVT7ofUfMtwSx7FvawqnXo7YgqQ,2303
-ltq-0.1.0.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,,