api/api_config.py - master - outthevoid/python_telegram_bot_project

api/api_config.py
55 строк2 KB
VladimirChukavin
Добавил функцию search_by_budget.
10 сен 2025, 17:10
10 сен 2025, 17:10d09aeee
from typing import Any

from config_data.config import API_TOKEN
from api.api_utils.api_handler import APISearchInterface

url_api = "https://api.kinopoisk.dev/v1.4/movie"

method = "GET"

headers = {
    "accept": "application/json",
    "X-API-KEY": API_TOKEN,
}


def search_by_name(name: str, limit: int) -> dict[Any, Any]:
    """
    Функция для уменьшения количества передаваемых
    в функцию аргументов.
    :param name: Название фильма
    :param limit: Количество вариантов в ответе
    :return: Словарь с ответом сервера
    """
    search = APISearchInterface.search_movies_by_name()
    return search(method, url_api, headers, name, limit)


def search_by_rating(
    min_rating: float, max_rating: float, limit: int
) -> dict[Any, Any]:
    """
    Функция для уменьшения количества передаваемых
    в функцию аргументов.
    :param min_rating: Минимальный рейтинг
    :param max_rating: Максимальный рейтинг
    :param limit: Количество вариантов в ответе
    :return: Словарь с ответом сервера
    """
    search = APISearchInterface.search_movies_by_rating()
    return search(method, url_api, headers, min_rating, max_rating, limit)


def search_by_budget(
    limit: int, min_budget: int = None, max_budget: int = None
) -> dict[Any, Any]:
    """
    Функция для уменьшения количества передаваемых
    в функцию аргументов.
    :param min_budget: Минимальный бюджет
    :param max_budget: Максимальный бюджет
    :param limit: Количество вариантов в ответе
    :return: Словарь с ответом сервера
    """
    search = APISearchInterface.search_movies_by_budget()
    return search(method, url_api, headers, min_budget, max_budget, limit)