ltq 0.1.0__tar.gz → 0.1.2__tar.gz

{ltq-0.1.0 → ltq-0.1.2}/PKG-INFO

@@ -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 → ltq-0.1.2}/README.md

@@ -1,5 +1,5 @@
 <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">
@@ -18,11 +18,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:
@@ -71,7 +69,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 → ltq-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ltq"
-version = "0.1.0"
+version = "0.1.2"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "Tom Clesius", email = "tomclesius@gmail.com" }]

{ltq-0.1.0 → ltq-0.1.2}/src/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-0.1.0 → ltq-0.1.2}/src/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()