mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
8f424bf0d4
♻️ 重构插件系统 ⚙️ 重写插件 🎨 改进代码结构 📝 完善文档 Co-authored-by: zhxy-CN <admin@owo.cab> Co-authored-by: 洛水居室 <luoshuijs@outlook.com> Co-authored-by: xtaodada <xtao@xtaolink.cn> Co-authored-by: Li Chuangbo <im@chuangbo.li>
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
import ujson as json
|
|
|
|
from core.base.redisdb import RedisDB
|
|
from modules.wiki.base import Model
|
|
|
|
|
|
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}"
|
|
# noinspection PyBroadException
|
|
try:
|
|
result = json.loads(await self.client.get(qname))
|
|
except Exception:
|
|
result = []
|
|
if isinstance(result, list) and len(result) > 0:
|
|
for num, item in enumerate(result):
|
|
result[num] = json.loads(item)
|
|
return result
|