From 48ccaabf66828594051de919c9b52098debf95a3 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 18 Jun 2022 00:19:13 +0800 Subject: =?UTF-8?q?=F0=9F=9A=9A=F0=9F=92=A9=20=E7=A7=BB=E5=8A=A8=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E5=BA=93=E7=9B=B8=E5=85=B3=E5=87=BD=E6=95=B0,=20?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/database/__init__.py | 2 +- ATRI/database/db.py | 12 +++++++++--- 2 files changed, 10 insertions(+), 4 deletions(-) (limited to 'ATRI/database') diff --git a/ATRI/database/__init__.py b/ATRI/database/__init__.py index 6c58d20..497bea4 100644 --- a/ATRI/database/__init__.py +++ b/ATRI/database/__init__.py @@ -1,2 +1,2 @@ -from .db import init_database +from .db import init_database, close_database_connection from .models import BilibiliSubscription, TwitterSubscription diff --git a/ATRI/database/db.py b/ATRI/database/db.py index 6e20ad3..c8782de 100644 --- a/ATRI/database/db.py +++ b/ATRI/database/db.py @@ -1,5 +1,5 @@ from pathlib import Path -from tortoise import Tortoise, run_async +from tortoise import Tortoise from ATRI.log import logger as log @@ -21,7 +21,13 @@ async def run(): await Tortoise.generate_schemas() -def init_database(): +async def init_database(): log.info("正在初始化数据库...") - run_async(run()) + await run() log.success("数据库初始化完成") + + +async def close_database_connection(): + log.info("正在关闭数据库连接...") + await Tortoise.close_connections() + log.info("数据库成功关闭") -- cgit v1.2.3