refactor code

This commit is contained in:
hibiki
2021-07-30 21:21:51 +08:00
parent 2ad891aa1e
commit cc24822dca
165 changed files with 7815 additions and 8174 deletions
+73 -70
View File
@@ -7,28 +7,34 @@ from nonebot.permission import SUPERUSER
import random
from nonebot.plugin import MatcherGroup
import re
from .open_cases_c import open_case, total_open_statistics, group_statistics, my_knifes_name, open_shilian_case
from .open_cases_c import (
open_case,
total_open_statistics,
group_statistics,
my_knifes_name,
open_shilian_case,
)
from .utils import util_get_buff_price, util_get_buff_img, update_count_daily
__plugin_name__ = '开箱'
__plugin_name__ = "开箱"
__plugin_usage__ = (
'用法:\n'
'看看你的人品罢了\n'
'目前只支持\n\t'
'1.狂牙大行动武器箱\n\t'
'2.突围大行动武器箱\n\t'
'3.命悬一线武器箱\n\t'
'4.裂空武器箱\n\t'
'5.光谱武器箱\n'
'示例:小真寻开箱 突围大行动(不输入指定武器箱则随机)\n'
'示例:我的开箱(开箱统计)\n'
'示例:群开箱统计\n'
'示例:我的金色'
"用法:\n"
"看看你的人品罢了\n"
"目前只支持\n\t"
"1.狂牙大行动武器箱\n\t"
"2.突围大行动武器箱\n\t"
"3.命悬一线武器箱\n\t"
"4.裂空武器箱\n\t"
"5.光谱武器箱\n"
"示例:小真寻开箱 突围大行动(不输入指定武器箱则随机)\n"
"示例:我的开箱(开箱统计)\n"
"示例:群开箱统计\n"
"示例:我的金色"
)
_flmt = FreqLimiter(3)
cases_name = ['狂牙大行动', '突围大行动', '命悬一线', '裂空', '光谱']
cases_name = ["狂牙大行动", "突围大行动", "命悬一线", "裂空", "光谱"]
cases_matcher_group = MatcherGroup(priority=5, permission=GROUP, block=True)
@@ -38,20 +44,24 @@ k_open_case = cases_matcher_group.on_command("开箱")
@k_open_case.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
if str(event.get_message()).strip() in ['帮助']:
if str(event.get_message()).strip() in ["帮助"]:
await k_open_case.finish(__plugin_usage__)
if not _flmt.check(event.user_id):
await k_open_case.finish('着什么急啊,慢慢来!', at_sender=True)
await k_open_case.finish("着什么急啊,慢慢来!", at_sender=True)
_flmt.start_cd(event.user_id)
case_name = get_message_text(event.json())
if case_name:
result = await open_case(event.user_id, event.group_id, case_name)
else:
result = await open_case(event.user_id, event.group_id, random.choice(cases_name))
result = await open_case(
event.user_id, event.group_id, random.choice(cases_name)
)
await k_open_case.finish(result, at_sender=True)
total_case_data = cases_matcher_group.on_command("我的开箱", aliases={'开箱统计', '开箱查询', '查询开箱'})
total_case_data = cases_matcher_group.on_command(
"我的开箱", aliases={"开箱统计", "开箱查询", "查询开箱"}
)
@total_case_data.handle()
@@ -75,7 +85,9 @@ my_kinfes = on_command("我的金色", priority=1, permission=GROUP, block=True)
@my_kinfes.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
await my_kinfes.finish(await my_knifes_name(event.user_id, event.group_id), at_sender=True)
await my_kinfes.finish(
await my_knifes_name(event.user_id, event.group_id), at_sender=True
)
open_shilian = cases_matcher_group.on_regex(".*连开箱")
@@ -87,7 +99,7 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
# await k_open_case.finish('着什么急啊,慢慢来!', at_sender=True)
_flmt.start_cd(event.user_id)
msg = get_message_text(event.json())
rs = re.search(r'(.*)连开箱(.*)', msg)
rs = re.search(r"(.*)连开箱(.*)", msg)
if rs:
num = rs.group(1).strip()
if is_number(num) or num_dict.get(num):
@@ -96,61 +108,53 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
except KeyError:
num = int(num)
if num > 30:
await open_shilian.finish('开箱次数不要超过30啊笨蛋!', at_sender=True)
await open_shilian.finish("开箱次数不要超过30啊笨蛋!", at_sender=True)
else:
await open_shilian.finish('必须要是数字切不要超过30啊笨蛋!中文也可!', at_sender=True)
await open_shilian.finish("必须要是数字切不要超过30啊笨蛋!中文也可!", at_sender=True)
case_name = rs.group(2).strip()
if case_name.find('武器箱') != -1:
case_name = case_name.replace('武器箱', '').strip()
if case_name.find("武器箱") != -1:
case_name = case_name.replace("武器箱", "").strip()
if not case_name:
case_name = random.choice(cases_name)
elif case_name not in cases_name:
await open_shilian.finish('武器箱未收录!', at_sender=True)
await open_shilian.finish(await open_shilian_case(event.user_id, event.group_id, case_name, num), at_sender=True)
# await open_shilian.send("嘟嘟嘟...开箱中...")
# num = 10
# if str(state["_prefix"]["raw_command"]).find('十') != -1:
# num = 10
# elif str(state["_prefix"]["raw_command"]).find('五') != -1:
# num = 5
# if case_name:
# await open_shilian.finish(await open_shilian_case(event.user_id, event.group_id, case_name, num), at_sender=True)
# else:
# await open_shilian.finish(await open_shilian_case(event.user_id, event.group_id, random.choice(cases_name), num),
# at_sender=True)
await open_shilian.finish("武器箱未收录!", at_sender=True)
await open_shilian.finish(
await open_shilian_case(event.user_id, event.group_id, case_name, num),
at_sender=True,
)
num_dict = {
'一': 1,
'二': 2,
'三': 3,
'四': 4,
'五': 5,
'六': 6,
'七': 7,
'八': 8,
'九': 9,
'十': 10,
'十一': 11,
'十二': 12,
'十三': 13,
'十四': 14,
'十五': 15,
'十六': 16,
'十七': 17,
'十八': 18,
'十九': 19,
'二十': 20,
'二十一': 21,
'二十二': 22,
'二十三': 23,
'二十四': 24,
'二十五': 25,
'二十六': 26,
'二十七': 27,
'二十八': 28,
'二十九': 29,
'三十': 30
"一": 1,
"二": 2,
"三": 3,
"四": 4,
"五": 5,
"六": 6,
"七": 7,
"八": 8,
"九": 9,
"十": 10,
"十一": 11,
"十二": 12,
"十三": 13,
"十四": 14,
"十五": 15,
"十六": 16,
"十七": 17,
"十八": 18,
"十九": 19,
"二十": 20,
"二十一": 21,
"二十二": 22,
"二十三": 23,
"二十四": 24,
"二十五": 25,
"二十六": 26,
"二十七": 27,
"二十八": 28,
"二十九": 29,
"三十": 30,
}
@@ -172,10 +176,9 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
# 重置开箱
@scheduler.scheduled_job(
'cron',
"cron",
hour=0,
minute=1,
)
async def _():
await update_count_daily()