learnhouse/src/services/courses/activities/activities.py

from pydantic import BaseModel
from src.security.security import verify_user_rights_with_roles
from src.services.users.schemas.users import PublicUser
from fastapi import HTTPException, status, Request
from uuid import uuid4
from datetime import datetime

#### Classes ####################################################


class Activity(BaseModel):
    name: str
    type: str
    content: object


class ActivityInDB(Activity):
    activity_id: str
    course_id: str
    coursechapter_id: str
    org_id: str
    creationDate: str
    updateDate: str


#### Classes ####################################################


####################################################
# CRUD
####################################################


async def create_activity(
    request: Request,
    activity_object: Activity,
    org_id: str,
    coursechapter_id: str,
    current_user: PublicUser,
):
    activities = request.app.db["activities"]
    courses = request.app.db["courses"]

    # generate activity_id
    activity_id = str(f"activity_{uuid4()}")

    hasRoleRights = await verify_user_rights_with_roles(
        request, "create", current_user.user_id, activity_id, org_id
    )

    # get course_id from activity
    course = await courses.find_one({"chapters": coursechapter_id})

    if not hasRoleRights:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT,
            detail="Roles : Insufficient rights to perform this action",
        )

    # create activity
    activity = ActivityInDB(
        **activity_object.dict(),
        creationDate=str(datetime.now()),
        coursechapter_id=coursechapter_id,
        updateDate=str(datetime.now()),
        activity_id=activity_id,
        org_id=org_id,
        course_id=course["course_id"],
    )
    await activities.insert_one(activity.dict())

    # update chapter
    await courses.update_one(
        {"chapters_content.coursechapter_id": coursechapter_id},
        {"$addToSet": {"chapters_content.$.activities": activity_id}},
    )

    return activity


async def get_activity(request: Request, activity_id: str, current_user: PublicUser):
    activities = request.app.db["activities"]
    courses = request.app.db["courses"]

    activity = await activities.find_one({"activity_id": activity_id})

    # get course_id from activity
    coursechapter_id = activity["coursechapter_id"]
    course = await courses.find_one({"chapters": coursechapter_id})

    isCoursePublic = course["public"]
    isAuthor = current_user.user_id in course["authors"]

    if isAuthor:
        activity = ActivityInDB(**activity)
        return activity

    # verify course rights
    hasRoleRights = await verify_user_rights_with_roles(
        request,
        "read",
        current_user.user_id,
        activity_id,
        element_org_id=activity["org_id"],
    )

    if not hasRoleRights and not isCoursePublic:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT,
            detail="Roles : Insufficient rights to perform this action",
        )

    if not activity:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT, detail="Course does not exist"
        )

    activity = ActivityInDB(**activity)
    return activity


async def update_activity(
    request: Request,
    activity_object: Activity,
    activity_id: str,
    current_user: PublicUser,
):
    activities = request.app.db["activities"]

    activity = await activities.find_one({"activity_id": activity_id})
    # verify course rights
    await verify_user_rights_with_roles(
        request,
        "update",
        current_user.user_id,
        activity_id,
        element_org_id=activity["org_id"],
    )

    if activity:
        creationDate = activity["creationDate"]

        # get today's date
        datetime_object = datetime.now()

        updated_course = ActivityInDB(
            activity_id=activity_id,
            coursechapter_id=activity["coursechapter_id"],
            creationDate=creationDate,
            updateDate=str(datetime_object),
            org_id=activity["org_id"],
            **activity_object.dict(),
        )

        await activities.update_one(
            {"activity_id": activity_id}, {"$set": updated_course.dict()}
        )

        return ActivityInDB(**updated_course.dict())

    else:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT, detail="activity does not exist"
        )


async def delete_activity(request: Request, activity_id: str, current_user: PublicUser):
    activities = request.app.db["activities"]

    activity = await activities.find_one({"activity_id": activity_id})

    # verify course rights
    await verify_user_rights_with_roles(
        request,
        "delete",
        current_user.user_id,
        activity_id,
        element_org_id=activity["org_id"],
    )

    if not activity:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT, detail="activity does not exist"
        )

    isDeleted = await activities.delete_one({"activity_id": activity_id})

    if isDeleted:
        return {"detail": "activity deleted"}
    else:
        raise HTTPException(
            status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
            detail="Unavailable database",
        )


####################################################
# Misc
####################################################


async def get_activities(
    request: Request, coursechapter_id: str, current_user: PublicUser
):
    activities = request.app.db["activities"]

    activities = activities.find({"coursechapter_id": coursechapter_id})

    if not activities:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT, detail="Course does not exist"
        )

    activities = [
        ActivityInDB(**activity) for activity in await activities.to_list(length=100)
    ]

    return activities