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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
from pathlib import Path
from tortoise import Tortoise
from ATRI.log import log
# 临时的实现，寻求更好的方式！欢迎pr
DB_DIR = Path(".") / "data" / "sql"
DB_DIR.mkdir(parents=True, exist_ok=True)
async def run():
from ATRI.database import models
await Tortoise.init(
{
"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"
},
},
"rrs": {
"engine": "tortoise.backends.sqlite",
"credentials": {
"file_path": f"{DB_DIR}/rssrsshubsubscription.sqlite3"
},
},
"rms": {
"engine": "tortoise.backends.sqlite",
"credentials": {
"file_path": f"{DB_DIR}/rssmikananisubscription.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"},
"rrs": {"models": [locals()["models"]], "default_connection": "rrs"},
"rms": {"models": [locals()["models"]], "default_connection": "rms"},
},
}
)
await Tortoise.generate_schemas()
async def init_database():
log.info("正在初始化数据库...")
await run()
log.success("数据库初始化完成")
async def close_database_connection():
log.info("正在关闭数据库连接...")
await Tortoise.close_connections()
log.info("数据库成功关闭")
|
