132 lines
2.4 KiB
Python
132 lines
2.4 KiB
Python
from fastapi import FastAPI
|
|
import models
|
|
from database import engine, SessionLocal
|
|
from pydantic import BaseModel
|
|
|
|
app = FastAPI()
|
|
|
|
models.Base.metadata.create_all(bind=engine)
|
|
|
|
|
|
def get_db():
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
class Post:
|
|
"""Post is a single forum post or reply"""
|
|
id: int
|
|
thread_id: int
|
|
author: str
|
|
title: str
|
|
content: str
|
|
|
|
def __init__(self, id, thread_id, author, title, content):
|
|
self.id = id
|
|
self.thread_id = thread_id
|
|
self.author = author
|
|
self.title = title
|
|
self.content = content
|
|
|
|
|
|
class PostCreate(BaseModel):
|
|
"""Used when creating posts and replies"""
|
|
author: str = 'anon'
|
|
title: str
|
|
content: str
|
|
|
|
|
|
class Thread:
|
|
"""Thread is a collection of a post and its replies"""
|
|
id: int
|
|
author: str
|
|
title: str
|
|
closed: bool
|
|
|
|
def __init__(self, id, author, title, closed=False):
|
|
self.id = id
|
|
self.author = author
|
|
self.title = title
|
|
self.closed = closed
|
|
|
|
|
|
POSTS = []
|
|
THREADS = []
|
|
|
|
|
|
@app.get("/")
|
|
async def get_threads():
|
|
return THREADS
|
|
|
|
|
|
@app.post("/post")
|
|
async def create_post(data: PostCreate):
|
|
id = len(POSTS) + 1
|
|
author = data.author
|
|
title = data.title
|
|
content = data.content
|
|
|
|
# Create thread
|
|
thread = Thread(
|
|
id=id,
|
|
author=author,
|
|
title=title
|
|
)
|
|
THREADS.append(thread)
|
|
|
|
# Create post
|
|
post = Post(
|
|
id=id,
|
|
thread_id=id,
|
|
author=author,
|
|
title=title,
|
|
content=content
|
|
)
|
|
POSTS.append(post)
|
|
|
|
return post
|
|
|
|
|
|
@app.get("/{thread_id}")
|
|
async def get_thread(thread_id: int):
|
|
result = []
|
|
for post in POSTS:
|
|
if post.thread_id == thread_id:
|
|
result.append(post)
|
|
|
|
if len(result) == 0:
|
|
return {'error': 'could not find parent thread'}
|
|
|
|
return result
|
|
|
|
|
|
@app.post("/{thread_id}/post")
|
|
async def reply_to_post(thread_id: int, data: PostCreate):
|
|
parent = None
|
|
for thread in THREADS:
|
|
if thread.id == thread_id:
|
|
parent = thread
|
|
|
|
if parent is None:
|
|
return {'error': 'could not find parent thread'}
|
|
|
|
id = len(POSTS) + 1
|
|
author = data.author
|
|
title = data.title
|
|
content = data.content
|
|
|
|
# Create post
|
|
post = Post(
|
|
id=id,
|
|
thread_id=parent.id,
|
|
author=author,
|
|
title=title,
|
|
content=content
|
|
)
|
|
POSTS.append(post)
|
|
|
|
return post
|