diff --git a/.gitignore b/.gitignore index 54cd5e6..dabdbca 100644 --- a/.gitignore +++ b/.gitignore @@ -4,4 +4,4 @@ __pycache__/ *.py[o|c] **_build/ geckodriver.log - +sql_app.db diff --git a/dev.py b/dev.py index 0ce2447..35a02a1 100644 --- a/dev.py +++ b/dev.py @@ -1,9 +1,23 @@ from fastapi import FastAPI import uvicorn +from database import SessionLocal, engine +import models + + +models.Base.metadata.create_all(bind=engine) app = FastAPI() +# Dependency +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() + + @app.get("/") async def root(): return {"message": "Hello World"} diff --git a/models.py b/models.py new file mode 100644 index 0000000..e5995ad --- /dev/null +++ b/models.py @@ -0,0 +1,6 @@ +from database import Base + + +class Movie(Base): + __tablename__ = "movies" + diff --git a/test_sql_database.py b/test_sql_database.py index e43e407..7ed0f43 100644 --- a/test_sql_database.py +++ b/test_sql_database.py @@ -4,7 +4,7 @@ from sqlalchemy.orm import sessionmaker from sqlalchemy.pool import StaticPool from database import Base -from main import app, get_db +from dev import app, get_db from models import Movie SQLALCHEMY_DATABASE_URL = "sqlite://"