resources.py

from databases import Database
from fastapi import APIRouter, Depends

from backend.db import get_db
from backend.models.dtos.mapping_badge_dto import (
    MappingBadgeDTO,
    MappingBadgeCreateDTO,
    MappingBadgeUpdateDTO,
    MappingBadgeListDTO,
    MappingBadgePublicListDTO,
)
from backend.models.dtos.user_dto import AuthUserDTO
from backend.services.mapping_badges import MappingBadgeService
from backend.services.users.authentication_service import pm_only

router = APIRouter(
    prefix="/badges",
    tags=["mapping_badges"],
    responses={404: {"description": "Not found"}},
)


@router.get("/")
async def get_mapping_badges(
    db: Database = Depends(get_db),
) -> MappingBadgeListDTO:
    """
    List mapping badges
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    return await MappingBadgeService.get_all(db)


@router.post("/")
async def create_mapping_badge(
    data: MappingBadgeCreateDTO,
    db: Database = Depends(get_db),
    user: AuthUserDTO = Depends(pm_only),
) -> MappingBadgeDTO:
    """
    Creates a new MappingBadge
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    return await MappingBadgeService.create(data, db)


@router.get("/{badge_id}/")
async def get_mapping_badge(
    badge_id: int,
    db: Database = Depends(get_db),
) -> MappingBadgeDTO:
    """
    Gets information for a single badge
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    return await MappingBadgeService.get_by_id(badge_id, db)


@router.patch("/{badge_id}/")
async def update_mapping_badge(
    data: MappingBadgeUpdateDTO,
    badge_id: int,
    db: Database = Depends(get_db),
    user: AuthUserDTO = Depends(pm_only),
) -> MappingBadgeDTO:
    """
    Updates a mapping badge
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    data.id = badge_id

    return await MappingBadgeService.update(data, db)


@router.delete("/{badge_id}/")
async def delete_mapping_badge(
    badge_id: int,
    db: Database = Depends(get_db),
    user: AuthUserDTO = Depends(pm_only),
):
    """
    Deletes a mapping badge
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    await MappingBadgeService.delete(badge_id, db)


@router.get("/user/{user_id}/")
async def get_for_user(
    user_id: int,
    db: Database = Depends(get_db),
) -> MappingBadgePublicListDTO:
    """
    Get all mapping badges that a user has earned
    ---
    tags:
      - badges
    produces:
      - application/json
    """
    return await MappingBadgeService.get_for_user(user_id, db)