taskiq-redis 1.0.9__tar.gz → 1.1.1__tar.gz

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: taskiq-redis
-Version: 1.0.9
+Version: 1.1.1
 Summary: Redis integration for taskiq
+License-File: LICENSE
 Keywords: taskiq,tasks,distributed,async,redis,result_backend
 Author: taskiq-team
 Author-email: taskiq@norely.com
@@ -13,9 +14,10 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.8
-Requires-Dist: redis (>=5,<6)
+Requires-Dist: redis (>=6,<7)
 Requires-Dist: taskiq (>=0.11.12,<1)
 Project-URL: Homepage, https://github.com/taskiq-python/taskiq-redis
 Project-URL: Repository, https://github.com/taskiq-python/taskiq-redis
@@ -213,3 +215,17 @@ scheduler = TaskiqScheduler(broker, [array_source])
 
 During startup the scheduler will try to migrate schedules from an old source to a new one. Please be sure to specify different prefixe just to avoid any kind of collision between these two.
 
+
+## Dynamic queue names
+
+
+Brokers supports dynamic queue names, allowing you to specify different queues when kicking tasks. This is useful for routing tasks to specific queues based on runtime conditions, such as priority levels, tenant isolation, or environment-specific processing.
+
+Simply pass the desired queue name as message's label when kicking a task to override the broker's default queue configuration.
+
+```python
+@broker.task(queue_name="low_priority")
+async def low_priority_task() -> None:
+    print("I don't mind waiting a little longer")
+```
+

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/README.md

