tubearchivist
/
tubearchivist
mirror of https://github.com/tubearchivist/tubearchivist.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

skip empty add

This commit is contained in:
Simon 2024-05-15 00:11:25 +02:00
parent 8f1a5c8557
commit d6f3fd883c
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tubearchivist/home/src/ta/ta_redis.py
View File

@ -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: