add get db session method

database.py

@@ -4,12 +4,21 @@ from sqlalchemy.orm import sessionmaker
 
 
 ## ENV LOAD
-SQLALCHEMY_DATABASE_URL = 'mysql://choozmo:pAssw0rd@db.ptt.cx:3306/DBNAME?charset=utf8mb4'
+SQLALCHEMY_DATABASE_URL = 'mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4'
 
 engine = create_engine(SQLALCHEMY_DATABASE_URL)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
 
-
+# Dependency
+def get_db_session():
+    db = SessionLocal()
+    try:
+        yield db
+    except:
+        db.rollback()
+        raise
+    finally:
+        db.close()
 
 ### define database session, deal with database setup and connection

main.py

@@ -48,15 +48,6 @@ models.Base.metadata.create_all(bind=engine)
 # Fastapi
 app = FastAPI()
 
-# Dependency
-def get_db():
-    db = SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
-
-
 
 @app.get('/')
 def hellow():