router.py

import uuid
from typing import List
import logging
from fastapi import APIRouter, Depends, Request, Response, status, HTTPException
from fastapi.responses import RedirectResponse
from sqlalchemy.orm import Session

import app.api.v1.endpoints.beverage.crud as beverage_crud
from app.api.v1.endpoints.beverage.schemas import BeverageSchema, BeverageCreateSchema, BeverageListItemSchema
from app.database.connection import SessionLocal


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


router = APIRouter()


@router.get('', response_model=List[BeverageListItemSchema], tags=['beverage'])
def get_all_beverages(db: Session = Depends(get_db)):
    beverages = beverage_crud.get_all_beverages(db)
    return beverages


@router.post('', response_model=BeverageSchema, status_code=status.HTTP_201_CREATED, tags=['beverage'])
def create_beverage(beverage: BeverageCreateSchema,
                    request: Request,
                    db: Session = Depends(get_db)):
    beverage_found = beverage_crud.get_beverage_by_name(beverage.name, db)

    if beverage_found:
        logging.info('Beverage {} already exists (redirect)'.format(beverage.name))
        url = request.url_for('get_beverage', beverage_id=beverage_found.id)
        return RedirectResponse(url=url, status_code=status.HTTP_303_SEE_OTHER)

    new_beverage = beverage_crud.create_beverage(beverage, db)
    logging.info('Create Beverage {}'.format(beverage.name))
    return new_beverage


@router.put('/{beverage_id}', response_model=BeverageSchema, tags=['beverage'])
def update_beverage(
        beverage_id: uuid.UUID,
        changed_beverage: BeverageCreateSchema,
        request: Request,
        response: Response,
        db: Session = Depends(get_db),
):
    beverage_found = beverage_crud.get_beverage_by_id(beverage_id, db)

    if beverage_found:
        if beverage_found.name == changed_beverage.name:
            logging.info('Update Beverage {}'.format(beverage_found.name))
            beverage_crud.update_beverage(beverage_found, changed_beverage, db)
            return Response(status_code=status.HTTP_204_NO_CONTENT)
        else:
            beverage_name_found = beverage_crud.get_beverage_by_name(changed_beverage.name, db)
            if beverage_name_found:
                logging.info('Update Beverage {} (Redirect)'.format(changed_beverage.name))
                url = request.url_for('get_beverage', beverage_id=beverage_name_found.id)
                return RedirectResponse(url=url, status_code=status.HTTP_303_SEE_OTHER)
            else:
                updated_beverage = beverage_crud.create_beverage(changed_beverage, db)
                logging.info('Update Beverage {} (Create)'.format(updated_beverage.name))
                response.status_code = status.HTTP_201_CREATED
    else:
        logging.error('Cannot Update Beverage {}, Beverage does not exist'.format(beverage_id))
        raise HTTPException(status_code=404, detail='Item not found')

    return updated_beverage


@router.get('/{beverage_id}', response_model=BeverageSchema, tags=['beverage'])
def get_beverage(
        beverage_id: uuid.UUID,
        db: Session = Depends(get_db),
):
    beverage = beverage_crud.get_beverage_by_id(beverage_id, db)

    if not beverage:
        logging.error('Cannot GET Beverage {}, Beverage does not exist'.format(beverage_id))
        raise HTTPException(status_code=404, detail='Item not found')

    logging.info('GET Beverage {}'.format(beverage.id))
    return beverage


@router.delete('/{beverage_id}', response_model=None, tags=['beverage'])
def delete_beverage(
        beverage_id: uuid.UUID,
        db: Session = Depends(get_db)):
    beverage = beverage_crud.get_beverage_by_id(beverage_id, db)

    if not beverage:
        logging.error('Cannot Delete Beverage {}, Beverage does not exist'.format(beverage_id))
        raise HTTPException(status_code=404, detail='Item not found')

    logging.info('Delete Beverage {}'.format(beverage.id))
    beverage_crud.delete_beverage_by_id(beverage_id, db)
    return Response(status_code=status.HTTP_204_NO_CONTENT)