From f5ceb8927f2e7f2a9e29d62c8e4cef876f917249 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sat, 6 Feb 2021 00:32:26 +0800
Subject: =?UTF-8?q?=F0=9F=8F=97=20=F0=9F=92=A9=20=E6=9B=B4=E6=94=B9?=
 =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E7=BB=93=E6=9E=84=EF=BC=8C=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=E5=95=A5b=20BUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/utils/apscheduler.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 ATRI/utils/apscheduler.py

(limited to 'ATRI/utils/apscheduler.py')

diff --git a/ATRI/utils/apscheduler.py b/ATRI/utils/apscheduler.py
new file mode 100644
index 0000000..2680f19
--- /dev/null
+++ b/ATRI/utils/apscheduler.py
@@ -0,0 +1,34 @@
+# Fork from: https://github.com/nonebot/plugin-apscheduler
+
+import logging
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+
+from nonebot import get_driver, export
+from nonebot.log import logger, LoguruHandler
+
+
+apscheduler_autostart: bool = True
+apscheduler_config: dict = {"apscheduler.timezone": "Asia/Shanghai"}
+
+
+driver = get_driver()
+scheduler = AsyncIOScheduler()
+export().scheduler = scheduler
+
+
+async def _start_scheduler():
+    if not scheduler.running:
+        scheduler.configure(apscheduler_config)
+        scheduler.start()
+        logger.info("Scheduler Started.")
+
+
+if apscheduler_autostart:
+    driver.on_startup(_start_scheduler)
+
+aps_logger = logging.getLogger("apscheduler")
+aps_logger.setLevel(logging.DEBUG)
+aps_logger.handlers.clear()
+aps_logger.addHandler(LoguruHandler())
+
+from apscheduler.triggers.date import DateTrigger
-- 
cgit v1.2.3