Mbonea commited on
Commit
839f19d
·
1 Parent(s): b06927a
Files changed (1) hide show
  1. db.py +21 -19
db.py CHANGED
@@ -7,37 +7,39 @@ TORTOISE_ORM = {
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
  async def init_db():
24
  await Tortoise.init(
25
- db_url=DATABASE_URL,
26
- modules={'models': [
27
- 'App.routers.stocks.models',
28
- 'App.routers.tasks.models',
29
- 'App.routers.utt.models',
30
- 'App.routers.users.models',
31
- 'App.routers.portfolio.models',
32
- 'App.routers.bonds.models'
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('models').values():
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()