|
|
|
import time
|
|
import json
|
|
import hashlib
|
|
|
|
|
|
from pydantic import BaseModel
|
|
|
|
class InputHotTopic(BaseModel):
|
|
start_time: str = ""
|
|
end_time: str = ""
|
|
query: str = ""
|
|
keywords: list = []
|
|
top_cluster: int = 5
|
|
prompt: str = ""
|
|
check_relevent: str = ""
|
|
class SessionProcess(object):
|
|
|
|
def __init__(self):
|
|
self.session = dict()
|
|
|
|
def hash_session(self, query: InputHotTopic):
|
|
hash_dict = query.dict()
|
|
hash_dict['time'] = int(time.time())
|
|
return hashlib.sha224(json.dumps(hash_dict).encode("utf-8")).hexdigest()
|
|
|
|
def insert_session(self, data_input):
|
|
print('data_input: ', data_input)
|
|
|
|
|
|
|
|
hash_id = self.hash_session(data_input)
|
|
if hash_id not in self.session:
|
|
self.session[hash_id] = {"status": 0, "created_time": time.time(), "update_time": time.time(),
|
|
"result": {}, "data": data_input}
|
|
return hash_id
|
|
|
|
def get_info_session(self, hash_id: str):
|
|
if hash_id in self.session:
|
|
return self.session[hash_id]
|
|
return {"status": -2, "result": {}, "meta": {}}
|
|
|
|
def update_session(self, hash_id: str, result: dict, status: int):
|
|
if hash_id in self.session:
|
|
self.session[hash_id]["status"] = status
|
|
self.session[hash_id]["result"] = result
|
|
self.session[hash_id]["update_time"] = time.time()
|
|
return True
|
|
return False
|
|
|
|
def delete_session(self, hash_id: str):
|
|
if hash_id in self.session:
|
|
del self.session[hash_id]
|
|
return True
|
|
return False
|
|
|
