PamGram/core/wiki/cache.py

import ujson as json

from models.wiki.base import Model
from utils.redisdb import RedisDB


class WikiCache:
    def __init__(self, redis: RedisDB):
        self.client = redis.client
        self.qname = "wiki"

    async def set(self, key: str, value):
        qname = f"{self.qname}:{key}"
        if isinstance(value, Model):
            value = value.json()
        elif isinstance(value, (dict, list)):
            value = json.dumps(value)
        await self.client.set(qname, value)

    async def delete(self, key: str):
        qname = f"{self.qname}:{key}"
        await self.client.delete(qname)

    async def get(self, key: str) -> dict:
        qname = f"{self.qname}:{key}"
        data = await self.client.get(qname)
        json_data = str(data, encoding="utf-8")
        result = json.loads(json_data)
        if isinstance(result, list) and len(result) > 0:
            for num, item in enumerate(result):
                result[num] = json.loads(item)
        return result