-
Notifications
You must be signed in to change notification settings - Fork 10
/
__init__.py
131 lines (108 loc) · 4.59 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
import asyncio
import base64
import json
import os
import shutil
from hoshino import Service, R, logger
from .plugin_utils.base_util import get_img_cq, get_interval
from .plugin_utils.plugin_update import init_plugin, plugin_update_auto
from .uma_comic.update_init import update as comic_update, auto_update as comic_auto
from .uma_compatibility.update_init import update as com_update, auto_update as com_auto
from .uma_face.update_init import update as face_update, auto_update as face_auto
from .uma_gacha_v2.update_init import update as gacha_update, auto_update as gacha_auto
from .uma_info.update_init import update as info_update, update_info_auto as info_auto
from .uma_skills.update_init import update as skills_update, auto_update as skills_auto
from .uma_support_chart.update_init import update as sup_update
from .uma_tasks.update_init import update as tasks_update, auto_update as tasks_auto
sv = Service('uma_help')
with open(os.path.join(os.path.dirname(__file__), f'{sv.name}.png'), 'rb') as f:
base64_data = base64.b64encode(f.read())
s = base64_data.decode()
sv.help = f''
@sv.on_fullmatch('马娘帮助')
async def get_help(bot, ev):
img_path = os.path.join(os.path.dirname(__file__), f'{sv.name}.png')
sv_help = await get_img_cq(img_path)
await bot.send(ev, sv_help)
@sv.on_fullmatch(('马娘插件-v', '马娘插件-version'))
async def get_plugin_version(bot, ev):
version_path = os.path.join(os.path.dirname(__file__), 'version.json')
with open(version_path, 'r', encoding="UTF-8") as file:
version_data = json.load(file)
version = version_data['version']
commit_time = version_data['commit_time']
msg = f'【马娘插件】\n基础版本：{version}\n更新时间：{commit_time}'
await bot.send(ev, msg)
# 马娘速查，一些支援卡对比的子网站就不列举了，去百科里找就行了
@sv.on_fullmatch('马娘速查')
async def uma_query(bot, ev):
query_path = os.path.join(os.path.dirname(__file__), 'query_data.json')
with open(query_path, 'r', encoding='UTF-8') as file:
query_data = json.load(file)
msg = '◎赛马娘常用网站◎'
for title in list(query_data.keys()):
msg += '\n' + title + '：' + query_data[title]
await bot.send(ev, msg)
# v1.5.2将图片文件夹合并至一个文件夹
root_path = R.img('umamusume').path
def move_dir(dir_name):
if os.path.exists(R.img(dir_name).path):
shutil.move(os.path.abspath(R.img(dir_name).path), os.path.abspath(root_path))
if not os.path.exists(root_path):
os.mkdir(root_path)
move_dir('uma_bir')
move_dir('uma_comic')
move_dir('uma_face')
move_dir('uma_support_chart')
move_dir('uma_voice')
move_dir('umamusume_news')
# 这样独立版马娘抽卡删除后再装本整合版可以通用
move_dir('uma_gacha')
# v1.8将首次启动事件合并到一块，防止首次启动时过多线程同时工作导致触发反爬虫
async def update():
logger.info('【马娘插件】启动时检查更新...')
try:
await init_plugin(False)
except:
logger.error('马娘插件版本获取失败，插件部分功能已停止启动！请检查是否能访问Github，如不能请设置代理或者关闭插件自动更新功能，重启生效')
await asyncio.sleep(0.1)
flag = await info_update()
if not flag:
logger.error('马娘基础数据库更新失败，后续部分更新操作已停止，请重新启动Bot更新')
await asyncio.sleep(0.1)
await comic_update()
await asyncio.sleep(0.1)
await com_update()
await asyncio.sleep(0.1)
await face_update()
await asyncio.sleep(0.1)
await skills_update()
await asyncio.sleep(0.1)
await tasks_update()
await asyncio.sleep(0.1)
await sup_update()
await asyncio.sleep(0.1)
await gacha_update()
logger.info('【马娘插件】更新检查完成...')
loop = asyncio.get_event_loop()
loop.run_until_complete(update())
# 部分统一自动更新时间点
@sv.scheduled_job('cron', id='daily_uma_res', day=f'1/{get_interval()}', hour='2', minute='30')
async def auto_update():
await plugin_update_auto()
await asyncio.sleep(0.1)
_ = await info_auto()
await asyncio.sleep(0.1)
await comic_auto()
await asyncio.sleep(0.1)
await com_auto()
await asyncio.sleep(0.1)
await face_auto()
await asyncio.sleep(0.1)
await gacha_auto()
# 其他每小时自动对比是否有更新
@sv.scheduled_job('cron', id='hourly_uma_res', hour='0-23', minute='00')
async def auto_update_per():
await skills_auto()
await asyncio.sleep(0.1)
await tasks_auto()