Feature/332 like dislike backend

backend/alembic/env.py

@@ -7,8 +7,8 @@
 from database import Base
 from stories import models as story_models
 from users import models as user_models
-
 from NytLiveCounty import models as nyt_models
+from likes import models as like_models
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.

backend/alembic/versions/7ecf335fdfc2_add_table_for_like_and_dislike.py

@@ -0,0 +1,41 @@
+"""add table for like and dislike
+
+Revision ID: 7ecf335fdfc2
+Revises: 527194b9fb18
+Create Date: 2020-10-13 09:42:37.480671
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = "7ecf335fdfc2"
+down_revision = "527194b9fb18"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "likes",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.Column("like", sa.Boolean(), nullable=True),
+        sa.Column("story_id", sa.Integer(), nullable=True),
+        sa.Column("liker_story_id", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(["liker_story_id"], ["stories.id"],),
+        sa.ForeignKeyConstraint(["story_id"], ["stories.id"],),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_likes_id"), "likes", ["id"], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f("ix_likes_id"), table_name="likes")
+    op.drop_table("likes")
+    # ### end Alembic commands ###

backend/likes/crud.py

@@ -0,0 +1,57 @@
+from sqlalchemy.orm import Session
+from . import models, schemas
+from sqlalchemy.sql.expression import and_
+from stories.crud import update
+
+
+def get_like_by_story_and_user(
+    db: Session, story_id: int, liker_story_id: int
+):
+    return (
+        db.query(models.Like)
+        .filter(
+            and_(
+                models.Like.story_id == story_id,
+                models.Like.liker_story_id == liker_story_id,
+            )
+        )
+        .first()
+    )
+
+
+def update_like(db, like_id, like: schemas.LikeCreate):
+    return update(like_id, like, models.Like, db)
+
+
+def create_like(db, like: schemas.LikeCreate, liker_story_id: int):
+    d = like.dict()
+    d["liker_story_id"] = liker_story_id
+    db_like = models.Like(**d)
+    db.add(db_like)
+    db.commit()
+    db.refresh(db_like)
+    return db_like
+
+
+def get_like_count(db, story_id):
+    return (
+        db.query(models.Like)
+        .filter(and_(models.Like.story_id == story_id, models.Like.like == 1))
+        .count()
+    )
+
+
+def get_dislike_count(db, story_id):
+    return (
+        db.query(models.Like)
+        .filter(and_(models.Like.story_id == story_id, models.Like.like == 0))
+        .count()
+    )
+
+
+def is_like_by(db, story_id, liker_story_id):
+    db_like = get_like_by_story_and_user(db, story_id, liker_story_id)
+    if db_like:
+        return db_like.like
+    else:
+        return None

backend/likes/models.py

@@ -0,0 +1,10 @@
+from database import Base
+from sqlalchemy import Column, ForeignKey, Integer, Boolean
+
+
+class Like(Base):
+    __tablename__ = "likes"
+
+    like = Column(Boolean)
+    story_id = Column(Integer, ForeignKey("stories.id"))
+    liker_story_id = Column(Integer, ForeignKey("stories.id"))

backend/likes/schemas.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel
+
+
+class LikeCreate(BaseModel):
+    like: bool = None
+    story_id: int
+
+
+class Like(LikeCreate):
+    liker_story_id: int
+
+    class Config:
+        orm_mode = True

backend/router/api.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter
 
-from router import auth, stories, users, symptoms, data, nyt_live_county
+from router import auth, stories, users, symptoms, data, nyt_live_county, likes
 
 router = APIRouter()
 
@@ -17,3 +17,5 @@
 router.include_router(
     nyt_live_county.router, prefix="/nyt_live_county", tags=["nyt_live_county"]
 )
+
+router.include_router(likes.router, prefix="/likes", tags=["likes"])

backend/router/likes.py

@@ -0,0 +1,68 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.orm import Session
+from starlette.responses import JSONResponse
+
+from auth import main
+from database import get_db
+from likes import crud, schemas
+from stories.crud import get_story
+from stories import schemas as stories_schemas
+
+router = APIRouter()
+
+
+@router.post("/", response_model=schemas.Like)
+async def create_like(
+    like: schemas.LikeCreate,
+    current_story: stories_schemas.Story = Depends(main.get_current_story),
+    db: Session = Depends(get_db),
+):
+    if not get_story(db, like.story_id):
+        raise HTTPException(
+            status_code=404,
+            detail="Target story cannot be found",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+
+    if not current_story:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="User haven't shared their story",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+
+    like_to_update = crud.get_like_by_story_and_user(
+        db, story_id=like.story_id, liker_story_id=current_story.id
+    )
+
+    if like_to_update:
+        return crud.update_like(db, like_to_update.id, like)
+    else:
+        if like.story_id == current_story.id:
+            raise HTTPException(
+                status_code=422,
+                detail="User cannot like or dislike their own stories",
+                headers={"WWW-Authenticate": "Bearer"},
+            )
+
+        return crud.create_like(db, like, current_story.id)
+
+
+@router.get("/{story_id}")
+async def get_like_count(
+    story_id: int,
+    current_story: stories_schemas.Story = Depends(main.get_current_story),
+    db: Session = Depends(get_db),
+):
+    like_count = crud.get_like_count(db, story_id)
+    dislike_count = crud.get_dislike_count(db, story_id)
+    is_like_by_me = crud.is_like_by(db, story_id, current_story.id)
+
+    return JSONResponse(
+        {
+            "like": like_count,
+            "dislike": dislike_count,
+            "like_by_me": is_like_by_me,
+        },
+        status_code=200,
+    )