twitch-tgbot/main.py

#!/usr/bin/python3
import time
import config_python
from twitchAPI.twitch import Twitch
import logging
from logging.handlers import TimedRotatingFileHandler
from termcolor import colored
import os
import sys
import telebot

streamer = config_python.streamer
app_id = config_python.appid
app_secret = config_python.appsecret

log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
log_file = 'output.log'


def stream_message_worker(data: dict):
    # Если это стрим, а не повтор, значит все в порядке
    if data['type'] == 'live':
        # Для удобства выносим в переменные
        game_name = data['game_name']
        stream_title = data['title']
        stream_viewers = str(data['viewer_count'])
        # Готовим сообщение
        message = stream_title + '\n' + \
                  "Игра: " + game_name + '\n' + \
                  "Зрителей: " + stream_viewers
        log.info(message)
        # Если файл с айди существует, читаем его и редактируем сообщение из него
        if os.path.exists('msgid'):
            with open('msgid', 'r') as f:
                msgid = f.read()
            if msgid == '':
                msgid = 0
        else:
            msgid = 0

        if msgid == 0:
            send_message = bot.send_message(config_python.tgchat, message)
            print(send_message)
            msgid = send_message.message_id
            with open('msgid', 'w') as f:
                f.write(str(msgid))
        else:
            try:
                edit_msg = bot.edit_message_text(message, config_python.tgchat, int(msgid))
            except telebot.apihelper.ApiTelegramException as e:
                log.error(e)


# Если стрим не идет
def no_stream_msg_worker():
    log.info(streamer + " Не стримит")
    # Проверяем наличие файла
    if os.path.exists('msgid'):
        # Если файл существует - читаем
        with open('msgid', 'r') as f:
            msgid = f.read()
        if msgid == '':
            # Если пустой - удаляем
            os.remove('msgid')
        else:
            # Если не пустой, то удаляем сообщение по айдишнику
            try:
                # Удаляем сообщение
                edit_msg = bot.delete_message(config_python.tgchat, int(msgid))
            except telebot.apihelper.ApiTelegramException as e:
                log.error(e)
            # В конце удаляем айди сообщения
            os.remove('msgid')


def check_alive():
    """
    1. Проверка на наличие стрима
    1.1 Если нет - удалить lock файл, если он есть
    1.2 Если есть - создать lock файл, запустить записывалку
    """

    # Получаем инфо о стримере, если не получается, выходим с ошибкой
    resolved_id = twitch_client.get_users(logins=[streamer])
    if not resolved_id['data']:
        log.error(
            colored(
                "Аккаунт " + streamer + " не найден",
                'red',
            )
        )
    # Достаем ID стримера из инфо
    user_id = resolved_id['data'][0]['id']
    # Получаем стримы
    user_stream = twitch_client.get_streams(user_id=user_id)
    # Если стрим/повтор идет, то идем дальше
    if user_stream['data']:
        stream_message_worker(user_stream['data'][0])
    else:
        # Если стрим не идет, то запускаем функцию без параметров
        no_stream_msg_worker()


def get_console_handler():
    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setFormatter(log_format)
    return console_handler


def get_file_handler():
    file_handler = TimedRotatingFileHandler(log_file, when='midnight')
    file_handler.setFormatter(log_format)
    return file_handler


def get_logger(logger_name):
    logger = logging.getLogger(logger_name)
    logger.setLevel(logging.DEBUG)
    logger.addHandler(get_console_handler())
    logger.addHandler(get_file_handler())
    logger.propagate = False
    return logger


if __name__ == '__main__':
    log = get_logger("main")
    log.info("Запущен")
    twitch_client = Twitch(app_id, app_secret)
    bot = telebot.TeleBot(config_python.tgtoken)

    while True:
        check_alive()
        time.sleep(300)