feat: Probably functional
This commit is contained in:
13
app/db.py
13
app/db.py
@@ -1,9 +1,11 @@
|
||||
from pathlib import Path
|
||||
|
||||
from flask import Flask
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import DeclarativeBase, scoped_session, sessionmaker
|
||||
from sqlalchemy.orm import DeclarativeBase, Session, scoped_session, sessionmaker
|
||||
|
||||
engine = None
|
||||
SessionLocal = None
|
||||
SessionLocal: scoped_session[Session]
|
||||
|
||||
|
||||
class Base(DeclarativeBase):
|
||||
@@ -13,8 +15,13 @@ class Base(DeclarativeBase):
|
||||
def init_db(app: Flask) -> None: # noqa: ARG001
|
||||
global engine, SessionLocal # noqa: PLW0603
|
||||
|
||||
Path("./instance").mkdir(exist_ok=True)
|
||||
|
||||
engine = create_engine("sqlite:///instance/app.db", echo=True, future=True)
|
||||
SessionLocal = scoped_session(sessionmaker(bind=engine))
|
||||
session_local = scoped_session(sessionmaker(bind=engine))
|
||||
if session_local is None:
|
||||
raise RuntimeError
|
||||
SessionLocal = session_local
|
||||
|
||||
from . import models # noqa: F401, PLC0415
|
||||
|
||||
|
||||
Reference in New Issue
Block a user