Skip to content

Commit

Permalink
更新一些抽卡的报错提示
Browse files Browse the repository at this point in the history
  • Loading branch information
@azmiao
azmiao committed Oct 19, 2023
1 parent 8418bb7 commit 771c6b5
Showing 1 changed file with 72 additions and 41 deletions.
113 changes: 72 additions & 41 deletions uma_gacha_v2/__init__.py
View file
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
import os
import base64
import os

from hoshino import Service, priv
from hoshino.util import DailyNumberLimiter, FreqLimiter

from .gacha import Gacha
from . update_init import auto_update
from .update_init import auto_update
from .util import get_pool, get_img_path, generate_img, random_comment, server_list, \
switch_server, switch_pool_id, get_pool_detail
from hoshino import Service, priv
from hoshino.util import DailyNumberLimiter, FreqLimiter
from ..plugin_utils.base_util import get_img_cq


Expand All @@ -19,14 +20,14 @@
single_limit = DailyNumberLimiter(30000)

# 抽井的次数(马娘池和支援卡池次数共通) | 默认每天15次
tenjo_limit = DailyNumberLimiter(15)
tenjou_limit = DailyNumberLimiter(15)

# 抽满破的次数 | 默认每天10次
full_limit = DailyNumberLimiter(10)

# 超过次数的消息
SINGLE_EXCEED_NOTICE = f'您今天已经抽过{single_limit.max}颗萝卜了，欢迎明早5点后再来哦！'
TENJO_EXCEED_NOTICE = f'您今天已经抽过{tenjo_limit.max}张天井券了，欢迎明早5点后再来哦！'
TENJOU_EXCEED_NOTICE = f'您今天已经抽过{tenjou_limit.max}张天井券了，欢迎明早5点后再来哦！'
FULL_EXCEED_NOTICE = f'您今天已经抽过{full_limit.max}次支援卡满破了，欢迎明早5点后再来哦！'

#############################
Expand All @@ -38,13 +39,15 @@
s = base64_data.decode()
sv.help = f'![](data:image/jpeg;base64,{s})'


# 帮助界面
@sv.on_fullmatch("马娘抽卡帮助")
async def help(bot, ev):
async def sv_help(bot, ev):
img_path = os.path.join(os.path.dirname(__file__), f'{sv.name}_help.png')
sv_help = await get_img_cq(img_path)
await bot.send(ev, sv_help)


# 马娘单抽
@sv.on_fullmatch(('马娘单抽', '单抽马娘'), only_to_me=True)
async def one_gacha_uma(bot, ev):
Expand All @@ -58,8 +61,9 @@ async def one_gacha_uma(bot, ev):
msg = await one_gacha(group_id, user_id, 'uma')
await bot.send(ev, msg)


# 支援卡单抽
@sv.on_fullmatch(('育成卡单抽', '支援卡单抽','s卡单抽','S卡单抽'), only_to_me=True)
@sv.on_fullmatch(('育成卡单抽', '支援卡单抽', 's卡单抽', 'S卡单抽'), only_to_me=True)
async def one_gacha_chart(bot, ev):
if not lmt.check(ev.user_id):
await bot.finish(ev, f'马娘抽卡功能冷却中(剩余 {int(lmt.left_time(ev.user_id)) + 1}秒)', at_sender=True)
Expand All @@ -71,6 +75,7 @@ async def one_gacha_chart(bot, ev):
msg = await one_gacha(group_id, user_id, 'chart')
await bot.send(ev, msg)


# 马娘十连
@sv.on_fullmatch(('马娘十连', '马十连'), only_to_me=True)
async def ten_gacha_uma(bot, ev):
Expand All @@ -84,8 +89,9 @@ async def ten_gacha_uma(bot, ev):
msg = await ten_gacha(group_id, user_id, 'uma')
await bot.send(ev, msg)


