23 lines
626 B
Python
23 lines
626 B
Python
from base.redis import redis
|
|
import json
|
|
|
|
|
|
async def get_unread_counter(chat_id: str, user_id: int):
|
|
try:
|
|
unread = await redis.execute("LLEN", f"chats/{chat_id.decode('utf-8')}/unread/{user_id}")
|
|
if unread:
|
|
return unread
|
|
except Exception:
|
|
return 0
|
|
|
|
|
|
async def get_total_unread_counter(user_id: int):
|
|
chats = await redis.execute("GET", f"chats_by_user/{str(user_id)}")
|
|
unread = 0
|
|
if chats:
|
|
chats = json.loads(chats)
|
|
for chat_id in chats:
|
|
n = await get_unread_counter(chat_id.decode('utf-8'), user_id)
|
|
unread += n
|
|
return unread
|