2021-10-12 15:02:36 +00:00
|
|
|
|
import re
|
|
|
|
|
|
2023-02-05 14:33:33 +00:00
|
|
|
|
from core.builtins import Bot, Plain, Image as BImage
|
2023-03-04 08:51:56 +00:00
|
|
|
|
from core.component import module
|
2022-08-04 07:52:42 +00:00
|
|
|
|
from core.logger import Logger
|
2021-10-12 15:02:36 +00:00
|
|
|
|
from modules.maimai.libraries.image import *
|
|
|
|
|
from modules.maimai.libraries.maimai_best_40 import generate
|
2022-07-12 17:27:34 +00:00
|
|
|
|
from modules.maimai.libraries.maimai_best_50 import generate50
|
2021-11-12 14:25:53 +00:00
|
|
|
|
from modules.maimai.libraries.maimaidx_music import *
|
2023-03-04 08:51:56 +00:00
|
|
|
|
from modules.maimai.libraries.tool import hash_
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
2022-05-13 13:54:12 +00:00
|
|
|
|
total_list = TotalList()
|
|
|
|
|
|
|
|
|
|
|
2021-10-12 15:02:36 +00:00
|
|
|
|
def song_txt(music: Music):
|
|
|
|
|
return [Plain(f"{music.id}. {music.title}\n"),
|
2022-07-12 17:27:34 +00:00
|
|
|
|
BImage(f"https://www.diving-fish.com/covers/{get_cover_len4_id(music.id)}.png", ),
|
2021-10-12 15:02:36 +00:00
|
|
|
|
Plain(f"\n{'/'.join(music.level)}")]
|
|
|
|
|
|
|
|
|
|
|
2022-05-13 13:54:12 +00:00
|
|
|
|
async def inner_level_q(ds1, ds2=None):
|
2021-10-12 15:02:36 +00:00
|
|
|
|
result_set = []
|
|
|
|
|
diff_label = ['Bas', 'Adv', 'Exp', 'Mst', 'ReM']
|
|
|
|
|
if ds2 is not None:
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music_data = (await total_list.get()).filter(ds=(ds1, ds2))
|
2021-10-12 15:02:36 +00:00
|
|
|
|
else:
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music_data = (await total_list.get()).filter(ds=ds1)
|
2021-10-24 10:55:45 +00:00
|
|
|
|
for music in sorted(music_data, key=lambda i: int(i['id'])):
|
2021-10-12 15:02:36 +00:00
|
|
|
|
for i in music.diff:
|
|
|
|
|
result_set.append((music['id'], music['title'], music['ds'][i], diff_label[i], music['level'][i]))
|
|
|
|
|
return result_set
|
|
|
|
|
|
|
|
|
|
|
2023-03-04 08:51:56 +00:00
|
|
|
|
mai = module('maimai', developers=['mai-bot', 'OasisAkari'], alias=['mai'],
|
2023-04-30 03:30:59 +00:00
|
|
|
|
desc='有关maimai相关的工具,移植自mai-bot。')
|
2021-10-24 10:55:45 +00:00
|
|
|
|
|
|
|
|
|
|
2022-05-13 13:54:12 +00:00
|
|
|
|
@mai.handle(['inner <rating> {根据定数查询对应歌曲}',
|
|
|
|
|
'inner <rating_min> <rating_max> {根据定数查询对应歌曲}'])
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2022-08-08 15:31:38 +00:00
|
|
|
|
if '<rating>' in msg.parsed_msg:
|
2022-05-13 13:54:12 +00:00
|
|
|
|
result_set = await inner_level_q(float(msg.parsed_msg['<rating>']))
|
2021-10-12 15:02:36 +00:00
|
|
|
|
else:
|
2022-05-13 13:54:12 +00:00
|
|
|
|
result_set = await inner_level_q(float(msg.parsed_msg['<rating_min>']), float(msg.parsed_msg['<rating_max>']))
|
2021-10-12 15:02:36 +00:00
|
|
|
|
if len(result_set) > 50:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
return await msg.finish(f"结果过多({len(result_set)} 条),请缩小搜索范围。")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
s = ""
|
|
|
|
|
for elem in result_set:
|
|
|
|
|
s += f"{elem[0]}. {elem[1]} {elem[3]} {elem[4]}({elem[2]})\n"
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish(s.strip())
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
2023-03-04 08:51:56 +00:00
|
|
|
|
@mai.handle(re.compile(r"随个((?:dx|sd|标准))?([绿黄红紫白]?)([0-9]+\+?)"),
|
|
|
|
|
desc="随个[dx/标准][绿黄红紫白]<难度> 随机一首指定条件的乐曲")
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2021-10-13 15:22:11 +00:00
|
|
|
|
res = msg.matched_msg
|
2021-10-12 15:02:36 +00:00
|
|
|
|
if res:
|
|
|
|
|
try:
|
|
|
|
|
if res.groups()[0] == "dx":
|
|
|
|
|
tp = ["DX"]
|
|
|
|
|
elif res.groups()[0] == "sd" or res.groups()[0] == "标准":
|
|
|
|
|
tp = ["SD"]
|
|
|
|
|
else:
|
|
|
|
|
tp = ["SD", "DX"]
|
|
|
|
|
level = res.groups()[2]
|
|
|
|
|
if res.groups()[1] == "":
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music_data = (await total_list.get()).filter(level=level, type=tp)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
else:
|
2022-07-31 08:33:20 +00:00
|
|
|
|
music_data = (await total_list.get()).filter(level=level, diff=['绿黄红紫白'.index(res.groups()[1])],
|
|
|
|
|
type=tp)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
if len(music_data) == 0:
|
|
|
|
|
rand_result = "没有这样的乐曲哦。"
|
|
|
|
|
else:
|
|
|
|
|
rand_result = song_txt(music_data.random())
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish(rand_result)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
except Exception as e:
|
2022-08-04 07:52:42 +00:00
|
|
|
|
Logger.error(e)
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish("随机命令错误,请检查语法")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
2023-03-04 08:51:56 +00:00
|
|
|
|
@mai.handle(re.compile(r".*maimai.*什么"), desc='XXXmaimaiXXX什么 随机一首歌')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish(song_txt((await total_list.get()).random()))
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
2023-03-04 08:51:56 +00:00
|
|
|
|
@mai.handle(re.compile(r"查歌(.+)"), desc='查歌<乐曲标题的一部分> 查询符合条件的乐曲')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2021-10-13 15:22:11 +00:00
|
|
|
|
name = msg.matched_msg.groups()[0].strip()
|
|
|
|
|
if name == "":
|
|
|
|
|
return
|
2022-05-13 13:54:12 +00:00
|
|
|
|
res = (await total_list.get()).filter(title_search=name)
|
2021-10-13 15:22:11 +00:00
|
|
|
|
if len(res) == 0:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish("没有找到这样的乐曲。")
|
2021-10-13 15:22:11 +00:00
|
|
|
|
elif len(res) < 50:
|
|
|
|
|
search_result = ""
|
2021-10-24 10:55:45 +00:00
|
|
|
|
for music in sorted(res, key=lambda i: int(i['id'])):
|
2021-10-13 15:22:11 +00:00
|
|
|
|
search_result += f"{music['id']}. {music['title']}\n"
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish([Plain(search_result.strip())])
|
2021-10-13 15:22:11 +00:00
|
|
|
|
else:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish(f"结果过多({len(res)} 条),请缩小查询范围。")
|
2021-10-13 15:22:11 +00:00
|
|
|
|
|
|
|
|
|
|
2023-03-04 08:51:56 +00:00
|
|
|
|
@mai.handle(re.compile(r"([绿黄红紫白]?)id([0-9]+)"), desc='[绿黄红紫白]id<歌曲编号> 查询乐曲信息或谱面信息')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(message: Bot.MessageSession):
|
2021-10-13 15:22:11 +00:00
|
|
|
|
groups = message.matched_msg.groups()
|
|
|
|
|
level_labels = ['绿', '黄', '红', '紫', '白']
|
|
|
|
|
if groups[0] != "":
|
|
|
|
|
try:
|
|
|
|
|
level_index = level_labels.index(groups[0])
|
|
|
|
|
level_name = ['Basic', 'Advanced', 'Expert', 'Master', 'Re: MASTER']
|
2021-10-12 15:02:36 +00:00
|
|
|
|
name = groups[1]
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music = (await total_list.get()).by_id(name)
|
2021-10-13 15:22:11 +00:00
|
|
|
|
chart = music['charts'][level_index]
|
|
|
|
|
ds = music['ds'][level_index]
|
|
|
|
|
level = music['level'][level_index]
|
2022-07-12 17:27:34 +00:00
|
|
|
|
file = f"https://www.diving-fish.com/covers/{get_cover_len4_id(music['id'])}.png"
|
2021-10-13 15:22:11 +00:00
|
|
|
|
if len(chart['notes']) == 4:
|
|
|
|
|
msg = f'''{level_name[level_index]} {level}({ds})
|
|
|
|
|
TAP: {chart['notes'][0]}
|
|
|
|
|
HOLD: {chart['notes'][1]}
|
|
|
|
|
SLIDE: {chart['notes'][2]}
|
|
|
|
|
BREAK: {chart['notes'][3]}
|
|
|
|
|
谱师: {chart['charter']}'''
|
|
|
|
|
else:
|
|
|
|
|
msg = f'''{level_name[level_index]} {level}({ds})
|
|
|
|
|
TAP: {chart['notes'][0]}
|
|
|
|
|
HOLD: {chart['notes'][1]}
|
|
|
|
|
SLIDE: {chart['notes'][2]}
|
|
|
|
|
TOUCH: {chart['notes'][3]}
|
|
|
|
|
BREAK: {chart['notes'][4]}
|
|
|
|
|
谱师: {chart['charter']}'''
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await message.finish([Plain(f"{music['id']}. {music['title']}\n"), BImage(f"{file}"), Plain(msg)])
|
2021-10-13 15:22:11 +00:00
|
|
|
|
except Exception:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await message.finish("未找到该谱面")
|
2021-10-13 15:22:11 +00:00
|
|
|
|
else:
|
|
|
|
|
name = groups[1]
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music = (await total_list.get()).by_id(name)
|
2021-10-13 15:22:11 +00:00
|
|
|
|
try:
|
2022-07-12 17:27:34 +00:00
|
|
|
|
file = f"https://www.diving-fish.com/covers/{get_cover_len4_id(music['id'])}.png"
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await message.finish([Plain(f"{music['id']}. {music['title']}\n"),
|
2022-07-31 08:33:20 +00:00
|
|
|
|
BImage(f"{file}"),
|
|
|
|
|
Plain(f"艺术家: {music['basic_info']['artist']}\n"
|
|
|
|
|
f"分类: {music['basic_info']['genre']}\n"
|
|
|
|
|
f"BPM: {music['basic_info']['bpm']}\n"
|
|
|
|
|
f"版本: {music['basic_info']['from']}\n"
|
|
|
|
|
f"难度: {'/'.join(music['level'])}")])
|
2021-10-13 15:22:11 +00:00
|
|
|
|
except Exception:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await message.finish("未找到该乐曲")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
wm_list = ['拼机', '推分', '越级', '下埋', '夜勤', '练底力', '练手法', '打旧框', '干饭', '抓绝赞', '收歌']
|
|
|
|
|
|
2021-10-24 10:55:45 +00:00
|
|
|
|
|
2022-05-13 13:54:12 +00:00
|
|
|
|
@mai.handle('today {查看今天的舞萌运势}')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2022-03-14 10:22:21 +00:00
|
|
|
|
if msg.target.senderFrom == "Discord|Client":
|
2023-03-04 08:51:56 +00:00
|
|
|
|
qq = msg.session.sender.id
|
2022-03-14 10:22:21 +00:00
|
|
|
|
else:
|
2023-03-04 08:51:56 +00:00
|
|
|
|
qq = msg.session.sender
|
|
|
|
|
h = hash_(qq)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
rp = h % 100
|
|
|
|
|
wm_value = []
|
|
|
|
|
for i in range(11):
|
|
|
|
|
wm_value.append(h & 3)
|
|
|
|
|
h >>= 2
|
|
|
|
|
s = f"今日人品值:{rp}\n"
|
|
|
|
|
for i in range(11):
|
|
|
|
|
if wm_value[i] == 3:
|
|
|
|
|
s += f'宜 {wm_list[i]}\n'
|
|
|
|
|
elif wm_value[i] == 0:
|
|
|
|
|
s += f'忌 {wm_list[i]}\n'
|
|
|
|
|
s += "刘大鸽提醒您:打机时不要大力拍打或滑动哦\n今日推荐歌曲:"
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music = (await total_list.get())[h % len((await total_list.get()))]
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish([Plain(s)] + song_txt(music))
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
2022-05-13 13:54:12 +00:00
|
|
|
|
@mai.handle(['scoreline <difficulty+sid> <scoreline> {查找某首歌的分数线}',
|
|
|
|
|
'scoreline help {查看分数线帮助}'])
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2021-10-12 15:02:36 +00:00
|
|
|
|
r = "([绿黄红紫白])(id)?([0-9]+)"
|
2022-08-12 08:36:55 +00:00
|
|
|
|
arg1 = msg.parsed_msg.get('<difficulty+sid>')
|
|
|
|
|
args2 = msg.parsed_msg.get('<scoreline>')
|
|
|
|
|
argh = msg.parsed_msg.get('help', False)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
if argh:
|
|
|
|
|
s = '''此功能为查找某首歌分数线设计。
|
2022-05-13 13:54:12 +00:00
|
|
|
|
命令格式:maimai scoreline <difficulty+sid> <scoreline>
|
2021-10-12 15:02:36 +00:00
|
|
|
|
例如:分数线 紫799 100
|
|
|
|
|
命令将返回分数线允许的 TAP GREAT 容错以及 BREAK 50落等价的 TAP GREAT 数。
|
|
|
|
|
以下为 TAP GREAT 的对应表:
|
|
|
|
|
GREAT/GOOD/MISS
|
|
|
|
|
TAP\t1/2.5/5
|
|
|
|
|
HOLD\t2/5/10
|
|
|
|
|
SLIDE\t3/7.5/15
|
|
|
|
|
TOUCH\t1/2.5/5
|
|
|
|
|
BREAK\t5/12.5/25(外加200落)'''
|
2022-05-22 15:10:24 +00:00
|
|
|
|
img = text_to_image(s)
|
|
|
|
|
if img:
|
|
|
|
|
await msg.finish([BImage(img)])
|
2021-10-12 15:02:36 +00:00
|
|
|
|
elif args2 is not None:
|
|
|
|
|
try:
|
|
|
|
|
grp = re.match(r, arg1).groups()
|
|
|
|
|
level_labels = ['绿', '黄', '红', '紫', '白']
|
|
|
|
|
level_labels2 = ['Basic', 'Advanced', 'Expert', 'Master', 'Re:MASTER']
|
|
|
|
|
level_index = level_labels.index(grp[0])
|
|
|
|
|
chart_id = grp[2]
|
|
|
|
|
line = float(arg1)
|
2022-05-13 13:54:12 +00:00
|
|
|
|
music = (await total_list.get()).by_id(chart_id)
|
2021-10-12 15:02:36 +00:00
|
|
|
|
chart: Dict[Any] = music['charts'][level_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 - line
|
|
|
|
|
if reduce <= 0 or reduce >= 101:
|
|
|
|
|
raise ValueError
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish(f'''{music['title']} {level_labels2[level_index]}
|
2021-10-12 15:02:36 +00:00
|
|
|
|
分数线 {line}% 允许的最多 TAP GREAT 数量为 {(total_score * reduce / 10000):.2f}(每个-{10000 / total_score:.4f}%),
|
|
|
|
|
BREAK 50落(一共{brk}个)等价于 {(break_50_reduce / 100):.3f} 个 TAP GREAT(-{break_50_reduce / total_score * 100:.4f}%)''')
|
|
|
|
|
except Exception:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish("格式错误,输入“~maimai scoreline help”以查看帮助信息")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
|
|
|
|
|
|
2023-05-19 14:56:46 +00:00
|
|
|
|
@mai.handle('b40 [<username>] {查询B40信息(仅限大陆版maimai使用)}')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2023-05-19 14:56:46 +00:00
|
|
|
|
username = msg.parsed_msg.get('<username>', None)
|
2023-05-19 15:00:36 +00:00
|
|
|
|
if username is None and msg.target.senderFrom == "QQ":
|
2021-10-12 15:02:36 +00:00
|
|
|
|
payload = {'qq': msg.session.sender}
|
|
|
|
|
else:
|
2023-05-19 15:00:36 +00:00
|
|
|
|
if username is None:
|
|
|
|
|
await msg.finish("请提供用户名!")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
payload = {'username': username}
|
|
|
|
|
img, success = await generate(payload)
|
|
|
|
|
if success == 400:
|
2022-08-23 08:50:35 +00:00
|
|
|
|
await msg.finish("未找到此玩家,请确保此玩家的用户名和查分器中的用户名相同。"
|
|
|
|
|
"\n查分器地址:https://www.diving-fish.com/maimaidx/prober/")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
elif success == 403:
|
2022-05-21 16:04:29 +00:00
|
|
|
|
await msg.finish("该用户禁止了其他人获取数据。")
|
2021-10-12 15:02:36 +00:00
|
|
|
|
else:
|
2022-05-22 14:12:30 +00:00
|
|
|
|
if img:
|
|
|
|
|
await msg.finish([BImage(img)])
|
2022-07-12 17:27:34 +00:00
|
|
|
|
|
|
|
|
|
|
2023-05-19 14:56:46 +00:00
|
|
|
|
@mai.handle('b50 [<username>] {查询B50信息(仅限大陆版maimai使用)}')
|
2023-02-05 14:33:33 +00:00
|
|
|
|
async def _(msg: Bot.MessageSession):
|
2023-05-19 14:56:46 +00:00
|
|
|
|
username = msg.parsed_msg.get('<username>', None)
|
2023-05-19 15:00:36 +00:00
|
|
|
|
if username is None and msg.target.senderFrom == "QQ":
|
2023-01-28 05:53:11 +00:00
|
|
|
|
payload = {'qq': msg.session.sender, 'b50': True}
|
2022-07-12 17:27:34 +00:00
|
|
|
|
else:
|
2023-05-19 15:00:36 +00:00
|
|
|
|
if username is None:
|
|
|
|
|
await msg.finish("请提供用户名!")
|
2023-01-28 05:53:11 +00:00
|
|
|
|
payload = {'username': username, 'b50': True}
|
2022-07-12 17:27:34 +00:00
|
|
|
|
img, success = await generate50(payload)
|
|
|
|
|
if success == 400:
|
2022-08-23 08:50:35 +00:00
|
|
|
|
await msg.finish("未找到此玩家,请确保此玩家的用户名和查分器中的用户名相同。"
|
|
|
|
|
"\n查分器地址:https://www.diving-fish.com/maimaidx/prober/")
|
2022-07-12 17:27:34 +00:00
|
|
|
|
elif success == 403:
|
|
|
|
|
await msg.finish("该用户禁止了其他人获取数据。")
|
|
|
|
|
else:
|
|
|
|
|
if img:
|
|
|
|
|
await msg.finish([BImage(img)])
|