@@ -189,3 +189,17 @@ scheduler = TaskiqScheduler(broker, [array_source])
 ```
 
 During startup the scheduler will try to migrate schedules from an old source to a new one. Please be sure to specify different prefixe just to avoid any kind of collision between these two.
+
+
+## Dynamic queue names
+
+
+Brokers supports dynamic queue names, allowing you to specify different queues when kicking tasks. This is useful for routing tasks to specific queues based on runtime conditions, such as priority levels, tenant isolation, or environment-specific processing.
+
+Simply pass the desired queue name as message's label when kicking a task to override the broker's default queue configuration.
+
+```python
+@broker.task(queue_name="low_priority")
+async def low_priority_task() -> None:
+    print("I don't mind waiting a little longer")
+```

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "taskiq-redis"
-version = "1.0.9"
+version = "1.1.1"
 description = "Redis integration for taskiq"
 authors = ["taskiq-team <taskiq@norely.com>"]
 readme = "README.md"
@@ -27,7 +27,7 @@ keywords = [
 [tool.poetry.dependencies]
 python = "^3.9"
 taskiq = ">=0.11.12,<1"
-redis = "^5"
+redis = "^6"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8"

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/taskiq_redis/list_schedule_source.py

@@ -130,7 +130,7 @@ class ListRedisScheduleSource(ScheduleSource):
                 if key_time and key_time <= minute_before:
                     time_keys.append(key.decode())
             for key in time_keys:
-                schedules.extend(await redis.lrange(key, 0, -1))  # type: ignore
+                schedules.extend(await redis.lrange(key, 0, -1))
 
         return schedules
 
@@ -146,10 +146,10 @@ class ListRedisScheduleSource(ScheduleSource):
                 )
                 # We need to remove the schedule from the cron or time list.
                 if schedule.cron is not None:
-                    await redis.lrem(self._get_cron_key(), 0, schedule_id)  # type: ignore
+                    await redis.lrem(self._get_cron_key(), 0, schedule_id)
                 elif schedule.time is not None:
                     time_key = self._get_time_key(schedule.time)
-                    await redis.lrem(time_key, 0, schedule_id)  # type: ignore
+                    await redis.lrem(time_key, 0, schedule_id)
 
     async def add_schedule(self, schedule: "ScheduledTask") -> None:
         """Add a schedule to the source."""
@@ -163,9 +163,9 @@ class ListRedisScheduleSource(ScheduleSource):
             # This is an optimization, so we can get all the schedules
             # for the current time much faster.
             if schedule.cron is not None:
-                await redis.rpush(self._get_cron_key(), schedule.schedule_id)  # type: ignore
+                await redis.rpush(self._get_cron_key(), schedule.schedule_id)
             elif schedule.time is not None:
-                await redis.rpush(  # type: ignore
+                await redis.rpush(
                     self._get_time_key(schedule.time),
                     schedule.schedule_id,
                 )
@@ -195,11 +195,11 @@ class ListRedisScheduleSource(ScheduleSource):
             self._is_first_run = False
         async with Redis(connection_pool=self._connection_pool) as redis:
             buffer = []
-            crons = await redis.lrange(self._get_cron_key(), 0, -1)  # type: ignore
+            crons = await redis.lrange(self._get_cron_key(), 0, -1)
             logger.debug("Got %d cron schedules", len(crons))
             if crons:
                 buffer.extend(crons)
-            timed.extend(await redis.lrange(self._get_time_key(current_time), 0, -1))  # type: ignore
+            timed.extend(await redis.lrange(self._get_time_key(current_time), 0, -1))
             logger.debug("Got %d timed schedules", len(timed))
             if timed:
                 buffer.extend(timed)

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/taskiq_redis/redis_backend.py

@@ -131,7 +131,7 @@ class RedisAsyncResultBackend(AsyncResultBackend[_ReturnType]):
             redis_set_params["px"] = self.result_px_time
 
         async with Redis(connection_pool=self.redis_pool) as redis:
-            await redis.set(**redis_set_params)  # type: ignore
+            await redis.set(**redis_set_params)
 
     async def is_result_ready(self, task_id: str) -> bool:
         """
@@ -205,7 +205,7 @@ class RedisAsyncResultBackend(AsyncResultBackend[_ReturnType]):
             redis_set_params["px"] = self.result_px_time
 
         async with Redis(connection_pool=self.redis_pool) as redis:
-            await redis.set(**redis_set_params)  # type: ignore
+            await redis.set(**redis_set_params)
 
     async def get_progress(
         self,

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/taskiq_redis/redis_broker.py

@@ -122,7 +122,7 @@ class ListQueueBroker(BaseRedisBroker):
         """
         queue_name = message.labels.get("queue_name") or self.queue_name
         async with Redis(connection_pool=self.connection_pool) as redis_conn:
-            await redis_conn.lpush(queue_name, message.message)  # type: ignore
+            await redis_conn.lpush(queue_name, message.message)
 
     async def listen(self) -> AsyncGenerator[bytes, None]:
         """
@@ -137,7 +137,7 @@ class ListQueueBroker(BaseRedisBroker):
         while True:
             try:
                 async with Redis(connection_pool=self.connection_pool) as redis_conn:
-                    yield (await redis_conn.brpop(self.queue_name))[  # type: ignore
+                    yield (await redis_conn.brpop(self.queue_name))[
                         redis_brpop_data_position
                     ]
             except ConnectionError as exc:
@@ -251,9 +251,10 @@ class RedisStreamBroker(BaseRedisBroker):
 
         :param message: message to append.
         """
+        queue_name = message.labels.get("queue_name") or self.queue_name
         async with Redis(connection_pool=self.connection_pool) as redis_conn:
             await redis_conn.xadd(
-                self.queue_name,
+                queue_name,
                 {b"data": message.message},
                 maxlen=self.maxlen,
                 approximate=self.approximate,
@@ -280,7 +281,7 @@ class RedisStreamBroker(BaseRedisBroker):
                     self.consumer_name,
                     {
                         self.queue_name: ">",
-                        **self.additional_streams,  # type: ignore
+                        **self.additional_streams,
                     },
                     block=self.block,
                     noack=False,
@@ -310,7 +311,7 @@ class RedisStreamBroker(BaseRedisBroker):
                         )
                         logger.debug(
                             "Found %d pending messages in stream %s",
-                            len(pending),
+                            len(pending[1]),
                             stream,
                         )
                         for msg_id, msg in pending[1]:

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/taskiq_redis/redis_cluster_broker.py

@@ -55,7 +55,8 @@ class ListQueueClusterBroker(BaseRedisClusterBroker):
 
         :param message: message to append.
         """
-        await self.redis.lpush(self.queue_name, message.message)  # type: ignore
+        queue_name = message.labels.get("queue_name") or self.queue_name
+        await self.redis.lpush(queue_name, message.message)  # type: ignore
 
     async def listen(self) -> AsyncGenerator[bytes, None]:
         """
@@ -162,8 +163,9 @@ class RedisStreamClusterBroker(BaseRedisClusterBroker):
 
         :param message: message to append.
         """
+        queue_name = message.labels.get("queue_name") or self.queue_name
         await self.redis.xadd(
-            self.queue_name,
+            queue_name,
             {b"data": message.message},
             maxlen=self.maxlen,
             approximate=self.approximate,

{taskiq_redis-1.0.9 → taskiq_redis-1.1.1}/taskiq_redis/redis_sentinel_broker.py

@@ -230,9 +230,10 @@ class RedisStreamSentinelBroker(BaseSentinelBroker):
 
         :param message: message to append.
         """
+        queue_name = message.labels.get("queue_name") or self.queue_name
         async with self._acquire_master_conn() as redis_conn:
             await redis_conn.xadd(
-                self.queue_name,
+                queue_name,
                 {b"data": message.message},
                 maxlen=self.maxlen,
                 approximate=self.approximate,