From 98fb742d4dbb7d9c9a62ce7f38585b018f3a0886 Mon Sep 17 00:00:00 2001
From: HibiKier <45528451+HibiKier@users.noreply.github.com>
Date: Thu, 17 Jun 2021 10:18:58 +0800
Subject: [PATCH] Update __init__.py

---
 plugins/draw_card/__init__.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/plugins/draw_card/__init__.py b/plugins/draw_card/__init__.py
index 685c0ac6..2fe78c25 100644
--- a/plugins/draw_card/__init__.py
+++ b/plugins/draw_card/__init__.py
@@ -4,7 +4,7 @@ from nonebot.permission import SUPERUSER
 from nonebot.typing import T_State
 from .genshin_handle import genshin_draw, update_genshin_info, reset_count, reload_genshin_pool
 from .prts_handle import update_prts_info, prts_draw, reload_prts_pool
-from .pretty_handle import update_pretty_info, pretty_draw
+from .pretty_handle import update_pretty_info, pretty_draw, reload_pretty_pool
 from .guardian_handle import update_guardian_info, guardian_draw
 from .pcr_handle import update_pcr_info, pcr_draw
 from .azur_handle import update_azur_info, azur_draw
@@ -34,6 +34,7 @@ genshin_up_reload = on_keyword({'重载原神卡池'}, priority=1, block=True)
 
 pretty = on_regex('.*?马娘卡?[1-9|一][0-9]{0,2}[抽|井]', rule=is_switch('pretty'), priority=5, block=True)
 pretty_update = on_keyword({'更新马娘信息', '更新赛马娘信息'}, permission=SUPERUSER, priority=1, block=True)
+pretty_up_reload = on_keyword({'重载赛马娘卡池'}, priority=1, block=True)
 
 guardian = on_regex('.*?坎公骑冠剑武?器?[1-9|一][0-9]{0,2}[抽|井]', rule=is_switch('guardian'), priority=5, block=True)
 guardian_update = on_keyword({'更新坎公骑冠剑信息'}, permission=SUPERUSER, priority=1, block=True)
@@ -141,6 +142,12 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
         f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}) 赛马娘 {num}抽")
 
 
+@pretty_up_reload.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    text = await reload_pretty_pool()
+    await genshin_reset.finish(Message(f'重载成功！\n{text}'))
+
+
 @guardian.handle()
 async def _(bot: Bot, event: MessageEvent, state: T_State):
     msg = str(event.get_message()).strip()
@@ -318,6 +325,18 @@ async def _():
         logger.info('自动重载方舟卡池UP成功')
 
 
+# 每天四点重载赛马娘up卡池
+@scheduler.scheduled_job(
+    'cron',
+    hour=4,
+    minute=1,
+)
+async def _():
+    if PRETTY_FLAG:
+        await reload_pretty_pool()
+        logger.info('自动重载赛马娘UP成功')
+
+
 # 每天下午六点点重载原神up卡池
 @scheduler.scheduled_job(
     'cron',