From d6f3fd883ce5e9d7b4c7aba760f61814d45bc8d1 Mon Sep 17 00:00:00 2001 From: Simon Date: Wed, 15 May 2024 00:11:25 +0200 Subject: [PATCH] skip empty add --- tubearchivist/home/src/ta/ta_redis.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tubearchivist/home/src/ta/ta_redis.py b/tubearchivist/home/src/ta/ta_redis.py index 53f6d57f..cba148aa 100644 --- a/tubearchivist/home/src/ta/ta_redis.py +++ b/tubearchivist/home/src/ta/ta_redis.py @@ -129,11 +129,17 @@ class RedisQueue(RedisBase): def add(self, to_add: str) -> None: """add single item to queue""" + if not to_add: + return + next_score = self._get_next_score() self.conn.zadd(self.key, {to_add: next_score}) def add_list(self, to_add: list) -> None: """add list to queue""" + if not to_add: + return + next_score = self._get_next_score() mapping = {i[1]: next_score + i[0] for i in enumerate(to_add)} self.conn.zadd(self.key, mapping) @@ -154,7 +160,7 @@ class RedisQueue(RedisBase): return last[0][1] + 1 - def get_next(self) -> tuple[str | None, int | None]: + def get_next(self) -> tuple[str | None, float | None]: """return next element in the queue, if available""" result = self.conn.zpopmin(self.key) if not result: