From 81a1eeef9d7c393e683167470e95c0a15e7af6c0 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Fri, 1 Jul 2022 00:05:36 +0800 Subject: =?UTF-8?q?=F0=9F=8E=A8=20=E4=BC=98=E5=8C=96=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E5=BA=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/database/__init__.py | 2 +- ATRI/database/db.py | 36 ++++++++++++++++++++++++++++++++++-- 2 files changed, 35 insertions(+), 3 deletions(-) diff --git a/ATRI/database/__init__.py b/ATRI/database/__init__.py index 497bea4..5c12c6a 100644 --- a/ATRI/database/__init__.py +++ b/ATRI/database/__init__.py @@ -1,2 +1,2 @@ from .db import init_database, close_database_connection -from .models import BilibiliSubscription, TwitterSubscription +from .models import * diff --git a/ATRI/database/db.py b/ATRI/database/db.py index c8782de..a5a04c8 100644 --- a/ATRI/database/db.py +++ b/ATRI/database/db.py @@ -15,8 +15,40 @@ async def run(): from ATRI.database import models await Tortoise.init( - db_url=f"sqlite://{DB_DIR}/db.sqlite3", - modules={"models": [locals()["models"]]}, + { + "connections": { + "bilibili": { + "engine": "tortoise.backends.sqlite", + "credentials": {"file_path": f"{DB_DIR}/bilibili.sqlite3"}, + }, + "twitter": { + "engine": "tortoise.backends.sqlite", + "credentials": {"file_path": f"{DB_DIR}/twitter.sqlite3"}, + }, + "ts": { + "engine": "tortoise.backends.sqlite", + "credentials": {"file_path": f"{DB_DIR}/thesaurusstoragor.sqlite3"}, + }, + "tal": { + "engine": "tortoise.backends.sqlite", + "credentials": { + "file_path": f"{DB_DIR}/thesaurusauditlist.sqlite3" + }, + }, + }, + "apps": { + "bilibili": { + "models": [locals()["models"]], + "default_connection": "bilibili", + }, + "twitter": { + "models": [locals()["models"]], + "default_connection": "twitter", + }, + "ts": {"models": [locals()["models"]], "default_connection": "ts"}, + "tal": {"models": [locals()["models"]], "default_connection": "tal"}, + }, + } ) await Tortoise.generate_schemas() -- cgit v1.2.3