xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

Merge branch 'master' of https://github.com/Teahouse-Studios/akari-bot

Browse source
This commit is contained in:
yzhh 2023-06-02 15:48:04 +08:00
commit e7ba5ec35b
17 changed files with 44 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
modules/arcaea/__init__.py
View file

@ -51,10 +51,11 @@ async def _(msg: Bot.MessageSession, friend_code: int = None):
get_ = await get_url(api + f'user/bests/session?user_name={query_code}', headers=headers,
fmt='json')
if get_['status'] == 0:
await msg.sendMessage([Plain(msg.locale.t("arcaea.message.b30.wait")),
Plain(msg.locale.t("arcaea.message.sb616")),
Image(os.path.abspath('./assets/noc.jpg')),
Image(os.path.abspath('./assets/aof.jpg'))])
await msg.sendMessage(msg.locale.t("arcaea.message.b30.wait")),
if msg.target.targetFrom not in ['Discord|Channel', 'Telegram|group', 'Telegram|supergroup']:
await msg.sendMessage([Plain(msg.locale.t("arcaea.message.sb616")),
Image(os.path.abspath('./assets/noc.jpg')),
Image(os.path.abspath('./assets/aof.jpg'))])
elif get_['status'] == -33:
await msg.sendMessage(msg.locale.t("arcaea.message.b30.wait.cached"))
elif get_['status'] == -23:

2
modules/color/__init__.py
View file

@ -10,7 +10,7 @@ from PIL import Image, ImageDraw, ImageFont
from core.builtins import Bot, Image as BotImage
from core.component import module
c = module('color', alias=['colour'], developers=['Dianliang233',], desc='{color.help.desc}')
c = module('color', alias='colour', developers=['Dianliang233'], desc='{color.help.desc}')
font = ImageFont.truetype('assets/SourceHanSansCN-Normal.ttf', 40)

2
modules/convert/__init__.py
View file

@ -8,7 +8,7 @@ from core.component import module
# ureg = UnitRegistry(os.path.dirname(os.path.abspath(__file__)) +
# '/default_bi_zh-cn_en.txt', non_int_type=Decimal)
ureg = UnitRegistry(non_int_type=Decimal)
i = module('convert', alias=('conv', 'unit'), desc='{convert.help.desc}',
i = module('convert', alias=['conv', 'unit'], desc='{convert.help.desc}',
developers=['Dianliang233'], support_languages=['en_us'])

4
modules/core/su_utils.py
View file

@ -17,7 +17,7 @@ from core.utils.cache import random_cache_path
from database import BotDBUtil
from core.utils.storedata import get_stored_list, update_stored_list
su = module('superuser', alias=['su'], developers=['OasisAkari', 'Dianliang233'], required_superuser=True)
su = module('superuser', alias='su', developers=['OasisAkari', 'Dianliang233'], required_superuser=True)
@su.handle('add <user>')
@ -132,7 +132,7 @@ async def _(msg: Bot.MessageSession):
await msg.finish(msg.locale.t("core.message.set.help.option.tion.tion.tion.tion.tion.success", k=k, v=v))
ae = module('abuse', alias=['ae'], developers=['Dianliang233'], required_superuser=True)
ae = module('abuse', alias='ae', developers=['Dianliang233'], required_superuser=True)
@ae.handle('check <user>')

5
modules/core/utils.py
View file

@ -206,8 +206,7 @@ leave = module(
base=True,
required_admin=True,
available_for='QQ|Group',
alias={
'dismiss': 'leave'},
alias='dismiss',
desc='{core.help.leave}')
@ -219,7 +218,7 @@ async def _(msg: Bot.MessageSession):
await msg.call_api('set_group_leave', group_id=msg.session.target)
petal = module('petal', developers=['Dianliang233'], base=True, alias={'petals': 'petal'},
petal = module('petal', developers=['Dianliang233'], base=True, alias='petals',
desc='{core.help.petal}')

2
modules/dictionary/__init__.py
View file

@ -3,7 +3,7 @@ from core.component import module
from .screenshot import get_pic
dict_module = module('dictionary', alias=["dict"],
dict_module = module('dictionary', alias="dict",
desc='{dictionary.help.desc}', developers=['Dianliang233'], support_languages=['en_us'])

3
modules/exchange_rate/__init__.py
View file

@ -10,8 +10,7 @@ api_key = Config('exchange_rate_api_key')
excr = module('exchange_rate',
desc='{exchange_rate.help.desc}',
alias={'exchangerate': 'exchange_rate',
'excr': 'exchange_rate'},
alias=['exchangerate', 'excr'],
developers=['DoroWolf'])

2
modules/github/__init__.py
View file

@ -2,7 +2,7 @@ from core.builtins import Bot
from core.component import module
from modules.github import repo, user, search
github = module('github', alias=['gh'], developers=['Dianliang233'])
github = module('github', alias='gh', developers=['Dianliang233'])
@github.handle('<name> {{github.help}}')

6
modules/hitokoto/__init__.py
View file

@ -6,11 +6,9 @@ hitokoto_types = ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l"]
hitokoto = module(
'hitokoto',
developers=[
'bugungu',
'DoroWolf'],
developers=['bugungu','DoroWolf'],
desc='{hitokoto.help.desc}',
alias=['htkt'],
alias='htkt',
support_languages=['zh_cn'])

29
modules/maimai/__init__.py
View file

@ -12,6 +12,7 @@ from modules.maimai.libraries.tool import hash_
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 = ['綠', '黃', '紅']
@ -23,19 +24,25 @@ def song_txt(music: Music):
def get_label(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:
level = diff_label.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}', support_languages=['zh_cn'])
mai = module('maimai', developers=['mai-bot', 'OasisAkari', 'DoroWolf'], alias='mai',
desc='{maimai.help.desc}')
@ -45,7 +52,7 @@ async def _(msg: Bot.MessageSession):
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]} {elem[3]} {elem[4]} ({elem[2]})\n"
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:
@ -59,7 +66,7 @@ async def diff_level_q(level):
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]))
result_set.append((music['id'], music['title'], music['ds'][i], diff_label[i], music['level'][i], music['type']))
return result_set
@ -84,7 +91,7 @@ async def _(msg: Bot.MessageSession):
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]} {elem[3]} {elem[4]} ({elem[2]})\n"
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:
@ -103,7 +110,7 @@ async def inner_level_q(ds1, ds2=None):
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]))
result_set.append((music['id'], music['title'], music['ds'][i], diff_label[i], music['level'][i], music['type']))
return result_set
@ -121,7 +128,7 @@ async def _(msg: Bot.MessageSession):
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']}\n"
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:
@ -198,7 +205,7 @@ async def _(message: Bot.MessageSession):
msg = message.locale.t("maimai.message.song.dx", diff=diff_label[level_index], level=level, ds=ds,
tap=chart['notes'][0], hold=chart['notes'][1], slide=chart['notes'][2], touch=chart['notes'][3], _break=chart['notes'][4],
charter=chart['charter'])
await message.finish([Plain(f"{music['id']} {music['title']}\n"), BImage(f"{file}"), Plain(msg)])
await message.finish([Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), BImage(f"{file}"), Plain(msg)])
except Exception:
await message.finish(message.locale.t("maimai.message.chart_not_found"))
else:
@ -206,7 +213,7 @@ async def _(message: Bot.MessageSession):
music = (await total_list.get()).by_id(name)
try:
file = f"https://www.diving-fish.com/covers/{get_cover_len4_id(music['id'])}.png"
await message.finish([Plain(f"{music['id']} {music['title']}\n"),
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'],

4
modules/maimai/locales/en_us.json
View file

@ -16,8 +16,8 @@
"maimai.message.no_username": "请提供用户名！",
"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：${_break}\n谱师：${charter}",
"maimai.message.song.sd": "${diff} ${level} (${ds})\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.song.dx": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nTOUCH：${touch}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.song.sd": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.too_much": "结果过多（${length} 条），请缩小搜索范围。",
"maimai.message.user_not_found": "未找到此玩家，请确保此玩家的用户名和查分器中的用户名相同。"
}

4
modules/maimai/locales/zh_cn.json
View file

@ -16,8 +16,8 @@
"maimai.message.no_username": "请提供用户名！",
"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：${_break}\n谱师：${charter}",
"maimai.message.song.sd": "${diff} ${level} (${ds})\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.song.dx": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nTOUCH：${touch}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.song.sd": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n谱师：${charter}",
"maimai.message.too_much": "结果过多（${length} 条），请缩小搜索范围。",
"maimai.message.user_not_found": "未找到此玩家，请确保此玩家的用户名和查分器中的用户名相同。"
}

