Automated deployment: Mon Sep 20 06:43:34 UTC 2021 ab89cdea2f

src/db_sqlalchemy.py

@@ -1,31 +1,12 @@
 import databases
 import sqlalchemy
-from fastapi import FastAPI
-from fastapi_users import models
 from fastapi_users.db import SQLAlchemyBaseUserTable, SQLAlchemyUserDatabase
 from sqlalchemy.ext.declarative import DeclarativeMeta, declarative_base
 
-
-class User(models.BaseUser):
-    pass
-
-
-class UserCreate(models.BaseUserCreate):
-    pass
-
-
-class UserUpdate(User, models.BaseUserUpdate):
-    pass
-
-
-class UserDB(User, models.BaseUserDB):
-    pass
-
+from .models import UserDB
 
 DATABASE_URL = "sqlite:///./test.db"
-
 database = databases.Database(DATABASE_URL)
-
 Base: DeclarativeMeta = declarative_base()
 
 
@@ -36,20 +17,10 @@ class UserTable(Base, SQLAlchemyBaseUserTable):
 engine = sqlalchemy.create_engine(
     DATABASE_URL, connect_args={"check_same_thread": False}
 )
-
 Base.metadata.create_all(engine)
 
 users = UserTable.__table__
-user_db = SQLAlchemyUserDatabase(UserDB, database, users)
-
-app = FastAPI()
 
 
-@app.on_event("startup")
-async def startup():
-    await database.connect()
-
-
-@app.on_event("shutdown")
-async def shutdown():
-    await database.disconnect()
+def get_user_db():
+    yield SQLAlchemyUserDatabase(UserDB, database, users)