# 育成卡十连
@sv.on_fullmatch(('育成卡十连', '支援卡十连','s卡十连','S卡十连'), only_to_me=True)
@sv.on_fullmatch(('育成卡十连', '支援卡十连', 's卡十连', 'S卡十连'), only_to_me=True)
async def ten_gacha_chart(bot, ev):
if not lmt.check(ev.user_id):
await bot.finish(ev, f'马娘抽卡功能冷却中(剩余 {int(lmt.left_time(ev.user_id)) + 1}秒)', at_sender=True)
Expand All @@ -97,32 +103,36 @@ async def ten_gacha_chart(bot, ev):
msg = await ten_gacha(group_id, user_id, 'chart')
await bot.send(ev, msg)


# 马娘井
@sv.on_fullmatch(('马之井', '马娘井', '马娘一井'), only_to_me=True)
async def tenjou_gacha_uma(bot, ev):
if not lmt.check(ev.user_id):
await bot.finish(ev, f'马娘抽卡功能冷却中(剩余 {int(lmt.left_time(ev.user_id)) + 1}秒)', at_sender=True)
if not tenjo_limit.check(ev.user_id):
await bot.finish(ev, TENJO_EXCEED_NOTICE, at_sender=True)
if not tenjou_limit.check(ev.user_id):
await bot.finish(ev, TENJOU_EXCEED_NOTICE, at_sender=True)
lmt.start_cd(ev.user_id)
tenjo_limit.increase(ev.user_id)
tenjou_limit.increase(ev.user_id)
group_id, user_id = str(ev.group_id), str(ev.user_id)
msg = await tenjou_gacha(group_id, user_id, 'uma')
await bot.send(ev, msg)


# 育成卡井
@sv.on_fullmatch(('育成卡井', '育成卡一井', '支援卡井', '支援卡一井','s卡井','s卡一井','S卡井','S卡一井'), only_to_me=True)
@sv.on_fullmatch(('育成卡井', '育成卡一井', '支援卡井', '支援卡一井', 's卡井', 's卡一井', 'S卡井', 'S卡一井'),
only_to_me=True)
async def tenjou_gacha_chart(bot, ev):
if not lmt.check(ev.user_id):
await bot.finish(ev, f'马娘抽卡功能冷却中(剩余 {int(lmt.left_time(ev.user_id)) + 1}秒)', at_sender=True)
if not tenjo_limit.check(ev.user_id):
await bot.finish(ev, TENJO_EXCEED_NOTICE, at_sender=True)
if not tenjou_limit.check(ev.user_id):
await bot.finish(ev, TENJOU_EXCEED_NOTICE, at_sender=True)
lmt.start_cd(ev.user_id)
tenjo_limit.increase(ev.user_id)
tenjou_limit.increase(ev.user_id)
group_id, user_id = str(ev.group_id), str(ev.user_id)
msg = await tenjou_gacha(group_id, user_id, 'chart')
await bot.send(ev, msg)


# 育成卡抽满破
@sv.on_fullmatch(('育成卡抽满破', '支援卡抽满破'), only_to_me=True)
async def full_singer_gacha_chart(bot, ev):
Expand All @@ -136,6 +146,7 @@ async def full_singer_gacha_chart(bot, ev):
msg = await full_singer_gacha(group_id, user_id, 'chart')
await bot.send(ev, msg)


# 选择卡池
@sv.on_prefix('切换马娘服务器')
async def change_server(bot, ev):
Expand All @@ -148,6 +159,7 @@ async def change_server(bot, ev):
msg = await switch_server(group_id, server)
await bot.send(ev, msg)


# 选择卡池
@sv.on_prefix('切换马娘卡池')
async def change_pool(bot, ev):
Expand All @@ -158,13 +170,15 @@ async def change_pool(bot, ev):
msg = await switch_pool_id(group_id, pool_id)
await bot.send(ev, msg)


# 查看卡池
@sv.on_fullmatch('查看马娘卡池')
async def change_pool(bot, ev):
group_id = str(ev.group_id)
msg = await get_pool_detail(group_id)
await bot.send(ev, msg)


# 手动更新卡池
@sv.on_fullmatch('更新马娘卡池')
async def uma_gacha_update(bot, ev):
Expand All @@ -174,48 +188,65 @@ async def uma_gacha_update(bot, ev):
msg = await auto_update()
await bot.send(ev, msg)


