PhotosAPI/requests/albums.py

import re
from os import makedirs, rename, sep
from shutil import rmtree
from typing import Union
from modules.utils import configGet
from modules.app import app, check_project_key, get_api_key
from modules.database import col_photos, col_albums
from bson.objectid import ObjectId
from bson.errors import InvalidId

from fastapi import HTTPException, Depends
from fastapi.responses import UJSONResponse, Response
from fastapi.openapi.models import APIKey
from starlette.status import HTTP_204_NO_CONTENT, HTTP_401_UNAUTHORIZED, HTTP_404_NOT_FOUND, HTTP_406_NOT_ACCEPTABLE, HTTP_409_CONFLICT

@app.post("/albums", response_class=UJSONResponse, include_in_schema=True)
async def album_create(name: str, title: str, apikey: APIKey = Depends(get_api_key)):

    if (check_project_key("photos", apikey)):

        if re.search(re.compile('^[a-z,0-9,_]*$'), name) is False:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name can only contain: a-z, 0-9 and _ characters.")
        
        if 2 > len(name) > 20:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name must be >2 and <20 characters.")

        if 2 > len(title) > 40:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album title must be >2 and <40 characters.")

        if col_albums.find_one( {"name": name} ) is not None:
            return HTTPException(status_code=HTTP_409_CONFLICT, detail=f"Album with name '{name}' already exists.")

        makedirs(f'{configGet("data_location")}{sep}photos{sep}images{sep}{name}', exist_ok=True)

        uploaded = col_albums.insert_one( {"name": name, "title": title} )

        return UJSONResponse(
            {
                "id": uploaded.inserted_id.__str__(),
                "name": name,
                "title": title
            }
        )

    else:
        raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail=configGet("key_invalid", "messages"))

@app.get("/albums", response_class=UJSONResponse, include_in_schema=True)
async def album_find(q: str, apikey: APIKey = Depends(get_api_key)):

    if (check_project_key("photos", apikey)):

        output = {"results": []}
        albums = list(col_albums.find( {"name": re.compile(q)} ))

        for album in albums:
            output["results"].append( {"id": album["_id"].__str__(), "name": album["name"]} )

        return UJSONResponse(output)

    else:
        raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail=configGet("key_invalid", "messages"))

@app.patch("/albums/{id}", response_class=UJSONResponse, include_in_schema=True)
async def album_patch(id: str, name: Union[str, None] = None, title: Union[str, None] = None, apikey: APIKey = Depends(get_api_key)):

    if (check_project_key("photos", apikey)):

        try:
            album = col_albums.find_one( {"_id": ObjectId(id)} )
            if album is None:
                raise InvalidId(id)
        except InvalidId:
            return HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Could not find an album with such id.")

        if title is not None:
            if 2 > len(title) > 40:
                return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album title must be >2 and <40 characters.")
        else:
            title = album["title"]

        if name is not None:
            if re.search(re.compile('^[a-z,0-9,_]*$'), name) is False:
                return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name can only contain: a-z, 0-9 and _ characters.")
            if 2 > len(name) > 20:
                return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name must be >2 and <20 characters.")
            rename(f'{configGet("data_location")}{sep}photos{sep}images{sep}{album["name"]}', f'{configGet("data_location")}{sep}photos{sep}images{sep}{name}')
            col_photos.update_many( {"album": album["name"]}, {"$set": {"album": name}} )
        else:
            name = album["name"]

        col_albums.update_one( {"_id": ObjectId(id)}, {"$set": {"name": name, "title": title}} )

        return UJSONResponse(
            {
                "name": name,
                "title": title
            }
        )

    else:
        raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail=configGet("key_invalid", "messages"))

@app.put("/albums/{id}", response_class=UJSONResponse, include_in_schema=True)
async def album_put(id: str, name: str, title: str, apikey: APIKey = Depends(get_api_key)):

    if (check_project_key("photos", apikey)):

        try:
            album = col_albums.find_one( {"_id": ObjectId(id)} )
            if album is None:
                raise InvalidId(id)
        except InvalidId:
            return HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Could not find an album with such id.")

        if re.search(re.compile('^[a-z,0-9,_]*$'), name) is False:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name can only contain: a-z, 0-9 and _ characters.")

        if 2 > len(name) > 20:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album name must be >2 and <20 characters.")

        if 2 > len(title) > 40:
            return HTTPException(status_code=HTTP_406_NOT_ACCEPTABLE, detail="Album title must be >2 and <40 characters.")

        rename(f'{configGet("data_location")}{sep}photos{sep}images{sep}{album["name"]}', f'{configGet("data_location")}{sep}photos{sep}images{sep}{name}')
        col_photos.update_many( {"album": album["name"]}, {"$set": {"album": name}} )

        col_albums.update_one( {"_id": ObjectId(id)}, {"$set": {"name": name, "title": title}} )

        return UJSONResponse(
            {
                "name": name,
                "title": title
            }
        )

    else:
        raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail=configGet("key_invalid", "messages"))

@app.delete("/album/{id}", response_class=UJSONResponse, include_in_schema=True)
async def album_delete(id: str, apikey: APIKey = Depends(get_api_key)):

    if (check_project_key("photos", apikey)):

        try:
            album = col_albums.find_one_and_delete( {"_id": ObjectId(id)} )
            if album is None:
                raise InvalidId(id)
        except InvalidId:
            return HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Could not find an album with such id.")
        
        col_photos.delete_many( {"album": album["name"]} )

        rmtree(f'{configGet("data_location")}{sep}photos{sep}images{sep}{album["name"]}')

        return Response(status_code=HTTP_204_NO_CONTENT)

    else:
        raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail=configGet("key_invalid", "messages"))