From a48b9375796f4f562a05edd2736bcd6a6e3220b8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=9A=E7=BE=85=E7=8B=BC?= Date: Sun, 18 Jun 2023 13:42:49 +0800 Subject: [PATCH] add mai_regex --- modules/maimai/__init__.py | 336 +----------------------------- modules/maimai/locales/zh_cn.json | 1 + modules/maimai/maimai.py | 334 +++++++++++++++++++++++++++++ modules/maimai/regex.py | 46 ++++ modules/mod_dl/__init__.py | 2 +- modules/mod_dl/locales/zh_cn.json | 1 + 6 files changed, 385 insertions(+), 335 deletions(-) create mode 100644 modules/maimai/maimai.py create mode 100644 modules/maimai/regex.py diff --git a/modules/maimai/__init__.py b/modules/maimai/__init__.py index c4331290..c0d436c9 100644 --- a/modules/maimai/__init__.py +++ b/modules/maimai/__init__.py @@ -1,334 +1,2 @@ -from core.builtins import Bot, Plain, Image as BImage -from core.builtins import command_prefix -from core.component import module -from core.logger import Logger -from core.utils.image import msgchain2image -from modules.maimai.libraries.maimai_best_50 import generate -from modules.maimai.libraries.maimaidx_api_data import get_alias, get_rank -from modules.maimai.libraries.maimaidx_music import * - -total_list = TotalList() - -diff_label = ['Basic', 'Advanced', 'Expert', 'Master', 'Re:MASTER'] -diff_label_abbr = ['bas', 'adv', 'exp', 'mas', 'rem'] -diff_label_zhs = ['绿', '黄', '红', '紫', '白'] -diff_label_zht = ['綠', '黃', '紅'] - - -def song_txt(music: Music): - return [Plain(f"{music.id} {music.title}{' (DX)' if music['type'] == 'DX' else ''}\n"), - BImage(f"https://www.diving-fish.com/covers/{get_cover_len5_id(music.id)}.png", ), - Plain(f"\n{'/'.join(str(ds) for ds in music.ds)}")] - - -def get_diff(diff): - diff = diff.lower() - diff_label_lower = [label.lower() for label in diff_label] - - if diff in diff_label_zhs: - level = diff_label_zhs.index(diff) - elif diff in diff_label_zht: - level = diff_label_zht.index(diff) - elif diff in diff_label_abbr: - level = diff_label_abbr.index(diff) - elif diff in diff_label_lower: - level = diff_label_lower.index(diff) - else: - level = None - return level - - -mai = module('maimai', developers=['mai-bot', 'OasisAkari', 'DoroWolf'], alias='mai', - desc='{maimai.help.desc}') - - -@mai.handle('level {{maimai.help.level}}') -async def _(msg: Bot.MessageSession, level: str): - result_set = await diff_level_q(level) - s = msg.locale.t("maimai.message.level", level=level) + "\n" - for elem in result_set: - s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n" - if len(result_set) == 0: - return await msg.finish(msg.locale.t("maimai.message.music_not_found")) - if len(result_set) <= 10: - await msg.finish(s.strip()) - else: - img = await msgchain2image([Plain(s)]) - await msg.finish([BImage(img)]) - - -async def diff_level_q(level): - result_set = [] - music_data = (await total_list.get()).filter(level=level) - for music in sorted(music_data, key=lambda i: int(i['id'])): - for i in music.diff: - result_set.append( - (music['id'], - music['title'], - music['ds'][i], - diff_label[i], - music['level'][i], - music['type'])) - return result_set - - -@mai.handle('inner [] {{maimai.help.inner}}') -async def _(msg: Bot.MessageSession, rating: float, rating_max: float = None): - if rating_max is not None: - if rating > rating_max: - return await msg.finish(msg.locale.t('error.range.invalid')) - result_set = await inner_level_q(rating, rating_max) - s = msg.locale.t("maimai.message.inner.range", rating=round(rating, 1), rating_max=round(rating_max, 1)) + "\n" - else: - result_set = await inner_level_q(rating) - s = msg.locale.t("maimai.message.inner", rating=round(rating, 1)) + "\n" - for elem in result_set: - s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n" - if len(result_set) == 0: - return await msg.finish(msg.locale.t("maimai.message.music_not_found")) - if len(result_set) > 200: - return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(result_set))) - if len(result_set) <= 10: - await msg.finish(s.strip()) - else: - img = await msgchain2image([Plain(s)]) - await msg.finish([BImage(img)]) - - -async def inner_level_q(ds1, ds2=None): - result_set = [] - if ds2 is not None: - music_data = (await total_list.get()).filter(ds=(ds1, ds2)) - else: - music_data = (await total_list.get()).filter(ds=ds1) - for music in sorted(music_data, key=lambda i: int(i['id'])): - for i in music.diff: - result_set.append( - (music['id'], - music['title'], - music['ds'][i], - diff_label[i], - music['level'][i], - music['type'])) - return result_set - - -@mai.handle('search {{maimai.help.search}}') -async def _(msg: Bot.MessageSession, keyword: str): - name = keyword.strip() - res = (await total_list.get()).filter(title_search=name) - if len(res) == 0: - return await msg.finish(msg.locale.t("maimai.message.music_not_found")) - if len(res) > 200: - return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(res))) - else: - search_result = msg.locale.t("maimai.message.search", keyword=name) + "\n" - for music in sorted(res, key=lambda i: int(i['id'])): - search_result += f"{music['id']} {music['title']}{' (DX)' if music['type'] == 'DX' else ''}\n" - if len(res) <= 10: - await msg.finish([Plain(search_result.strip())]) - else: - img = await msgchain2image([Plain(search_result)]) - await msg.finish([BImage(img)]) - - -@mai.handle('alias {{maimai.help.alias}}') -async def _(msg: Bot.MessageSession, sid: str): - if not sid.isdigit(): - await msg.finish(msg.locale.t('maimai.message.error.non_digital')) - music = (await total_list.get()).by_id(sid) - title = f"{music['title']}{' (DX)' if music['type'] == 'DX' else ''}" - alias = await get_alias(sid) - if len(alias) == 0: - return await msg.finish(msg.locale.t("maimai.message.alias_not_found")) - else: - result = msg.locale.t("maimai.message.alias", title=title) + "\n" - result += "\n".join(alias) - await msg.finish([Plain(result.strip())]) - - -@mai.handle('b40 [] {{maimai.help.b40}}') -async def _(msg: Bot.MessageSession, username: str = None): - if username is None and msg.target.senderFrom == "QQ": - payload = {'qq': msg.session.sender, 'b50': True} - else: - if username is None: - await msg.finish(msg.locale.t("maimai.message.no_username")) - payload = {'username': username, 'b50': True} - await msg.sendMessage(msg.locale.t("maimai.message.b40.deprecated")) - img, success = await generate(payload) - if success == 400: - await msg.finish(msg.locale.t("maimai.message.user_not_found")) - elif success == 403: - await msg.finish(msg.locale.t("maimai.message.forbidden")) - else: - if img: - await msg.finish([BImage(img)]) - - -@mai.handle('b50 [] {{maimai.help.b50}}') -async def _(msg: Bot.MessageSession, username: str = None): - if username is None and msg.target.senderFrom == "QQ": - payload = {'qq': msg.session.sender, 'b50': True} - else: - if username is None: - await msg.finish(msg.locale.t("maimai.message.no_username")) - payload = {'username': username, 'b50': True} - img, success = await generate(payload) - if success == 400: - await msg.finish(msg.locale.t("maimai.message.user_not_found")) - elif success == 403: - await msg.finish(msg.locale.t("maimai.message.forbidden")) - else: - if img: - await msg.finish([BImage(img)]) - - -@mai.handle('random [] {{maimai.help.random.filter}}') -async def _(msg: Bot.MessageSession, dx_type: str = None): - filter = msg.parsed_msg[''] - level = '' - diff = '' - try: - if dx_type in ["dx", "DX"]: - dx_type = ["DX"] - elif dx_type in ["sd", "SD", "标准", "標準"]: - dx_type = ["SD"] - else: - dx_type = ["SD", "DX"] - - for char in filter: - if char.isdigit() or char == '+': - level += char - else: - diff += char - - if level == "": - if diff == "#": - music_data = (await total_list.get()).filter(type=dx_type) - else: - raise ValueError - else: - if diff == "": - music_data = (await total_list.get()).filter(level=level, type=dx_type) - else: - music_data = (await total_list.get()).filter(level=level, diff=[get_diff(diff)], type=dx_type) - - if len(music_data) == 0: - rand_result = msg.locale.t("maimai.message.music_not_found") - else: - rand_result = song_txt(music_data.random()) - await msg.finish(rand_result) - except Exception as e: - Logger.error(e) - await msg.finish(msg.locale.t("maimai.message.random.error")) - - -@mai.handle('random {{maimai.help.random}}') -async def _(msg: Bot.MessageSession): - await msg.finish(song_txt((await total_list.get()).random())) - - -@mai.handle('song [] {{maimai.help.song}}') -async def _(message: Bot.MessageSession, sid: str, diff: str = None): - if not sid.isdigit(): - await message.finish(message.locale.t('maimai.message.error.non_digital')) - if diff is not None: - diff_index = get_diff(diff) - music = (await total_list.get()).by_id(sid) - if music is None: - await message.finish(message.locale.t("maimai.message.chart_not_found")) - chart = music['charts'][diff_index] - ds = music['ds'][diff_index] - level = music['level'][diff_index] - file = f"https://www.diving-fish.com/covers/{get_cover_len5_id(music['id'])}.png" - if len(chart['notes']) == 4: - msg = message.locale.t( - "maimai.message.song.sd", - diff=diff_label[diff_index], - level=level, - ds=ds, - tap=chart['notes'][0], - hold=chart['notes'][1], - slide=chart['notes'][2], - brk=chart['notes'][3], - charter=chart['charter']) - else: - msg = message.locale.t( - "maimai.message.song.dx", - diff=diff_label[diff_index], - level=level, - ds=ds, - tap=chart['notes'][0], - hold=chart['notes'][1], - slide=chart['notes'][2], - touch=chart['notes'][3], - brk=chart['notes'][4], - charter=chart['charter']) - await message.finish( - [Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), - BImage(f"{file}"), Plain(msg)]) - else: - music = (await total_list.get()).by_id(sid) - if music is None: - await message.finish(message.locale.t("maimai.message.music_not_found")) - file = f"https://www.diving-fish.com/covers/{get_cover_len5_id(music['id'])}.png" - await message.finish( - [Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), - BImage(f"{file}"), - Plain(message.locale.t("maimai.message.song", - artist=music['basic_info']['artist'], genre=music['basic_info']['genre'], - bpm=music['basic_info']['bpm'], version=music['basic_info']['from'], - level='/'.join((str(ds) for ds in music['ds']))))]) - - -@mai.handle('scoreline {{maimai.help.scoreline}}') -async def _(msg: Bot.MessageSession, diff: str, sid: str, scoreline: float): - try: - diff_index = get_diff(diff) - music = (await total_list.get()).by_id(sid) - chart: Dict[Any] = music['charts'][diff_index] - tap = int(chart['notes'][0]) - slide = int(chart['notes'][2]) - hold = int(chart['notes'][1]) - touch = int(chart['notes'][3]) if len(chart['notes']) == 5 else 0 - brk = int(chart['notes'][-1]) - total_score = 500 * tap + slide * 1500 + hold * 1000 + touch * 500 + brk * 2500 - break_bonus = 0.01 / brk - break_50_reduce = total_score * break_bonus / 4 - reduce = 101 - scoreline - if reduce <= 0 or reduce >= 101: - raise ValueError - tap_great = "{:.2f}".format(total_score * reduce / 10000) - tap_great_prop = "{:.4f}".format(10000 / total_score) - b2t_great = "{:.3f}".format(break_50_reduce / 100) - b2t_great_prop = "{:.4f}".format(break_50_reduce / total_score * 100) - await msg.finish(f'''{music['title']}{' (DX)' if music['type'] == 'DX' else ''} {diff_label[diff_index]} -{msg.locale.t('maimai.message.scoreline', - scoreline=scoreline, - tap_great=tap_great, - tap_great_prop=tap_great_prop, - brk=brk, - b2t_great=b2t_great, - b2t_great_prop=b2t_great_prop)}''') - except Exception: - await msg.finish(msg.locale.t('maimai.message.scoreline.error', prefix=command_prefix[0])) - - -@mai.handle('rank [] {{maimai.help.rank}}') -async def _(msg: Bot.MessageSession, username: str = None): - if username is None and msg.target.senderFrom == "QQ": - payload = {'qq': msg.session.sender} - else: - if username is None: - await msg.finish(msg.locale.t("maimai.message.no_username")) - payload = {'username': username} - - result = await get_rank(msg, payload) - time, total_rank, average_rating, username, rating, rank, surpassing_rate = result - formatted_average_rating = "{:.4f}".format(average_rating) - formatted_surpassing_rate = "{:.2f}".format(surpassing_rate) - - await msg.finish(msg.locale.t('maimai.message.rank', time=time, total_rank=total_rank, user=username, - rating=rating, rank=rank, average_rating=formatted_average_rating, - surpassing_rate=formatted_surpassing_rate)) +from .maimai import * +from .regex import * \ No newline at end of file diff --git a/modules/maimai/locales/zh_cn.json b/modules/maimai/locales/zh_cn.json index ae0537c7..cc909179 100644 --- a/modules/maimai/locales/zh_cn.json +++ b/modules/maimai/locales/zh_cn.json @@ -29,6 +29,7 @@ "maimai.message.search": "“${keyword}”的搜索结果：", "maimai.message.song": "艺术家：${artist}\n分类：${genre}\nBPM：${bpm}\n版本：${version}\n难度：${level}", "maimai.message.song.dx": "${diff} ${level} (${ds})\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nTOUCH：${touch}\nBREAK：${brk}\n谱师：${charter}", + "maimai.message.song.prompt": "你要找的是不是：", "maimai.message.song.sd": "${diff} ${level} (${ds})\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${brk}\n谱师：${charter}", "maimai.message.too_much": "结果过多（${length} 条），请缩小搜索范围。", "maimai.message.user_not_found": "未找到此玩家，请确保此玩家的用户名和查分器中的用户名相同。" diff --git a/modules/maimai/maimai.py b/modules/maimai/maimai.py new file mode 100644 index 00000000..75654f9b --- /dev/null +++ b/modules/maimai/maimai.py @@ -0,0 +1,334 @@ +from core.builtins import Bot, Plain, Image as BImage +from core.builtins import command_prefix +from core.component import module +from core.logger import Logger +from core.utils.image import msgchain2image +from modules.maimai.libraries.maimai_best_50 import generate +from modules.maimai.libraries.maimaidx_api_data import get_alias, get_rank +from modules.maimai.libraries.maimaidx_music import * + +total_list = TotalList() + +diff_label = ['Basic', 'Advanced', 'Expert', 'Master', 'Re:MASTER'] +diff_label_abbr = ['bas', 'adv', 'exp', 'mas', 'rem'] +diff_label_zhs = ['绿', '黄', '红', '紫', '白'] +diff_label_zht = ['綠', '黃', '紅'] + + +def song_txt(music: Music): + return [Plain(f"{music.id} {music.title}{' (DX)' if music['type'] == 'DX' else ''}\n"), + BImage(f"https://www.diving-fish.com/covers/{get_cover_len5_id(music.id)}.png", ), + Plain(f"\n{'/'.join(str(ds) for ds in music.ds)}")] + + +def get_diff(diff): + diff = diff.lower() + diff_label_lower = [label.lower() for label in diff_label] + + if diff in diff_label_zhs: + level = diff_label_zhs.index(diff) + elif diff in diff_label_zht: + level = diff_label_zht.index(diff) + elif diff in diff_label_abbr: + level = diff_label_abbr.index(diff) + elif diff in diff_label_lower: + level = diff_label_lower.index(diff) + else: + level = None + return level + + +mai = module('maimai', developers=['mai-bot', 'OasisAkari', 'DoroWolf'], alias='mai', + desc='{maimai.help.desc}') + + +@mai.handle('level {{maimai.help.level}}') +async def _(msg: Bot.MessageSession, level: str): + result_set = await diff_level_q(level) + s = msg.locale.t("maimai.message.level", level=level) + "\n" + for elem in result_set: + s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n" + if len(result_set) == 0: + return await msg.finish(msg.locale.t("maimai.message.music_not_found")) + if len(result_set) <= 10: + await msg.finish(s.strip()) + else: + img = await msgchain2image([Plain(s)]) + await msg.finish([BImage(img)]) + + +async def diff_level_q(level): + result_set = [] + music_data = (await total_list.get()).filter(level=level) + for music in sorted(music_data, key=lambda i: int(i['id'])): + for i in music.diff: + result_set.append( + (music['id'], + music['title'], + music['ds'][i], + diff_label[i], + music['level'][i], + music['type'])) + return result_set + + +@mai.handle('inner [] {{maimai.help.inner}}') +async def _(msg: Bot.MessageSession, rating: float, rating_max: float = None): + if rating_max is not None: + if rating > rating_max: + return await msg.finish(msg.locale.t('error.range.invalid')) + result_set = await inner_level_q(rating, rating_max) + s = msg.locale.t("maimai.message.inner.range", rating=round(rating, 1), rating_max=round(rating_max, 1)) + "\n" + else: + result_set = await inner_level_q(rating) + s = msg.locale.t("maimai.message.inner", rating=round(rating, 1)) + "\n" + for elem in result_set: + s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n" + if len(result_set) == 0: + return await msg.finish(msg.locale.t("maimai.message.music_not_found")) + if len(result_set) > 200: + return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(result_set))) + if len(result_set) <= 10: + await msg.finish(s.strip()) + else: + img = await msgchain2image([Plain(s)]) + await msg.finish([BImage(img)]) + + +async def inner_level_q(ds1, ds2=None): + result_set = [] + if ds2 is not None: + music_data = (await total_list.get()).filter(ds=(ds1, ds2)) + else: + music_data = (await total_list.get()).filter(ds=ds1) + for music in sorted(music_data, key=lambda i: int(i['id'])): + for i in music.diff: + result_set.append( + (music['id'], + music['title'], + music['ds'][i], + diff_label[i], + music['level'][i], + music['type'])) + return result_set + + +@mai.handle('search {{maimai.help.search}}') +async def _(msg: Bot.MessageSession, keyword: str): + name = keyword.strip() + res = (await total_list.get()).filter(title_search=name) + if len(res) == 0: + return await msg.finish(msg.locale.t("maimai.message.music_not_found")) + if len(res) > 200: + return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(res))) + else: + search_result = msg.locale.t("maimai.message.search", keyword=name) + "\n" + for music in sorted(res, key=lambda i: int(i['id'])): + search_result += f"{music['id']} {music['title']}{' (DX)' if music['type'] == 'DX' else ''}\n" + if len(res) <= 10: + await msg.finish([Plain(search_result.strip())]) + else: + img = await msgchain2image([Plain(search_result)]) + await msg.finish([BImage(img)]) + + +@mai.handle('alias {{maimai.help.alias}}') +async def _(msg: Bot.MessageSession, sid: str): + if not sid.isdigit(): + await msg.finish(msg.locale.t('maimai.message.error.non_digital')) + music = (await total_list.get()).by_id(sid) + title = f"{music['title']}{' (DX)' if music['type'] == 'DX' else ''}" + alias = await get_alias(sid) + if len(alias) == 0: + return await msg.finish(msg.locale.t("maimai.message.alias_not_found")) + else: + result = msg.locale.t("maimai.message.alias", title=title) + "\n" + result += "\n".join(alias) + await msg.finish([Plain(result.strip())]) + + +@mai.handle('b40 [] {{maimai.help.b40}}') +async def _(msg: Bot.MessageSession, username: str = None): + if username is None and msg.target.senderFrom == "QQ": + payload = {'qq': msg.session.sender, 'b50': True} + else: + if username is None: + await msg.finish(msg.locale.t("maimai.message.no_username")) + payload = {'username': username, 'b50': True} + await msg.sendMessage(msg.locale.t("maimai.message.b40.deprecated")) + img, success = await generate(payload) + if success == 400: + await msg.finish(msg.locale.t("maimai.message.user_not_found")) + elif success == 403: + await msg.finish(msg.locale.t("maimai.message.forbidden")) + else: + if img: + await msg.finish([BImage(img)]) + + +@mai.handle('b50 [] {{maimai.help.b50}}') +async def _(msg: Bot.MessageSession, username: str = None): + if username is None and msg.target.senderFrom == "QQ": + payload = {'qq': msg.session.sender, 'b50': True} + else: + if username is None: + await msg.finish(msg.locale.t("maimai.message.no_username")) + payload = {'username': username, 'b50': True} + img, success = await generate(payload) + if success == 400: + await msg.finish(msg.locale.t("maimai.message.user_not_found")) + elif success == 403: + await msg.finish(msg.locale.t("maimai.message.forbidden")) + else: + if img: + await msg.finish([BImage(img)]) + + +@mai.handle('rank [] {{maimai.help.rank}}') +async def _(msg: Bot.MessageSession, username: str = None): + if username is None and msg.target.senderFrom == "QQ": + payload = {'qq': msg.session.sender} + else: + if username is None: + await msg.finish(msg.locale.t("maimai.message.no_username")) + payload = {'username': username} + + result = await get_rank(msg, payload) + time, total_rank, average_rating, username, rating, rank, surpassing_rate = result + formatted_average_rating = "{:.4f}".format(average_rating) + formatted_surpassing_rate = "{:.2f}".format(surpassing_rate) + + await msg.finish(msg.locale.t('maimai.message.rank', time=time, total_rank=total_rank, user=username, + rating=rating, rank=rank, average_rating=formatted_average_rating, + surpassing_rate=formatted_surpassing_rate)) + + +@mai.handle('random [] {{maimai.help.random.filter}}') +async def _(msg: Bot.MessageSession, dx_type: str = None): + filter = msg.parsed_msg[''] + level = '' + diff = '' + try: + if dx_type in ["dx", "DX"]: + dx_type = ["DX"] + elif dx_type in ["sd", "SD", "标准", "標準"]: + dx_type = ["SD"] + else: + dx_type = ["SD", "DX"] + + for char in filter: + if char.isdigit() or char == '+': + level += char + else: + diff += char + + if level == "": + if diff == "#": + music_data = (await total_list.get()).filter(type=dx_type) + else: + raise ValueError + else: + if diff == "": + music_data = (await total_list.get()).filter(level=level, type=dx_type) + else: + music_data = (await total_list.get()).filter(level=level, diff=[get_diff(diff)], type=dx_type) + + if len(music_data) == 0: + rand_result = msg.locale.t("maimai.message.music_not_found") + else: + rand_result = song_txt(music_data.random()) + await msg.finish(rand_result) + except Exception as e: + Logger.error(e) + await msg.finish(msg.locale.t("maimai.message.random.error")) + + +@mai.handle('random {{maimai.help.random}}') +async def _(msg: Bot.MessageSession): + await msg.finish(song_txt((await total_list.get()).random())) + + +@mai.handle('song [] {{maimai.help.song}}') +async def _(msg: Bot.MessageSession, sid: str, diff: str = None): + if not sid.isdigit(): + await msg.finish(msg.locale.t('maimai.message.error.non_digital')) + if diff is not None: + diff_index = get_diff(diff) + music = (await total_list.get()).by_id(sid) + if music is None: + await msg.finish(msg.locale.t("maimai.message.chart_not_found")) + chart = music['charts'][diff_index] + ds = music['ds'][diff_index] + level = music['level'][diff_index] + file = f"https://www.diving-fish.com/covers/{get_cover_len5_id(music['id'])}.png" + if len(chart['notes']) == 4: + msg = msg.locale.t( + "maimai.message.song.sd", + diff=diff_label[diff_index], + level=level, + ds=ds, + tap=chart['notes'][0], + hold=chart['notes'][1], + slide=chart['notes'][2], + brk=chart['notes'][3], + charter=chart['charter']) + else: + msg = msg.locale.t( + "maimai.message.song.dx", + diff=diff_label[diff_index], + level=level, + ds=ds, + tap=chart['notes'][0], + hold=chart['notes'][1], + slide=chart['notes'][2], + touch=chart['notes'][3], + brk=chart['notes'][4], + charter=chart['charter']) + await msg.finish( + [Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), + BImage(f"{file}"), Plain(msg)]) + else: + music = (await total_list.get()).by_id(sid) + if music is None: + await msg.finish(msg.locale.t("maimai.message.music_not_found")) + file = f"https://www.diving-fish.com/covers/{get_cover_len5_id(music['id'])}.png" + await msg.finish( + [Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), + BImage(f"{file}"), + Plain(msg.locale.t("maimai.message.song", + artist=music['basic_info']['artist'], genre=music['basic_info']['genre'], + bpm=music['basic_info']['bpm'], version=music['basic_info']['from'], + level='/'.join((str(ds) for ds in music['ds']))))]) + + +@mai.handle('scoreline {{maimai.help.scoreline}}') +async def _(msg: Bot.MessageSession, diff: str, sid: str, scoreline: float): + try: + diff_index = get_diff(diff) + music = (await total_list.get()).by_id(sid) + chart: Dict[Any] = music['charts'][diff_index] + tap = int(chart['notes'][0]) + slide = int(chart['notes'][2]) + hold = int(chart['notes'][1]) + touch = int(chart['notes'][3]) if len(chart['notes']) == 5 else 0 + brk = int(chart['notes'][-1]) + total_score = 500 * tap + slide * 1500 + hold * 1000 + touch * 500 + brk * 2500 + break_bonus = 0.01 / brk + break_50_reduce = total_score * break_bonus / 4 + reduce = 101 - scoreline + if reduce <= 0 or reduce >= 101: + raise ValueError + tap_great = "{:.2f}".format(total_score * reduce / 10000) + tap_great_prop = "{:.4f}".format(10000 / total_score) + b2t_great = "{:.3f}".format(break_50_reduce / 100) + b2t_great_prop = "{:.4f}".format(break_50_reduce / total_score * 100) + await msg.finish(f'''{music['title']}{' (DX)' if music['type'] == 'DX' else ''} {diff_label[diff_index]} +{msg.locale.t('maimai.message.scoreline', + scoreline=scoreline, + tap_great=tap_great, + tap_great_prop=tap_great_prop, + brk=brk, + b2t_great=b2t_great, + b2t_great_prop=b2t_great_prop)}''') + except Exception: + await msg.finish(msg.locale.t('maimai.message.scoreline.error', prefix=command_prefix[0])) diff --git a/modules/maimai/regex.py b/modules/maimai/regex.py new file mode 100644 index 00000000..62567a17 --- /dev/null +++ b/modules/maimai/regex.py @@ -0,0 +1,46 @@ +import re + +from core.builtins import Bot, Plain, Image as BImage +from core.component import module +from modules.maimai.libraries.maimaidx_api_data import get_alias +from modules.maimai.libraries.maimaidx_music import * + +total_list = TotalList() + +mai_regex = module('maimai_regex', + desc='{wiki.help.maimai_regex.desc}', + alias='maimai_regex', developers=['DoroWolf']) + + +@mai_regex.handle(re.compile(r"(.+)是什么歌"), desc='查歌<乐曲标题的一部分> 查询符合条件的乐曲') +async def _(msg: Bot.MessageSession): + name = msg.matched_msg.groups()[0].strip() + if name == "": + return + elif name.isdigit(): + music = (await total_list.get()).by_id(name) + if music is None: + await msg.finish(msg.locale.t("maimai.message.music_not_found")) + else: + sid_list = await get_alias(name, get_music=True) + if len(sid_list) == 0: + await msg.finish(msg.locale.t("maimai.message.music_not_found")) + elif len(sid_list) > 1: + res = msg.locale.t("maimai.message.song.prompt") + "\n" + for sid in sorted(sid_list, key=int): + s = (await total_list.get()).by_id(sid) + res += f"{s['id']} {s['title']}{' (DX)' if s['type'] == 'DX' else ''}\n" + + reply = await msg.waitReply(res) + music = (await total_list.get()).by_id(str(reply)) + else: + music = (await total_list.get()).by_id(str(sid_list[0])) + + file = f"https://www.diving-fish.com/covers/{get_cover_len5_id(music['id'])}.png" + await msg.finish( + [Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), + BImage(f"{file}"), + Plain(msg.locale.t("maimai.message.song", + artist=music['basic_info']['artist'], genre=music['basic_info']['genre'], + bpm=music['basic_info']['bpm'], version=music['basic_info']['from'], + level='/'.join((str(ds) for ds in music['ds']))))]) \ No newline at end of file diff --git a/modules/mod_dl/__init__.py b/modules/mod_dl/__init__.py index dd7d1cec..baa2892e 100644 --- a/modules/mod_dl/__init__.py +++ b/modules/mod_dl/__init__.py @@ -37,7 +37,7 @@ async def main(msg: Bot.MessageSession, mod_name: str, version: str = None): mod_name += ' ' + version ver = False if cn_chk(mod_name): - return {'msg': '暂不支持中文搜索。', 'success': False} + await msg.finish(msg.locale.t("mod_dl.message.unsupport")) async def search_modrinth(name: str, ver: str): url = f'https://api.modrinth.com/v2/search?query={name}&limit=10' diff --git a/modules/mod_dl/locales/zh_cn.json b/modules/mod_dl/locales/zh_cn.json index 70b6a91d..1c36a83f 100644 --- a/modules/mod_dl/locales/zh_cn.json +++ b/modules/mod_dl/locales/zh_cn.json @@ -11,6 +11,7 @@ "mod_dl.message.modrinth.result": "Modrinth 结果：", "mod_dl.message.not_found": "未找到结果。", "mod_dl.message.prompt": "请回复编号来选择 Mod。", + "mod_dl.message.unsupport": "不支持中文搜索，请检查输入。", "mod_dl.message.version": "此 Mod 拥有以下版本：", "mod_dl.message.version.not_found": "未找到指定版本。", "mod_dl.message.version.prompt": "请回复版本号来选择版本。"