add database support

backend/main.py

@@ -0,0 +1,98 @@
+from typing import Annotated
+from fastapi import FastAPI, Depends, HTTPException, Path
+from sqlalchemy.orm import Session
+from pydantic import BaseModel, Field
+from starlette import status
+
+from database import engine, SessionLocal
+from models import Base, Post, Thread
+
+app = FastAPI()
+
+Base.metadata.create_all(bind=engine)
+
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+
+db_dependency = Annotated[Session, Depends(get_db)]
+
+
+class PostCreate(BaseModel):
+    author: str = Field('anon')
+    title: str = Field('')
+    content: str = Field('')
+
+
+@app.get('/', status_code=status.HTTP_200_OK)
+async def get_threads(db: db_dependency):
+    return db.query(Thread).all()
+
+
+@app.post('/', status_code=status.HTTP_201_CREATED)
+async def create_thread(db: db_dependency, data: PostCreate):
+    # Create the post
+    post = Post(
+        author=data.author,
+        title=data.title,
+        content=data.content
+    )
+    db.add(post)
+    db.commit()
+
+    # Create the thread
+    thread = Thread(
+        author=post.author,
+        title=post.title,
+        content=post.content
+    )
+    db.add(thread)
+    db.commit()
+
+    # Update the Post with thread_id
+    post.thread_id = thread.id
+    db.commit()
+
+    return {
+        'id': post.id,
+        'author': post.author,
+        'title': post.title,
+        'content': post.content
+    }
+
+
+@app.get('/{thread_id}', status_code=status.HTTP_200_OK)
+async def get_thread_by_id(db: db_dependency, thread_id: int = Path(gt=0)):
+    posts = db.query(Post).filter(Post.thread_id == thread_id).all()
+    if posts:
+        return posts
+
+    raise HTTPException(404, f'Could not find thread')
+
+
+@app.post('/{thread_id}', status_code=status.HTTP_201_CREATED)
+async def create_reply(db: db_dependency, data: PostCreate, thread_id: int = Path(gt=0)):
+    thread = db.query(Thread).filter(Thread.id == thread_id).first()
+    if thread:
+        post = Post(
+            thread_id=thread_id,
+            author=data.author,
+            title=data.title,
+            content=data.content
+        )
+        db.add(post)
+        db.commit()
+
+        return {
+            'id': post.id,
+            'author': post.author,
+            'title': post.title,
+            'content': post.content
+        }
+
+    raise HTTPException(404, f'Could not find thread')

backend/models.py

@@ -1,8 +1,10 @@
+from sqlalchemy.orm import relationship
+
 from database import Base
 from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 
 
-class Posts(Base):
+class Post(Base):
     __tablename__ = 'posts'
 
     id = Column(Integer, primary_key=True)
@@ -12,11 +14,10 @@ class Posts(Base):
     content = Column(String)
 
 
-class Threads(Base):
+class Thread(Base):
     __tablename__ = 'threads'
 
     id = Column(Integer, primary_key=True)
     author = Column(String)
     title = Column(String)
     content = Column(String)
-    is_closed = Column(Boolean, default=False)