core/services/unread.py

from services.redis import redis
import json


async def get_unread_counter(chat_id: str, author_id: int):
    try:
        unread = await redis.execute(
            "LLEN", f"chats/{chat_id}/unread/{author_id}"
        )
        return unread or 0
    except Exception:
        return 0


async def get_total_unread_counter(author_id: int):
    print(f"[services.unread] get_total_unread_counter({author_id})")
    chats = await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")
    unread = 0
    for chat_id in list(chats):
        n = await get_unread_counter(chat_id, author_id)
        unread += n
    return unread