homework.py

import os
from sqlite3 import Timestamp
import sys
import logging
import requests
import time
from telegram import Bot
from dotenv import load_dotenv

load_dotenv()

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

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler(sys.stdout)
logger.addHandler(handler)
formatter = logging.Formatter(
    '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)


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


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


def send_message(bot, message):
    """Посылает сообщение от бота в чат с TELEGRAM_CHAT_ID."""
    try:
        bot.send_message(chat_id=TELEGRAM_CHAT_ID, text=message)
    except Exception as error:
        logger.error(error)
    else:
        logger.info(f'Отправлено сообщение: "{message}"')


def get_api_answer(current_timestamp):
    """Получает json со списком работ."""
    timestamp = current_timestamp or int(time.time())
    params = {'from_date': timestamp}
    try:
        response = requests.get(ENDPOINT, headers=HEADERS, params=params)
    except Exception as error:
        logger.error(error)
    else:
        if response.status_code != 200:
            raise OSError("Response " + str(response.status_code)
                          + ": " + response.content)
        else:
            return response.json()


def check_response(response):
    """Проверяет наличие работы в ответе от сервера."""
    homeworks = response['homeworks']
    if homeworks is None:
        logger.error('В ответе сервера нет данных о работе.')
    if isinstance(homeworks, list):
        return homeworks


def parse_status(homework):
    """Выносит вердикт о проверке работы."""
    homework_name = homework.get('homework_name')
    homework_status = homework.get('status')
    if homework_name is None:
        logger.error('В ответе сервера нет названия домашней работы.')
    if homework_status is None:
        logger.error('В ответе сервера нет статуса домашней работы.')
    verdict = HOMEWORK_STATUSES[homework_status]
    return f'Изменился статус проверки работы "{homework_name}". {verdict}'


def check_tokens():
    """Проверяет доступность токенов."""
    tokens = {
        'PRACTICUM_TOKEN': PRACTICUM_TOKEN,
        'TELEGRAM_TOKEN': TELEGRAM_TOKEN,
        'TELEGRAM_CHAT_ID': TELEGRAM_CHAT_ID,
    }
    for token, value in tokens.items():
        if not value:
            logger.critical(
                f'Отсутствует обязательная переменная окружения: {token}'
            )
            return False
    return True


def main():
    """Основная логика работы бота."""
    check_tokens_result = check_tokens()
    if not check_tokens_result:
        exit()
    bot = Bot(token=TELEGRAM_TOKEN)
    homework_preload_id = 0
    homework_preload_status = 'initial'
    while True:
        try:
            # current_timestamp = int(time.time())
            timestamp = 1652289700
            response = get_api_answer(timestamp)
            homeworks = check_response(response)
            homework = homeworks[0]
            status_received = homework['status']
            if (homework_preload_status != status_received
                    or homework_preload_id != homework['id']):
                verdict = parse_status(homework)
                send_message(bot, verdict)
                homework_preload_status = status_received
            else:
                logger.debug('Статус работы не обновлен.')
            if homework_preload_id != homework['id']:
                homework_preload_id = homework['id']
                homework_preload_status = status_received
        except Exception as error:
            message = f'Сбой в работе программы: {error}'
            logger.error(message)
            send_message(bot, message)
        else:
            current_timestamp = response['current_date']
        finally:
            time.sleep(RETRY_TIME)


if __name__ == '__main__':
    main()