4
modules/maimai/locales/zh_tw.json
View file

@ -16,8 +16,8 @@
"maimai.message.no_username": "請提供使用者名稱！",
"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：${_break}\n譜師：${charter}",
"maimai.message.song.sd": "${diff} ${level} (${ds})\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n譜師：${charter}",
"maimai.message.song.dx": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nTOUCH：${touch}\nBREAK：${_break}\n譜師：${charter}",
"maimai.message.song.sd": "${diff} ${level}（${ds}）\nTAP：${tap}\nHOLD：${hold}\nSLIDE：${slide}\nBREAK：${_break}\n譜師：${charter}",
"maimai.message.too_much": "結果過多（${length} 條），請縮小搜尋範圍。",
"maimai.message.user_not_found": "未找到此使用者，請確認此使用者的名稱和查分器中的名稱相同。"
}

2
modules/meme/__init__.py
View file

@ -9,7 +9,7 @@ from modules.meme.urban import urban
meme = module(
bind_prefix='meme',
# well, people still use it though it only lived for an hour or so
alias=['nbnhhsh'],
alias='nbnhhsh',
desc='{meme.help.desc}',
developers=['Dianliang233'],
support_languages=['zh_cn', 'en_us'])

2
modules/natural/__init__.py
View file

@ -4,7 +4,7 @@ from config import Config
from core.builtins import Bot
from core.component import module
n = module('natural', alias=['nl2c'], developers=['Dianliang233'], desc='{natural.help}', required_superuser=True)
n = module('natural', alias='nl2c', developers=['Dianliang233'], desc='{natural.help}', required_superuser=True)
# Load your API key from an environment variable or secret management service
openai.api_key = Config('openai_api_key')

2
modules/user/__init__.py
View file

@ -5,7 +5,7 @@ from core.component import module
from modules.wiki.utils.dbutils import WikiTargetInfo
from .user import get_user_info
usr = module('user', alias=['u'],
usr = module('user', alias='u',
developers=['OasisAkari'])

3
modules/wolframalpha/__init__.py
View file

@ -11,8 +11,7 @@ client = wolframalpha.Client(Config('wolfram_alpha_appid'))
w = module(
'wolframalpha',
alias={
'wolfram': 'wolframalpha'},
alias='wolfram',
developers=['Dianliang233'],
desc='{wolframalpha.help.desc}',
support_languages=['en_us'])