agatha/forum-app
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

add get_db method

Browse Source
This commit is contained in:
agatha 2024-04-01 17:25:32 -04:00
parent 79ec14516a
commit fdea61b784
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/forum.py
View File

@ -1,6 +1,6 @@
from fastapi import FastAPI from fastapi import FastAPI
import models import models
from database import engine from database import engine, SessionLocal
from pydantic import BaseModel from pydantic import BaseModel
app = FastAPI() app = FastAPI()
@ -8,6 +8,14 @@ app = FastAPI()
models.Base.metadata.create_all(bind=engine) models.Base.metadata.create_all(bind=engine)
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
class Post: class Post:
"""Post is a single forum post or reply""" """Post is a single forum post or reply"""
id: int id: int