polka_billy/porridger
1
0
Fork 1
Code Issues 20 Pull Requests Packages Releases Activity

Fixed initial tables creation

Browse Source
This commit is contained in:
Dmitriy Shishkov 2023-09-08 19:40:41 +03:00
parent 22dc21bda1
commit acd0a8fbf7
Signed by: dm1sh
GPG Key ID: 027994B0AA357688
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
back/db.py
View File

@ -17,5 +17,4 @@ Base = declarative_base()
# Создаем таблицы # Создаем таблицы
async def init_models(): async def init_models():
async with engine.begin() as conn: async with engine.begin() as conn:
await conn.run_sync(Base.metadata.drop_all)
await conn.run_sync(Base.metadata.create_all) await conn.run_sync(Base.metadata.create_all)

5
back/main.py
View File

@ -3,6 +3,8 @@ import uvicorn
from .api import app as app_fastapi from .api import app as app_fastapi
from .scheduler import app as app_rocketry from .scheduler import app as app_rocketry
from .db import init_models
class Server(uvicorn.Server): class Server(uvicorn.Server):
"""Customized uvicorn.Server """Customized uvicorn.Server
@ -16,6 +18,9 @@ class Server(uvicorn.Server):
async def main(): async def main():
"Run scheduler and the API" "Run scheduler and the API"
await init_models()
server = Server(config=uvicorn.Config(app_fastapi, workers=1, loop="asyncio", host="0.0.0.0")) server = Server(config=uvicorn.Config(app_fastapi, workers=1, loop="asyncio", host="0.0.0.0"))
api = asyncio.create_task(server.serve()) api = asyncio.create_task(server.serve())