mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-23 00:01:45 +00:00
69 lines
2.8 KiB
Python
69 lines
2.8 KiB
Python
from typing import List
|
|
|
|
import ujson
|
|
|
|
from core.base.redisdb import RedisDB
|
|
from .models import Answer, Question
|
|
|
|
|
|
class QuizCache:
|
|
def __init__(self, redis: RedisDB):
|
|
self.client = redis.client
|
|
self.question_qname = "quiz:question"
|
|
self.answer_qname = "quiz:answer"
|
|
|
|
async def get_all_question(self) -> List[Question]:
|
|
temp_list = []
|
|
qname = self.question_qname + "id_list"
|
|
data_list = [self.question_qname + f":{question_id}" for question_id in await self.client.lrange(qname, 0, -1)]
|
|
data = await self.client.mget(data_list)
|
|
for i in data:
|
|
temp_list.append(Question.de_json(ujson.loads(i)))
|
|
return temp_list
|
|
|
|
async def get_all_question_id_list(self) -> List[str]:
|
|
qname = self.question_qname + ":id_list"
|
|
return await self.client.lrange(qname, 0, -1)
|
|
|
|
async def get_one_question(self, question_id: int) -> Question:
|
|
qname = f"{self.question_qname}:{question_id}"
|
|
data = await self.client.get(qname)
|
|
json_data = str(data, encoding="utf-8")
|
|
return Question.de_json(ujson.loads(json_data))
|
|
|
|
async def get_one_answer(self, answer_id: int) -> Answer:
|
|
qname = f"{self.answer_qname}:{answer_id}"
|
|
data = await self.client.get(qname)
|
|
json_data = str(data, encoding="utf-8")
|
|
return Answer.de_json(ujson.loads(json_data))
|
|
|
|
async def add_question(self, question_list: List[Question] = None) -> int:
|
|
if not question_list:
|
|
return 0
|
|
for question in question_list:
|
|
await self.client.set(f"{self.question_qname}:{question.question_id}", ujson.dumps(question.to_dict()))
|
|
question_id_list = [question.question_id for question in question_list]
|
|
await self.client.lpush(f"{self.question_qname}:id_list", *question_id_list)
|
|
return await self.client.llen(f"{self.question_qname}:id_list")
|
|
|
|
async def del_all_question(self):
|
|
keys = await self.client.keys(f"{self.question_qname}*")
|
|
if keys is not None:
|
|
for key in keys:
|
|
await self.client.delete(key)
|
|
|
|
async def del_all_answer(self):
|
|
keys = await self.client.keys(f"{self.answer_qname}*")
|
|
if keys is not None:
|
|
for key in keys:
|
|
await self.client.delete(key)
|
|
|
|
async def add_answer(self, answer_list: List[Answer] = None) -> int:
|
|
if not answer_list:
|
|
return 0
|
|
for answer in answer_list:
|
|
await self.client.set(f"{self.answer_qname}:{answer.answer_id}", ujson.dumps(answer.to_dict()))
|
|
answer_id_list = [answer.answer_id for answer in answer_list]
|
|
await self.client.lpush(f"{self.answer_qname}:id_list", *answer_id_list)
|
|
return await self.client.llen(f"{self.answer_qname}:id_list")
|