feed WIP

src/api/swipes.py

@@ -0,0 +1,147 @@
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from psycopg2._psycopg import connection
+
+import db.swipes as db
+import settings.settings as settings
+from api.models import Group, Swipe, User
+from api.utils import get_current_user
+from db.feeds import get_groupname_by_feed_id
+from db.groups import get_group
+from db.internal import get_db_connection
+from db.memberships import check_membership_exists
+
+swipes_router = APIRouter(prefix="/api/swipes", tags=["swipes"])
+
+
+# Maybe endpoints should be remade
+# to return id and then work with swipe id
+
+@swipes_router.post("/swipe")
+async def read_swipe(
+    username: str,
+    feed_id: int,
+    picture_id: int,
+    conn: Annotated[connection, Depends(get_db_connection)],
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    groupname = get_groupname_by_feed_id(conn, feed_id)
+    if groupname is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such feed or feed is not linked to group",
+        )
+    if not check_membership_exists(conn, current_user.username, groupname) and current_user.role not in settings.settings.admin_roles:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not allowed",
+        )
+
+    swipe = Swipe()
+    swipe_data = db.get_swipe(conn, username, feed_id, picture_id)
+    if swipe_data is None:
+        return HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such swipe",
+        )
+    swipe.fill(swipe_data)
+    return swipe
+
+
+@swipes_router.post("/swipe/picture_id")
+async def read_swipes_by_picture_id(
+    feed_id: int,
+    picture_id: int,
+    conn: Annotated[connection, Depends(get_db_connection)],
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    groupname = get_groupname_by_feed_id(conn, feed_id)
+    if groupname is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such feed or feed is not linked to group",
+        )
+    if not check_membership_exists(conn, current_user.username, groupname) and current_user.role not in settings.settings.admin_roles:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not allowed",
+        )
+    return db.get_swipes_by_picture_id(conn, picture_id, feed_id)
+
+
+@swipes_router.post("/swipe/user")
+async def read_user_swipes(
+    username: str,
+    feed_id: int,
+    conn: Annotated[connection, Depends(get_db_connection)],
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    groupname = get_groupname_by_feed_id(conn, feed_id)
+    if groupname is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such feed or feed is not linked to group",
+        )
+    if not check_membership_exists(conn, current_user.username, groupname) and current_user.role not in settings.settings.admin_roles:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not allowed",
+        )
+    return db.get_swipes_by_user(conn, username, feed_id)
+
+
+@swipes_router.post("/add")
+async def add_swipe(
+    feed_id: int,
+    picture_id: int,
+    value: int,
+    conn: Annotated[connection, Depends(get_db_connection)],
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    groupname = get_groupname_by_feed_id(conn, feed_id)
+    if groupname is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such feed or feed is not linked to group",
+        )
+
+    group = Group()
+    group_data = get_group(conn, groupname)
+    if group_data is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No such feed or feed is not linked to group",
+        )
+    group.fill(group_data)
+
+    # Check for trying to skip in
+    # a group with skips disabled
+    if value == 0 and group.allow_skips is False:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Skips are disallowed",
+        )
+
+    result = db.create_swipe(conn, current_user.username, feed_id, picture_id, value)
+    if result:
+        # TODO: call function to like picture on the platform
+        pass
+    return result
+
+
+@swipes_router.post("/delete")
+async def delete_swipe(
+    username: str,
+    feed_id: int,
+    picture_id: int,
+    conn: Annotated[connection, Depends(get_db_connection)],
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    if current_user.role in settings.settings.admin_roles:
+        return db.delete_swipe(conn, username, feed_id, picture_id)
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not allowed",
+        )