refactor: 数据存储重构,分离不同类型的存储实现

This commit is contained in:
Relakkes
2024-01-14 22:06:31 +08:00
parent e31aebbdfb
commit 894dabcf63
37 changed files with 1427 additions and 864 deletions

10
db.py
View File

@@ -1,14 +1,20 @@
from typing import List
from tortoise import Tortoise, run_async
from config.db_config import *
from tools import utils
def get_platform_models() -> List[str]:
models = ["store.xhs", "store.douyin", "store.bilibili", "store.kuaishou", "store.weibo"]
return models
async def init_db(create_db: bool = False) -> None:
await Tortoise.init(
db_url=RELATION_DB_URL,
modules={'models': ['models']},
# modules={'models': ['models.kuaishou']}, # generate special table
modules={'models': get_platform_models()},
_create_db=create_db
)