Update backend/post_router.py

backend/post_router.py

@@ -1,119 +1,119 @@
-from fastapi import APIRouter, Depends, HTTPException
-from sqlalchemy.orm import Session
-from models import Post, User
-from db import get_db
-from pydantic import BaseModel
-from typing import Optional
-from fastapi.security import OAuth2PasswordBearer
-import jwt
-from auth.routes import get_current_user
-
-class PostCreate(BaseModel):
-    title: str
-    content: Optional[str] = None
-
-class PostUpdate(BaseModel):
-    post_id: int
-    title: str
-    content: Optional[str] = None
-
-SECRET_KEY = "52a6206f34a1c479da043cdeee17fd859a35e54978a6733a6a7ebadcbd11f0ca"
-ALGORITHM = "HS256"
-
-router = APIRouter()
-
-oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login")  # 로그인 경로 설정
-
-
-def get_auth_user(
-    token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)
-) -> User:
-    try:
-        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
-        email: str = payload.get("sub")
-        if email is None:
-            raise HTTPException(status_code=401, detail="Token payload invalid")
-    except jwt.PyJWTError:
-        raise HTTPException(status_code=401, detail="Invalid token")
-
-    user = db.query(User).filter(User.user_email == email).first()
-    if user is None:
-        raise HTTPException(status_code=401, detail="User not found")
-
-    return user
-
-
-@router.post("/posts")
-def create_post(
-    post: PostCreate,
-    db: Session = Depends(get_db),
-    current_user: User = Depends(get_auth_user),
-):
-    new_post = Post(
-        title=post.title, content=post.content, user_id=current_user.user_id
-    )
-    db.add(new_post)
-    db.commit()
-    db.refresh(new_post)
-    return {"message": "Post created", "post_id": new_post.post_id}
-
-
-@router.get("/list")
-def get_my_posts(
-    current_user: User = Depends(get_auth_user), db: Session = Depends(get_db)
-):
-    posts = db.query(Post).filter(Post.user_id == current_user.user_id).all()
-    return [{"post_id": post.post_id, "title": post.title} for post in posts]
-
-
-@router.get("/{post_id}")
-def read_user_post(
-    post_id: int,
-    current_user: User = Depends(get_auth_user),
-    db: Session = Depends(get_db),
-):
-    post = (
-        db.query(Post)
-        .filter(Post.post_id == post_id, Post.user_id == current_user.user_id)
-        .first()
-    )
-    if post is None:
-        raise HTTPException(status_code=404, detail="포스트가 없거나 권한이 없습니다.")
-    return {"post_id": post.post_id, "title": post.title, "content": post.content}
-
-@router.put("/save")
-def update_post(
-    post: PostUpdate,
-    db: Session = Depends(get_db),
-    current_user: User = Depends(get_auth_user),
-):
-    db_post = (
-        db.query(Post)
-        .filter(Post.post_id == post.post_id, Post.user_id == current_user.user_id)
-        .first()
-    )
-    if not db_post:
-        raise HTTPException(status_code=404, detail="Post not found or access denied")
-
-    db_post.title = post.title
-    db_post.content = post.content
-    db.commit()
-    db.refresh(db_post)
-
-    return {"message": "Post updated successfully"}
-
-
-@router.delete("/{post_id}")
-def delete_post(
-    post_id: int, db: Session = Depends(get_db), user=Depends(get_auth_user)
-):
-    post = (
-        db.query(Post)
-        .filter(Post.post_id == post_id, Post.user_id == user.user_id)
-        .first()
-    )
-    if not post:
-        raise HTTPException(status_code=404, detail="Post not found")
-    db.delete(post)
-    db.commit()
-    return {"message": "Post deleted"}
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from backend.models import Post, User
+from backend.db import get_db
+from pydantic import BaseModel
+from typing import Optional
+from fastapi.security import OAuth2PasswordBearer
+import jwt
+from backend.auth.routes import get_current_user
+
+class PostCreate(BaseModel):
+    title: str
+    content: Optional[str] = None
+
+class PostUpdate(BaseModel):
+    post_id: int
+    title: str
+    content: Optional[str] = None
+
+SECRET_KEY = "52a6206f34a1c479da043cdeee17fd859a35e54978a6733a6a7ebadcbd11f0ca"
+ALGORITHM = "HS256"
+
+router = APIRouter()
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login")  # 로그인 경로 설정
+
+
+def get_auth_user(
+    token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)
+) -> User:
+    try:
+        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
+        email: str = payload.get("sub")
+        if email is None:
+            raise HTTPException(status_code=401, detail="Token payload invalid")
+    except jwt.PyJWTError:
+        raise HTTPException(status_code=401, detail="Invalid token")
+
+    user = db.query(User).filter(User.user_email == email).first()
+    if user is None:
+        raise HTTPException(status_code=401, detail="User not found")
+
+    return user
+
+
+@router.post("/posts")
+def create_post(
+    post: PostCreate,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_auth_user),
+):
+    new_post = Post(
+        title=post.title, content=post.content, user_id=current_user.user_id
+    )
+    db.add(new_post)
+    db.commit()
+    db.refresh(new_post)
+    return {"message": "Post created", "post_id": new_post.post_id}
+
+
+@router.get("/list")
+def get_my_posts(
+    current_user: User = Depends(get_auth_user), db: Session = Depends(get_db)
+):
+    posts = db.query(Post).filter(Post.user_id == current_user.user_id).all()
+    return [{"post_id": post.post_id, "title": post.title} for post in posts]
+
+
+@router.get("/{post_id}")
+def read_user_post(
+    post_id: int,
+    current_user: User = Depends(get_auth_user),
+    db: Session = Depends(get_db),
+):
+    post = (
+        db.query(Post)
+        .filter(Post.post_id == post_id, Post.user_id == current_user.user_id)
+        .first()
+    )
+    if post is None:
+        raise HTTPException(status_code=404, detail="포스트가 없거나 권한이 없습니다.")
+    return {"post_id": post.post_id, "title": post.title, "content": post.content}
+
+@router.put("/save")
+def update_post(
+    post: PostUpdate,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_auth_user),
+):
+    db_post = (
+        db.query(Post)
+        .filter(Post.post_id == post.post_id, Post.user_id == current_user.user_id)
+        .first()
+    )
+    if not db_post:
+        raise HTTPException(status_code=404, detail="Post not found or access denied")
+
+    db_post.title = post.title
+    db_post.content = post.content
+    db.commit()
+    db.refresh(db_post)
+
+    return {"message": "Post updated successfully"}
+
+
+@router.delete("/{post_id}")
+def delete_post(
+    post_id: int, db: Session = Depends(get_db), user=Depends(get_auth_user)
+):
+    post = (
+        db.query(Post)
+        .filter(Post.post_id == post_id, Post.user_id == user.user_id)
+        .first()
+    )
+    if not post:
+        raise HTTPException(status_code=404, detail="Post not found")
+    db.delete(post)
+    db.commit()
+    return {"message": "Post deleted"}