diff options
| -rw-r--r-- | ATRI/database/__init__.py | 2 | ||||
| -rw-r--r-- | 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() |