async def one_gacha(group_id, user_id, gacha_type):
server, pool_id = await get_pool(group_id)
gacha = Gacha(pool_id, gacha_type, server)
chara, res_type = gacha.gacha_one(gacha.up_prob, gacha.s3_prob, gacha.s2_prob, gacha.s1_prob)
img_path = await get_img_path(chara, gacha_type)
img = await get_img_cq(img_path)
msg = f'{img}\n[CQ:at,qq={user_id}]\n抽到了 {chara}'
if res_type == 'up':
msg += '\nPS.兄弟姐妹们，有挂！'
elif res_type == 's3':
msg += '\nPS.这就是欧皇附体吗？'
try:
chara, res_type = gacha.gacha_one(gacha.up_prob, gacha.s3_prob, gacha.s2_prob, gacha.s1_prob)
img_path = await get_img_path(chara, gacha_type)
img = await get_img_cq(img_path)
msg = f'{img}\n[CQ:at,qq={user_id}]\n抽到了 {chara}'
if res_type == 'up':
msg += '\nPS.兄弟姐妹们，有挂！'
elif res_type == 's3':
msg += '\nPS.这就是欧皇附体吗？'
except IndexError as _:
msg = f'卡池数据还未自动更新完全，请耐心等待几小时后再尝试'
return msg


async def ten_gacha(group_id, user_id, gacha_type):
server, pool_id = await get_pool(group_id)
gacha = Gacha(pool_id, gacha_type, server)
first_up, result = gacha.gacha_ten(gacha.result, gacha.first_up)
result_list = result['up'] + result['s3'] + result['s2'] + result['s1']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '十连')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}'
try:
first_up, result = gacha.gacha_ten(gacha.result, gacha.first_up)
result_list = result['up'] + result['s3'] + result['s2'] + result['s1']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '十连')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}'
except IndexError as _:
msg = f'卡池数据还未自动更新完全，请耐心等待几小时后再尝试'
return msg


async def tenjou_gacha(group_id, user_id, gacha_type):
server, pool_id = await get_pool(group_id)
gacha = Gacha(pool_id, gacha_type, server)
first_up, result = gacha.gacha_tenjou(gacha.result, gacha.first_up)
result_list = result['up'] + result['s3']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '天井')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}'
try:
first_up, result = gacha.gacha_tenjou(gacha.result, gacha.first_up)
result_list = result['up'] + result['s3']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '天井')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}'
except IndexError as _:
msg = f'卡池数据还未自动更新完全，请耐心等待几小时后再尝试'
return msg


async def full_singer_gacha(group_id, user_id, gacha_type):
server, pool_id = await get_pool(group_id)
if pool_id == '00000000':
return '初始卡池00000000不支持该功能哦'
gacha = Gacha(pool_id, gacha_type, server)
select_chart, up_num, ten_num, exchange, first_up, result = gacha.gacha_full_singer(gacha.result, gacha.first_up)
result_list = result['up'] + result['s3']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '抽满破')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}\n'
msg += f'▼目标：{select_chart}\n最终获得{up_num}张\n其中兑换了{exchange}张\n总共花费{ten_num * 10}抽'
return msg
try:
select_chart, up_num, ten_num, exchange, first_up, result = gacha.gacha_full_singer(gacha.result,
gacha.first_up)
result_list = result['up'] + result['s3']
result_image = await generate_img(result_list, gacha_type)
msg_com = await random_comment(result, gacha_type, first_up, '抽满破')
msg = f'[CQ:image,file={result_image}]\n[CQ:at,qq={user_id}]\n{msg_com}\n'
msg += f'▼目标：{select_chart}\n最终获得{up_num}张\n其中兑换了{exchange}张\n总共花费{ten_num * 10}抽'
except IndexError as _:
msg = f'卡池数据还未自动更新完全，请耐心等待几小时后再尝试'
return msg

0 comments on commit 771c6b5

Please sign in to comment.