flask-demo-api/flask_demo_api/repository/redis.py

import pickle

import redis  # type: ignore

from flask_demo_api.protocols.models import KeyDTO
from flask_demo_api.protocols.repository import Repository


class RedisRepository(Repository):
    def __init__(self, redis_pool: redis.Redis) -> None:
        self.pool = redis_pool

    def get_key(self, obj: KeyDTO) -> KeyDTO | None:
        data = self.pool.get(obj.key)
        if not data:
            return None
        return KeyDTO(key=str(obj.key), val=data)

    def add_key(self, obj: KeyDTO) -> KeyDTO:
        self.pool.set(str(obj.key), pickle.dumps(obj.val))
        return obj

    def put_key(self, obj: KeyDTO) -> KeyDTO:
        return KeyDTO(key="5", val="6")

    def delete_key(self, obj: KeyDTO) -> None:
        self.pool.delete(str(obj.key))
        return None