From b3c35c2ca00106fe475d82baa580ee26ee64e1ba Mon Sep 17 00:00:00 2001
From: Botond Hende <nettingman@gmail.com>
Date: Sun, 25 Aug 2024 16:30:51 +0200
Subject: create/connect to database on startup

---
 __main__.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to '__main__.py')

diff --git a/__main__.py b/__main__.py
index deacefa..7ae3047 100644
--- a/__main__.py
+++ b/__main__.py
@@ -1,9 +1,28 @@
+import os.path
+import sqlite3
+from pathlib import Path
+
+from contextlib import asynccontextmanager
 from typing import Annotated
 
 from fastapi import FastAPI, Depends, Request
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 
-app = FastAPI()
+from .config import Config
+
+
+sq_con = None
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    db_dir = os.path.dirname(Config.DATABASE_PATH)
+    Path(db_dir).mkdir(parents=True, exist_ok=True)
+
+    sq_con = sqlite3.connect(Config.DATABASE_PATH)
+    yield
+
+
+app = FastAPI(lifespan=lifespan)
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
-- 
cgit v1.2.3-70-g09d2