Skip to content

Commit

Permalink
Database connect/disconnect
Browse files Browse the repository at this point in the history
  • Loading branch information
mekanix committed Feb 12, 2022
1 parent ad4e148 commit 05dfb1c
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 2 deletions.
14 changes: 14 additions & 0 deletions freenit/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,18 @@

config = getConfig()
app = FastAPI()


@app.on_event("startup")
async def startup() -> None:
if not config.database.is_connected:
await config.database.connect()


@app.on_event("shutdown")
async def shutdown() -> None:
if config.database.is_connected:
await config.database.disconnect()


app.mount("/api/v1", api)
17 changes: 16 additions & 1 deletion freenit/project/project/app.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,22 @@
from fastapi import FastAPI
from freenit.config import getConfig

from .api import api
from .config import getConfig

config = getConfig()
app = FastAPI()


@app.on_event("startup")
async def startup() -> None:
if not config.database.is_connected:
await config.database.connect()


@app.on_event("shutdown")
async def shutdown() -> None:
if config.database.is_connected:
await config.database.disconnect()


app.mount("/api/v1", api)
2 changes: 1 addition & 1 deletion freenit/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
version = "0.2.18"
version = "0.2.19"

0 comments on commit 05dfb1c

Please sign in to comment.