From 3721881813ed7cbe12d6140680284803e3161c5e Mon Sep 17 00:00:00 2001
From: Colin Goutte <cgte@bk.ru>
Date: Fri, 25 Aug 2023 12:03:32 +0200
Subject: [PATCH] RED: Fix import, still need to create proper database
 strcuture

---
 .gitignore           |  2 +-
 dev.py               | 14 ++++++++++++++
 models.py            |  6 ++++++
 test_sql_database.py |  2 +-
 4 files changed, 22 insertions(+), 2 deletions(-)
 create mode 100644 models.py

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://"