mirror of
https://github.com/baiqwerdvd/StarRailUID.git
synced 2025-06-19 05:45:03 +08:00
96 lines
2.8 KiB
Python
96 lines
2.8 KiB
Python
import re
|
|
|
|
from gsuid_core.bot import Bot
|
|
from gsuid_core.models import Event
|
|
from gsuid_core.sv import SV
|
|
from gsuid_core.utils.error_reply import UID_HINT
|
|
|
|
from ..utils.convert import get_uid
|
|
from ..utils.sr_prefix import PREFIX
|
|
from .draw_rogue_card import draw_rogue_img, draw_rogue_locust_img
|
|
|
|
sv_srabyss = SV('sr查询模拟宇宙')
|
|
sv_srabyss_locust = SV('sr查询寰宇蝗灾')
|
|
|
|
|
|
@sv_srabyss.on_command(
|
|
(
|
|
f'{PREFIX}查询宇宙',
|
|
f'{PREFIX}yz',
|
|
f'{PREFIX}查询上期宇宙',
|
|
f'{PREFIX}sqyz',
|
|
f'{PREFIX}上期宇宙',
|
|
f'{PREFIX}宇宙',
|
|
f'{PREFIX}查询模拟宇宙',
|
|
f'{PREFIX}上期模拟宇宙',
|
|
f'{PREFIX}查询上期模拟宇宙',
|
|
),
|
|
block=True,
|
|
)
|
|
async def send_srabyss_info(bot: Bot, ev: Event):
|
|
name = ''.join(re.findall('[\u4e00-\u9fa5]', ev.text))
|
|
if name:
|
|
return None
|
|
|
|
await bot.logger.info('开始执行[sr查询模拟宇宙信息]')
|
|
get_uid_ = await get_uid(bot, ev, True)
|
|
if get_uid_ is None:
|
|
return await bot.send(UID_HINT)
|
|
uid, user_id = get_uid_
|
|
if uid is None:
|
|
return await bot.send(UID_HINT)
|
|
await bot.logger.info(f'[sr查询模拟宇宙信息]uid: {uid}')
|
|
|
|
if 'sq' in ev.command or '上期' in ev.command:
|
|
schedule_type = '2'
|
|
else:
|
|
schedule_type = '3'
|
|
await bot.logger.info(f'[sr查询模拟宇宙信息]模拟宇宙期数: {schedule_type}')
|
|
|
|
if ev.text in ['一', '二', '三', '四', '五', '六']:
|
|
floor = (
|
|
ev.text.replace('一', '1')
|
|
.replace('二', '2')
|
|
.replace('三', '3')
|
|
.replace('四', '4')
|
|
.replace('五', '5')
|
|
.replace('六', '6')
|
|
)
|
|
else:
|
|
floor = ev.text
|
|
if floor and floor.isdigit():
|
|
floor = int(floor)
|
|
else:
|
|
floor = None
|
|
await bot.logger.info(f'[sr查询模拟宇宙信息]模拟宇宙世界数: {floor}')
|
|
im = await draw_rogue_img(user_id, uid, floor, schedule_type)
|
|
await bot.send(im)
|
|
return None
|
|
|
|
|
|
@sv_srabyss_locust.on_command(
|
|
(
|
|
f'{PREFIX}寰宇蝗灾',
|
|
f'{PREFIX}hyhz',
|
|
f'{PREFIX}查询寰宇蝗灾',
|
|
f'{PREFIX}sqhyhz',
|
|
),
|
|
block=True,
|
|
)
|
|
async def send_srabyss_locust_info(bot: Bot, ev: Event):
|
|
name = ''.join(re.findall('[\u4e00-\u9fa5]', ev.text))
|
|
if name:
|
|
return None
|
|
|
|
await bot.logger.info('开始执行[sr查询寰宇蝗灾信息]')
|
|
get_uid_ = await get_uid(bot, ev, True)
|
|
if get_uid_ is None:
|
|
return await bot.send(UID_HINT)
|
|
uid, user_id = get_uid_
|
|
if uid is None:
|
|
return await bot.send(UID_HINT)
|
|
await bot.logger.info(f'[sr查询寰宇蝗灾信息]uid: {uid}')
|
|
im = await draw_rogue_locust_img(user_id, uid)
|
|
await bot.send(im)
|
|
return None
|