homework.py

import logging
import os
import sys
import time
from http import HTTPStatus

import requests
from dotenv import load_dotenv
from telebot.apihelper import ApiException
from telebot import TeleBot

from exceptions import MissingEnvironmentVariableError, ApiRequestError

load_dotenv()

logger = logging.getLogger(__name__)


PRACTICUM_TOKEN = os.getenv('PRACTICUM_TOKEN')
TELEGRAM_TOKEN = os.getenv('TELEGRAM_TOKEN')
TELEGRAM_CHAT_ID = os.getenv('TELEGRAM_CHAT_ID')

RETRY_PERIOD = 600
ENDPOINT = 'https://practicum.yandex.ru/api/user_api/homework_statuses/'
HEADERS = {'Authorization': f'OAuth {PRACTICUM_TOKEN}'}


HOMEWORK_VERDICTS = {
    'approved': 'Работа проверена: ревьюеру всё понравилось. Ура!',
    'reviewing': 'Работа взята на проверку ревьюером.',
    'rejected': 'Работа проверена: у ревьюера есть замечания.'
}


def check_tokens():
    """Проверяет все необходимые переменные окружения."""
    tokens = {
        'PRACTICUM_TOKEN': PRACTICUM_TOKEN,
        'TELEGRAM_TOKEN': TELEGRAM_TOKEN,
        'TELEGRAM_CHAT_ID': TELEGRAM_CHAT_ID
    }
    missing_tokens = []
    for name, value in tokens.items():
        if not value:
            missing_tokens.append(name)
    if missing_tokens:
        error_msg = (
            'Отсутствуют обязательные переменные '
            f'окружения: {', '.join(missing_tokens)}. '
            'Программа принудительно остановлена.'
        )
        logger.critical(error_msg)
        raise MissingEnvironmentVariableError(error_msg)
    logger.info("Все необходимые переменные окружения доступны")


def send_message(bot, message):
    """Для отправки сообщения."""
    try:
        logger.info(
            'Начинается отправка сообщения в чат '
            f'{TELEGRAM_CHAT_ID}: {message}'
        )
        bot.send_message(chat_id=TELEGRAM_CHAT_ID, text=message)
        logger.debug('Сообщение успешно отправлено')
        return True
    except (ApiException, requests.exceptions.RequestException) as error:
        logger.error(f'Сообщение не удалось отправить, ошибка: {error}')
        return False


def get_api_answer(timestamp):
    """Проверяет доступность API сервиса."""
    params = {'from_date': timestamp}
    logger.info(
        f'Начинается запрос к API: '
        f'url={ENDPOINT}, заголовок={HEADERS}, параметры={params}'
    )

    try:
        response = requests.get(ENDPOINT, headers=HEADERS, params=params)
    except requests.exceptions.RequestException as e:
        error_msg = (
            f'Сбой в работе программы: Эндпоинт {ENDPOINT} '
            f'недоступен. Запрос вернул ошибку: {e}'
        )
        raise ApiRequestError(error_msg) from e

    if response.status_code != HTTPStatus.OK:
        error_msg = (
            f'Сбой в работе программы: Эндпоинт {ENDPOINT} '
            f'Код ответа API: {response.status_code}'
        )
        raise ApiRequestError(error_msg)

    return response.json()


def check_response(response):
    """Проверяем соответсвие параметров ответа из запроса."""
    if not isinstance(response, dict):
        error_msg = (
            f'Ответ API должен быть словарём. Сейчас это {type(response)}'
        )
        raise TypeError(error_msg)

    if 'homeworks' not in response:
        error_msg = 'В ответе API отсутствует обязательное поле: homeworks'
        raise KeyError(error_msg)

    homeworks_list = response['homeworks']

    if not isinstance(homeworks_list, list):
        error_msg = (
            'Поле homeworks должно быть типа list, '
            f'получено {type(homeworks_list).__name__}'
        )
        raise TypeError(error_msg)

    return homeworks_list


def parse_status(homework):
    """Проверяет соответсвие статусов с указанными в API сервисе яндекс."""
    for key in ['homework_name', 'status']:
        if key not in homework:
            error_msg = f'В ответе API нет ключа {key}'
            logger.error(error_msg)
            raise KeyError(error_msg)

    homework_name = homework['homework_name']
    homework_status = homework['status']
    if homework_status not in HOMEWORK_VERDICTS:
        error_msg = f'Статус "{homework_status}" не входит в ожидаемые.'
        logger.error(error_msg)
        raise ValueError(error_msg)
    verdict = HOMEWORK_VERDICTS[homework_status]
    return f'Изменился статус проверки работы "{homework_name}". {verdict}'


def main():
    """Основная логика работы бота."""
    check_tokens()

    bot = TeleBot(token=TELEGRAM_TOKEN)
    timestamp = int(time.time())
    # Переменная для исключения повторного сообщения об отправленной ошибке
    previous_error_message = None

    while True:
        try:
            response = get_api_answer(timestamp=timestamp)
            homeworks = check_response(response)

            # Начинает цикл заново если у домашки нет статуса
            if not homeworks:
                logger.debug(
                    'На текущий момент домашнюю работу не '
                    'взяли в проверку или она отсутствует'
                )
                continue
            new_status = parse_status(homework=homeworks[0])

            if send_message(bot=bot, message=new_status):
                timestamp = response.get('current_date', timestamp)

            previous_error_message = None

        except Exception as error:
            logger.error(error)
            message = 'Сбой в работе программы: ' + str(error)
            if previous_error_message != message:
                send_message(bot=bot, message=message)
                previous_error_message = message
        finally:
            time.sleep(RETRY_PERIOD)


if __name__ == '__main__':

    logging.basicConfig(
        level=logging.INFO,
        format='%(asctime)s - %(levelname)s - %(message)s',
        stream=sys.stdout
    )

    main()