db?
Browse files
db.py
CHANGED
@@ -7,37 +7,39 @@ TORTOISE_ORM = {
|
|
7 |
"apps": {
|
8 |
"models": {
|
9 |
"models": [
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
],
|
18 |
"default_connection": "default",
|
19 |
}
|
20 |
-
}
|
21 |
}
|
22 |
|
|
|
23 |
async def init_db():
|
24 |
await Tortoise.init(
|
25 |
-
db_url=DATABASE_URL,
|
26 |
-
modules={'models': [
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
]}
|
34 |
)
|
35 |
await Tortoise.generate_schemas()
|
36 |
|
|
|
37 |
async def close_db():
|
38 |
await Tortoise.close_connections()
|
39 |
|
40 |
|
41 |
async def clear_db():
|
42 |
-
for model in Tortoise.apps.get(
|
43 |
-
await model.all().delete()
|
|
|
7 |
"apps": {
|
8 |
"models": {
|
9 |
"models": [
|
10 |
+
"App.routers.stocks.models",
|
11 |
+
"App.routers.tasks.models",
|
12 |
+
"App.routers.utt.models",
|
13 |
+
"App.routers.users.models",
|
14 |
+
"App.routers.portfolio.models",
|
15 |
+
"App.routers.bonds.models",
|
16 |
+
"aerich.models",
|
17 |
],
|
18 |
"default_connection": "default",
|
19 |
}
|
20 |
+
},
|
21 |
}
|
22 |
|
23 |
+
|
24 |
async def init_db():
|
25 |
await Tortoise.init(
|
26 |
+
TORTOISE_ORM # db_url=DATABASE_URL,
|
27 |
+
# modules={'models': [
|
28 |
+
# 'App.routers.stocks.models',
|
29 |
+
# 'App.routers.tasks.models',
|
30 |
+
# 'App.routers.utt.models',
|
31 |
+
# 'App.routers.users.models',
|
32 |
+
# 'App.routers.portfolio.models',
|
33 |
+
# 'App.routers.bonds.models'
|
34 |
+
# ]}
|
35 |
)
|
36 |
await Tortoise.generate_schemas()
|
37 |
|
38 |
+
|
39 |
async def close_db():
|
40 |
await Tortoise.close_connections()
|
41 |
|
42 |
|
43 |
async def clear_db():
|
44 |
+
for model in Tortoise.apps.get("models").values():
|
45 |
+
await model.all().delete()
|