forked from benx1n/wows-stats-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
182 lines (174 loc) · 7.48 KB
/
__init__.py
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
from nonebot.typing import State_T
from nonebot.exceptions import CQHttpError
import hoshino
from hoshino import R, Service, priv, get_bot
from hoshino.util import FreqLimiter, DailyNumberLimiter
from hoshino.typing import CQEvent, MessageSegment
from nonebot import NoticeSession, on_command
from .html_render import text_to_pic
from .publicAPI import get_nation_list,get_ship_name,get_ship_byName
from .wws_info import get_AccountInfo
from .wws_recent import get_RecentInfo
from .wws_bind import set_BindInfo,get_BindInfo,change_BindInfo,set_special_BindInfo,delete_BindInfo
from .wws_ship import get_ShipInfo,get_ShipInfoRecent,ShipSecletProcess
from .wws_clan import get_ClanInfo,ClanSecletProcess
from .wws_shiprank import get_ShipRank
from .data_source import command_list
from .utils import match_keywords,find_and_replace_keywords,bytes2b64
import traceback
import httpx
import json
import re
_max = 100
EXCEED_NOTICE = f'您今天已经冲过{_max}次了，请明早5点后再来！'
_nlmt = DailyNumberLimiter(_max)
_flmt = FreqLimiter(3)
_version = "0.3.0.1"
WWS_help ="""请发送wws help查看帮助"""
sv_help = WWS_help.strip()
sv = Service('wows-stats-bot', manage_priv=priv.SUPERUSER, enable_on_default=True,help_ = sv_help)
@sv.on_fullmatch(('wws帮助','wws 帮助','wws help'))
async def get_help(bot, ev):
url = 'https://benx1n.oss-cn-beijing.aliyuncs.com/version.json'
async with httpx.AsyncClient() as client:
resp = await client.get(url, timeout=10)
result = json.loads(resp.text)
latest_version = result['latest_version']
url = 'https://benx1n.oss-cn-beijing.aliyuncs.com/wws_help.txt'
async with httpx.AsyncClient() as client:
resp = await client.get(url, timeout=10)
result = resp.text
result = f'''帮助列表 当前版本{_version} 最新版本{latest_version}\n{result}'''
img = await text_to_pic(text = result, width = 800)
await bot.send(ev,str(MessageSegment.image(bytes2b64(img))))
return
@sv.on_prefix(('wws 地区','wws地区'))
async def send_nation_info(bot, ev:CQEvent):
msg = await get_nation_list()
if msg:
await bot.send(ev,msg)
else:
await bot.send(ev,"wuwuwu~好像出了点问题")
return
@sv.on_prefix(('wws'))
async def selet_command(bot,ev:CQEvent):
try:
qqid = ev['user_id']
if not _nlmt.check(qqid):
await bot.send(ev, EXCEED_NOTICE, at_sender=True)
return
if not _flmt.check(qqid):
await bot.send(ev, '您冲得太快了，请稍候再冲', at_sender=True)
return
_flmt.start_cd(qqid)
_nlmt.increase(qqid)
select_command = None
searchtag = str(ev.message).strip()
if not searchtag or searchtag=="":
await bot.send(ev,WWS_help.strip())
return
match = re.search(r"(\(|（)(.*?)(\)|）)",str(ev.message))
replace_name = None
if match:
replace_name = match.group(2)
search_list = str(ev.message).replace(match.group(0),'').split()
else:
search_list = str(ev.message).split()
select_command,search_list = await find_and_replace_keywords(search_list,command_list)
if not select_command:
if replace_name:
search_list.append(replace_name)
msg = await get_AccountInfo(qqid,search_list)
elif select_command == 'ship':
select_command = None
select_command,search_list = await find_and_replace_keywords(search_list,command_list) #第二次匹配
if not select_command:
if replace_name:
search_list.append(replace_name)
msg = await get_ShipInfo(qqid,search_list,bot,ev)
elif select_command == 'recent':
msg = await get_ShipInfoRecent(qqid,search_list,bot,ev)
else:
msg = '看不懂指令QAQ'
elif select_command == 'recent':
select_command = None
select_command,search_list = await find_and_replace_keywords(search_list,command_list) #第二次匹配
if not select_command:
if replace_name:
search_list.append(replace_name)
msg = await get_RecentInfo(qqid,search_list)
elif select_command == 'ship':
msg = await get_ShipInfoRecent(qqid,search_list,bot,ev)
else:
msg = '：看不懂指令QAQ'
elif select_command == 'clan':
msg = await get_ClanInfo(qqid,search_list,bot,ev)
elif select_command == 'ship_rank':
msg = await get_ShipRank(qqid,search_list,bot,ev)
elif select_command == 'special_bind':
msg = await set_special_BindInfo(qqid,search_list)
elif select_command == 'bind':
if replace_name:
search_list.append(replace_name)
msg = await set_BindInfo(qqid,search_list)
elif select_command == 'bindlist':
msg = await get_BindInfo(qqid,search_list)
elif select_command == 'changebind':
msg = await change_BindInfo(qqid,search_list)
elif select_command == 'delete_bind':
msg = await delete_BindInfo(qqid,search_list)
elif select_command == 'searchship':
msg = await get_ship_name(search_list)
else:
msg = '看不懂指令QAQ'
if msg:
if isinstance(msg,str):
await bot.send(ev,msg)
else:
await bot.send(ev,str(MessageSegment.image(bytes2b64(msg))))
else:
await bot.send('呜呜呜发生了错误，可能是网络问题，如果过段时间不能恢复请联系麻麻哦~')
return
except Exception:
traceback.print_exc()
await bot.send(ev,'呜呜呜发生了错误，可能是网络问题，如果过段时间不能恢复请联系麻麻哦~')
return
@sv.on_message()
async def change_select_state(bot, ev):
msg = ev["raw_message"]
qqid = ev['user_id']
if ShipSecletProcess[qqid].SelectList and str(msg).isdigit():
ShipSecletProcess[qqid] = ShipSecletProcess[qqid]._replace(state = True)
ShipSecletProcess[qqid] = ShipSecletProcess[qqid]._replace(SlectIndex = int(msg))
if ClanSecletProcess[qqid].SelectList and str(msg).isdigit():
ShipSecletProcess[qqid] = ShipSecletProcess[qqid]._replace(state = True)
ShipSecletProcess[qqid] = ShipSecletProcess[qqid]._replace(SlectIndex = int(msg))
return
@sv.on_fullmatch('wws 检查更新')
async def check_version(bot, ev:CQEvent):
url = 'https://benx1n.oss-cn-beijing.aliyuncs.com/version.json'
async with httpx.AsyncClient() as client:
resp = await client.get(url, timeout=10)
result = json.loads(resp.text)
bot = hoshino.get_bot()
superid = hoshino.config.SUPERUSERS[0]
match,msg = False,f'发现新版本'
for each in result['data']:
if each['version'] > _version:
match = True
msg += f"\n{each['date']} v{each['version']}\n"
for i in each['description']:
msg += f"{i}\n"
if match:
await bot.send_private_msg(user_id=superid, message=msg)
try:
await bot.send(ev,msg)
except Exception:
traceback.print_exc()
return
return
@sv.scheduled_job('cron',hour='12')
async def job1():
bot = get_bot()
ev = CQEvent
await check_version(bot,ev)