summaryrefslogtreecommitdiff
path: root/ATRI/plugins/__init__.py
blob: 327855b6ab68d5d8119b071946d0852f7aeb3167 (plain)
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
import ATRI
from ATRI.log import log
from ATRI.utils.apscheduler import scheduler
from ATRI.utils.check_update import CheckUpdate
from ATRI.database import init_database, close_database_connection

from time import sleep

from ATRI.patch import *

driver = ATRI.driver()


@driver.on_startup
async def startup():
    await init_database()

    log.info(f"当前版本: {ATRI.__version__}")

    log.info("开始检查更新...")
    commit_info = await CheckUpdate.show_latest_commit_info()
    if commit_info:
        log.info(commit_info)

    l_v, l_v_t = await CheckUpdate.show_latest_version()
    if l_v and l_v_t:
        if l_v != ATRI.__version__:
            log.warning("新版本已发布, 请更新")
            log.warning(f"最新版本: {l_v} 更新时间: {l_v_t}")
            sleep(3)

    if not scheduler.running:
        scheduler.start()
        log.info("定时任务已启用")

    log.info("アトリは、高性能ですから!")


@driver.on_shutdown
async def shutdown():
    await close_database_connection()

    scheduler.shutdown(False)

    log.info("感谢使用")