mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
29 lines
849 B
Python
29 lines
849 B
Python
|
from typing import Any
|
||
|
|
import pickle # nosec B403
|
||
|
|
import gzip
|
||
|
|
|
||
|
|
from core.base.redisdb import RedisDB
|
||
|
|
|
||
|
|
|
||
|
|
class TemplatePreviewCache:
|
||
|
|
'''暂存渲染模板的数据用于预览'''
|
||
|
|
|
||
|
|
def __init__(self, redis: RedisDB):
|
||
|
|
self.client = redis.client
|
||
|
|
self.qname = "bot:template:preview"
|
||
|
|
|
||
|
|
async def get_data(self, key: str) -> Any:
|
||
|
|
data = await self.client.get(self.cache_key(key))
|
||
|
|
if data:
|
||
|
|
# skipcq: BAN-B301
|
||
|
|
return pickle.loads(gzip.decompress(data)) # nosec B301
|
||
|
|
|
||
|
|
async def set_data(self, key: str, data: Any, ttl: int = 8 * 60 * 60):
|
||
|
|
ck = self.cache_key(key)
|
||
|
|
await self.client.set(ck, gzip.compress(pickle.dumps(data)))
|
||
|
|
if ttl != -1:
|
||
|
|
await self.client.expire(ck, ttl)
|
||
|
|
|
||
|
|
def cache_key(self, key: str) -> str:
|
||
|
|
return f"{self.qname}:{key}"
|