xuthus/LittlePaimon
1
0
Fork 0
mirror of https://github.com/xuthus83/LittlePaimon.git synced 2024-10-21 16:27:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix bug

Browse Source
This commit is contained in:
CMHopeSunshine 2022-08-17 16:38:55 +08:00
parent 6cd5bce0b1
commit bb46af8f57
1 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
utils/alias_handler.py
View File

@ -22,7 +22,7 @@ def get_id_by_name(name: str):
:return: id字符串 :return: id字符串
""" """
alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json') alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json')
name_list = alias_file['roles'] name_list = alias_file['角色']
for role_id, alias in name_list.items(): for role_id, alias in name_list.items():
if name in alias: if name in alias:
return role_id return role_id
@ -35,11 +35,8 @@ def get_name_by_id(role_id: str):
:return: 角色名字符串 :return: 角色名字符串
""" """
alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json') alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json')
name_list = alias_file['roles'] name_list = alias_file['角色']
if role_id in name_list: return name_list[role_id][0] if role_id in name_list else None
return name_list[role_id][0]
else:
return None
def get_alias_by_name(name: str): def get_alias_by_name(name: str):
@ -49,11 +46,8 @@ def get_alias_by_name(name: str):
:return: 别名列表 :return: 别名列表
""" """
alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json') alias_file = load_json(path=Path(__file__).parent / 'json_data' / 'alias.json')
name_list = alias_file['roles'] name_list = alias_file['角色']
for r in name_list.values(): return next((r for r in name_list.values() if name in r), None)
if name in r:
return r
return None
def get_match_alias(msg: str, type: str = '角色', single_to_dict: bool = False) -> Union[str, list, dict]: def get_match_alias(msg: str, type: str = '角色', single_to_dict: bool = False) -> Union[str, list, dict]:
@ -78,8 +72,9 @@ def get_match_alias(msg: str, type: str = '角色', single_to_dict: bool = False
possible[alias[0]] = role_id possible[alias[0]] = role_id
if len(possible) == 1: if len(possible) == 1:
return {list(possible.keys())[0]: possible[list(possible.keys())[0]]} if single_to_dict else list(possible.keys())[0] return {list(possible.keys())[0]: possible[list(possible.keys())[0]]} if single_to_dict else list(possible.keys())[0]
return possible return possible
elif type in ['武器', '圣遗物']: elif type in {'武器', '圣遗物'}:
possible = [] possible = []
for name, alias in alias_list.items(): for name, alias in alias_list.items():
match_list = difflib.get_close_matches(msg, alias, cutoff=0.4, n=3) match_list = difflib.get_close_matches(msg, alias, cutoff=0.4, n=3)