Выделение БД и кэша как отдельные модули

This commit is contained in:
2024-12-04 16:38:30 +03:00
parent 76525f19d3
commit 580b7d07e5
15 changed files with 11 additions and 129 deletions

42
App/cache/__init__.py vendored Normal file
View File

@@ -0,0 +1,42 @@
import json
from redis import Redis, ConnectionError, DataError
class RedisCache:
def __init__(self, config: dict):
self.config = config
self.conn = self._connect()
def _connect(self):
try:
conn = Redis(**self.config)
return conn
except DataError as err:
print(err)
return None
def set_value(self, name: str, value_dict: dict, ttl: int):
try:
value_js = json.dumps(value_dict)
self.conn.set(name=name, value=value_js)
if ttl > 0:
self.conn.expire(name, ttl)
return True
except ConnectionError as err:
print(err)
return False
def get_value(self, name: str):
try:
value_js = self.conn.get(name)
except:
return None
value_js = self.conn.get(name)
if value_js:
value_dict = json.loads(value_js)
return value_dict
else:
return None
def __exit__(self, exc_type, exc_val, exc_tb):
self.conn.close()
return True

26
App/cache/wrapper.py vendored Normal file
View File

@@ -0,0 +1,26 @@
from functools import wraps
from . import RedisCache
def fetch_from_cache(cache_name: str, cache_config: dict):
cache_conn = RedisCache(cache_config['redis'])
ttl = cache_config['ttl']
def decorator(f):
# как оно работает
# лезем в кэш и смотрим, есть ли там что-то
# если есть, то возвращаем кэшированную информацию
# если нет, то запускаем декорируемую функцию
# достаём оттуда информацию
# заносим её в кэш
# возвращаем кэшированную информацию
@wraps(f)
def wrapper(*args, **kwargs):
cached_value = cache_conn.get_value(cache_name)
print("cached_value=", cached_value)
if cached_value:
return cached_value
response = f(*args, **kwargs)
print("response=", response)
cache_conn.set_value(cache_name,response,ttl)
return response
return wrapper
return decorator