fastapi_solution/main.py - main - vladmialkin/AsyncAPI

fastapi_solution/main.py
70 строк2 KB
vladmialkin
Merge branch 'dev' into new_conftest
18 авг 2024, 13:32
18 авг 2024, 13:32484aeb9
from contextlib import asynccontextmanager

from fastapi_solution.src.api.v1 import films, genres, persons
from fastapi_solution.src.api.v2 import film as films_v2
from fastapi_solution.src.api.v2 import genre as genres_v2
from fastapi_solution.src.api.v2 import person as persons_v2
from fastapi_solution.src.db import elastic, redis

import backoff
import asyncio

from elasticsearch import (
    AsyncElasticsearch,
    ConnectionError as ElasticConError,
    ConnectionTimeout as ElasticConnectionTimeout
)
from fastapi import FastAPI
from fastapi.responses import ORJSONResponse
from fastapi_pagination import add_pagination
from redis import ConnectionError as RedisConError, TimeoutError as RedisTimeoutError
from redis.asyncio import Redis

from fastapi_solution.src.core.config import settings



@backoff.on_exception(
    backoff.expo, (RedisConError, RedisTimeoutError), max_tries=settings.MAX_TRIES
)
async def setup_redis():
    redis.redis = Redis(host=settings.REDIS_HOST, port=settings.REDIS_PORT)
    await redis.redis.ping()


@backoff.on_exception(
    backoff.expo, (ElasticConError, ElasticConnectionTimeout), max_tries=settings.MAX_TRIES
)
async def setup_elasticsearch():
    elastic.es = AsyncElasticsearch(
        hosts=[f"http://{settings.ELASTIC_HOST}:{settings.ELASTIC_PORT}"]
    )
    await elastic.es.ping()


@asynccontextmanager
async def lifespan(app_):
    await asyncio.gather(setup_redis(), setup_elasticsearch())
    yield
    await redis.redis.close()
    await elastic.es.close()


app = FastAPI(
    lifespan=lifespan,
    title=settings.PROJECT_NAME,
    docs_url='/api/openapi',
    openapi_url='/api/openapi.json',
    default_response_class=ORJSONResponse,
    description="Информация о фильмах, жанрах и людях, участвовавших в создании произведения"
)

add_pagination(app)


app.include_router(films.router, prefix='/api/v1/films', tags=['films'])
app.include_router(genres.router, prefix='/api/v1/genres', tags=['genres'])
app.include_router(persons.router, prefix='/api/v1/persons', tags=['persons'])
app.include_router(persons_v2.router, prefix="/api/v2/persons", tags=["persons"])
app.include_router(films_v2.router, prefix="/api/v2/films", tags=["films"])
app.include_router(genres_v2.router, prefix="/api/v2/genres", tags=["genres"])