Database connect/disconnect

Author: mekanix

freenit/app.py

@@ -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)

freenit/project/project/app.py

@@ -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)

freenit/version.py

@@ -1 +1 @@
-version = "0.2.18"
+version = "0.2.19"