module_26_fastapi/homework/main.py - master - mialdr/python_advanced

module_26_fastapi/homework/main.py
78 строк3 KB
Mikhail Druchek
module26 is solved
18 апр 2024, 23:23
18 апр 2024, 23:23163890b
import sys

from contextlib import asynccontextmanager

from fastapi import FastAPI, HTTPException, Depends
import uvicorn
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker

from database import engine, async_session
from crud import models, schemas
import crud


@asynccontextmanager
async def lifespan(app: FastAPI):
    async with engine.begin() as conn:
        await conn.run_sync(models.Base.metadata.drop_all)
        await conn.run_sync(models.Base.metadata.create_all)
        yield
        await engine.dispose()
        # await conn.run_sync(models.Base.metadata.drop_all)



app = FastAPI(
    lifespan=lifespan
)


async def get_db():
    db = async_session()
    try:
        yield db
    finally:
        await db.close()


@app.post('/dishes/', response_model=schemas.DishOutCreateDish)
async def create_dish(dish: schemas.DishIn, session: AsyncSession = Depends(get_db)) -> models.Dish:
    db_dish = await crud.get_dish_by_title(session, dish.title)
    if db_dish is not None:
        raise HTTPException(status_code=400, detail='Блюдо с таким названием уже есть')
    new_dish: models.Dish = await crud.create_dish(dish, session)
    return new_dish


@app.get('/dishes/', response_model=list[schemas.DishOutReadDishes], response_model_exclude={'title'})
async def read_dishes(session: AsyncSession = Depends(get_db)) -> list[models.Dish]:
    dishes = await crud.get_dishes(session)
    if not dishes:
        raise HTTPException(status_code=404, detail='В базеданных ничего нет')
    return dishes


@app.get('/dishes/{dish_id}', response_model=schemas.DishOutReadDish)
async def read_dish(dish_id: int, session: AsyncSession = Depends(get_db)) -> models.Dish:
    dish = await crud.get_dish_by_id(session, dish_id=dish_id, adding_number_views=True)
    if dish is None:
        raise HTTPException(status_code=404, detail='Блюдо не найдено')
    return dish


@app.patch('/dishes/{dish_id}', response_model=schemas.DishOutCreateDish)
async def update_dish(dish: schemas.DishUpdate, dish_id: int, session: AsyncSession = Depends(get_db)) -> models.Dish:
    dish = await crud.update_dish(dish, dish_id, session)
    if dish is None:
        raise HTTPException(status_code=404, detail='Блюдо не найдено')
    return dish


@app.delete('/dishes/{dish_id}')
async def delete_dish(dish_id, session: AsyncSession = Depends(get_db)):
    if await crud.del_dish(dish_id, session):
        return {'message': f'Блюдо {dish_id} успешно удалено'}
    raise HTTPException(status_code=404, detail='Блюдо не найдено')

if __name__ == '__main__':
    uvicorn.run(app, host="0.0.0.0", port=8000)