updated

Author: wpcodevo

alembic/versions/4917da928a79_added_post_table.py

@@ -0,0 +1,39 @@
+"""added post table
+
+Revision ID: 4917da928a79
+Revises: 39256113e8e5
+Create Date: 2022-07-14 09:05:17.444518
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = '4917da928a79'
+down_revision = '39256113e8e5'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('posts',
+    sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
+    sa.Column('user_id', postgresql.UUID(as_uuid=True), nullable=False),
+    sa.Column('title', sa.String(), nullable=False),
+    sa.Column('content', sa.String(), nullable=False),
+    sa.Column('category', sa.String(), nullable=False),
+    sa.Column('image', sa.String(), nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('posts')
+    # ### end Alembic commands ###

app/main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from app.config import settings
-from app.routers import user, auth
+from app.routers import user, auth, post
 
 app = FastAPI()
 
@@ -20,6 +20,7 @@
 
 app.include_router(auth.router, tags=['Auth'], prefix='/api/auth')
 app.include_router(user.router, tags=['Users'], prefix='/api/users')
+app.include_router(post.router, tags=['Posts'], prefix='/api/posts')
 
 
 @app.get('/api/healthchecker')

app/models.py

@@ -1,8 +1,8 @@
-from enum import unique
 import uuid
 from .database import Base
-from sqlalchemy import TIMESTAMP, Column, String, Boolean, text
+from sqlalchemy import TIMESTAMP, Column, ForeignKey, String, Boolean, text
 from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import relationship
 
 
 class User(Base):
@@ -20,3 +20,20 @@ class User(Base):
                         nullable=False, server_default=text("now()"))
     updated_at = Column(TIMESTAMP(timezone=True),
                         nullable=False, server_default=text("now()"))
+
+
+class Post(Base):
+    __tablename__ = 'posts'
+    id = Column(UUID(as_uuid=True), primary_key=True, nullable=False,
+                default=uuid.uuid4)
+    user_id = Column(UUID(as_uuid=True), ForeignKey(
+        'users.id', ondelete='CASCADE'), nullable=False)
+    title = Column(String, nullable=False)
+    content = Column(String, nullable=False)
+    category = Column(String, nullable=False)
+    image = Column(String, nullable=False)
+    created_at = Column(TIMESTAMP(timezone=True),
+                        nullable=False, server_default=text("now()"))
+    updated_at = Column(TIMESTAMP(timezone=True),
+                        nullable=False, server_default=text("now()"))
+    user = relationship('User')

app/routers/post.py

@@ -0,0 +1,68 @@
+import uuid
+from .. import schemas, models
+from sqlalchemy.orm import Session
+from fastapi import Depends, HTTPException, status, APIRouter, Response
+from ..database import get_db
+from app.oauth2 import require_user
+
+router = APIRouter()
+
+
+@router.get('/', response_model=schemas.ListPostResponse)
+def get_posts(db: Session = Depends(get_db), limit: int = 10, page: int = 1, search: str = '', user_id: str = Depends(require_user)):
+    skip = (page - 1) * limit
+
+    posts = db.query(models.Post).group_by(models.Post.id).filter(
+        models.Post.title.contains(search)).limit(limit).offset(skip).all()
+    return {'status': 'success', 'results': len(posts), 'posts': posts}
+
+
+@router.post('/', status_code=status.HTTP_201_CREATED, response_model=schemas.PostResponse)
+def create_post(post: schemas.CreatePostSchema, db: Session = Depends(get_db), owner_id: str = Depends(require_user)):
+    post.user_id = uuid.UUID(owner_id)
+    new_post = models.Post(**post.dict())
+    db.add(new_post)
+    db.commit()
+    db.refresh(new_post)
+    return new_post
+
+
+@router.put('/{id}', response_model=schemas.PostResponse)
+def update_post(id: str, post: schemas.UpdatePostSchema, db: Session = Depends(get_db), user_id: str = Depends(require_user)):
+    post_query = db.query(models.Post).filter(models.Post.id == id)
+    updated_post = post_query.first()
+
+    if not updated_post:
+        raise HTTPException(status_code=status.HTTP_200_OK,
+                            detail=f'No post with this id: {id} found')
+    if updated_post.user_id != uuid.UUID(user_id):
+        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN,
+                            detail='You are not allowed to perform this action')
+    post_query.update(post.dict(), synchronize_session=False)
+    db.commit()
+    return updated_post
+
+
+@router.get('/{id}', response_model=schemas.PostResponse)
+def get_post(id: str, db: Session = Depends(get_db), user_id: str = Depends(require_user)):
+    post = db.query(models.Post).filter(models.Post.id == id).first()
+    if not post:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
+                            detail=f"No post with this id: {id} found")
+    return post
+
+
+@router.delete('/{id}')
+def delete_post(id: str, db: Session = Depends(get_db), user_id: str = Depends(require_user)):
+    post_query = db.query(models.Post).filter(models.Post.id == id)
+    post = post_query.first()
+    if not post:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
+                            detail=f'No post with this id: {id} found')
+
+    if post.owner_id != user_id:
+        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN,
+                            detail='You are not allowed to perform this action')
+    post_query.delete(synchronize_session=False)
+    db.commit()
+    return Response(status_code=status.HTTP_204_NO_CONTENT)

app/schemas.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from typing import List
 import uuid
 from pydantic import BaseModel, EmailStr, constr
 
@@ -31,4 +32,43 @@ class UserResponse(UserBaseSchema):
 
 
 class FilteredUserResponse(UserBaseSchema):
+    id: uuid.UUID
+
+
+class PostBaseSchema(BaseModel):
+    title: str
+    content: str
+    category: str
+    image: str
+    user_id: uuid.UUID | None = None
+
+    class Config:
+        orm_mode = True
+
+
+class CreatePostSchema(PostBaseSchema):
     pass
+
+
+class PostResponse(PostBaseSchema):
+    id: uuid.UUID
+    user: FilteredUserResponse
+    created_at: datetime
+    updated_at: datetime
+
+
+class UpdatePostSchema(BaseModel):
+    title: str | None = None
+    content: str | None = None
+    category: str | None = None
+    image: str | None = None
+    user_id: uuid.UUID | None = None
+
+    class Config:
+        orm_mode = True
+
+
+class ListPostResponse(BaseModel):
+    status: str
+    results: int
+    posts: List[PostResponse]