修改配置文件,修改部分代码

This commit is contained in:
BadHappy 2023-12-30 17:27:02 +08:00
parent 1652fe4775
commit 6b7a6481a2
5 changed files with 53 additions and 41 deletions

View File

@ -1,5 +1,4 @@
private_key = "d76a25fea1d5c3c8ccfdb31179798f31f87a2afc7e8feda3f67d515790ee9ee5" # 与 icalingua 客户端使用的 private_key 一致 private_key = "" # 与 icalingua 客户端使用的 private_key 一致
host = "http://192.168.1.6:6789" # docker 版 icalingua 服务的地址 host = "" # docker 版 icalingua 服务的地址
self_id = 2163534756 # 机器人的 qq 号 self_id = 0 # 机器人的 qq 号

View File

@ -18,7 +18,7 @@ from module import hitokoto
# 功能包结束引用 # 功能包结束引用
def get_config() -> Tuple[str, str, int]: def get_config() -> Tuple[str, str, int]:
with open('config-temp.toml', 'r', encoding='utf-8') as f: with open('config.toml', 'r', encoding='utf-8') as f:
config = qtoml.load(f) config = qtoml.load(f)
return config['host'], config['private_key'], config['self_id'] return config['host'], config['private_key'], config['self_id']
@ -223,13 +223,17 @@ async def add_message(data: Dict[str, Any]):
elif content.startswith("!!hitokoto"): elif content.startswith("!!hitokoto"):
ctypet = content[len("!!hitokoto "):] ctypet = content[len("!!hitokoto "):]
if content == "!!hitokoto":
result_code, result_data = hitokoto.hitokoto("没有")
else:
result_code, result_data = hitokoto.hitokoto(ctypet) result_code, result_data = hitokoto.hitokoto(ctypet)
if result_code == 3: if result_code == 3:
message = Message(content=f"参数不对哦!",room_id=data['roomId']) message = Message(content=f"参数不对哦!",room_id=data['roomId'])
elif result_code == 2:
message = Message(content=f"发生了不可描述的错误X_X", room_id=data['roomId'])
elif result_code == 0: elif result_code == 0:
message = Message(f"Hitokoto: {result_data['hitokoto']}") message = Message(content=f"Hitokoto: {result_data['hitokoto']}", room_id=data['roomId'])
else: else:
message = Message(content=f"但你看到这条消息就代表有bug出炉", room_id=data['roomId']) message = Message(content=f"但你看到这条消息就代表有bug出炉", room_id=data['roomId'])

View File

@ -3,6 +3,8 @@ import random
def hitokoto(ctypet): def hitokoto(ctypet):
try:
if ctypet == '动画' or ctypet == 'a': if ctypet == '动画' or ctypet == 'a':
ctype = 'a' ctype = 'a'
elif ctypet == '漫画' or ctypet == 'b': elif ctypet == '漫画' or ctypet == 'b':
@ -27,8 +29,14 @@ def hitokoto(ctypet):
ctype = 'k' ctype = 'k'
elif ctypet == '抖机灵' or ctypet == 'l': elif ctypet == '抖机灵' or ctypet == 'l':
ctype = 'l' ctype = 'l'
else:
elif ctypet == '没有':
ctype = random.choice('abcdefghijkl') ctype = random.choice('abcdefghijkl')
params_value = {'c': ctype}
hitokoto = requests.get('https://v1.hitokoto.cn', params=params_value)
return 0, hitokoto.json()
else:
return 3, None return 3, None
params_value = {'c': ctype} params_value = {'c': ctype}
@ -36,8 +44,9 @@ def hitokoto(ctypet):
return 0, hitokoto.json() return 0, hitokoto.json()
# except Exception as e: except IOError:
# return 2, None return 2, None

Binary file not shown.

Binary file not shown.