Automated deployment: Mon Jan 3 10:27:19 UTC 2022 c1083f77b5117f7fbb358080d88b6569ff2d4311

src/db_sqlalchemy_access_tokens.py

@@ -0,0 +1,39 @@
+import databases
+import sqlalchemy
+from fastapi_users.db import SQLAlchemyBaseUserTable, SQLAlchemyUserDatabase
+from fastapi_users_db_sqlalchemy.access_token import (
+    SQLAlchemyAccessTokenDatabase,
+    SQLAlchemyBaseAccessTokenTable,
+)
+from sqlalchemy.ext.declarative import DeclarativeMeta, declarative_base
+
+from .models import AccessToken, UserDB
+
+DATABASE_URL = "sqlite:///./test.db"
+database = databases.Database(DATABASE_URL)
+Base: DeclarativeMeta = declarative_base()
+
+
+class UserTable(Base, SQLAlchemyBaseUserTable):
+    pass
+
+
+class AccessTokenTable(SQLAlchemyBaseAccessTokenTable, Base):
+    pass
+
+
+engine = sqlalchemy.create_engine(
+    DATABASE_URL, connect_args={"check_same_thread": False}
+)
+Base.metadata.create_all(engine)
+
+users = UserTable.__table__
+access_tokens = AccessTokenTable.__table__
+
+
+async def get_user_db():
+    yield SQLAlchemyUserDatabase(UserDB, database, users)
+
+
+async def get_access_token_db():
+    yield SQLAlchemyAccessTokenDatabase(AccessToken, database, access_tokens)