2022-12-14 18:19:31 +03:00
|
|
|
|
#!/usr/bin/python3
|
2022-12-14 19:24:12 +03:00
|
|
|
|
import time
|
2022-12-14 18:19:31 +03:00
|
|
|
|
import config_python
|
|
|
|
|
from twitchAPI.twitch import Twitch
|
2022-12-14 19:24:12 +03:00
|
|
|
|
import logging
|
|
|
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
|
|
|
from termcolor import colored
|
|
|
|
|
import os
|
|
|
|
|
import sys
|
2022-12-14 21:38:44 +03:00
|
|
|
|
import telebot
|
2022-12-14 23:12:26 +03:00
|
|
|
|
from telebot.types import InputFile
|
2022-12-14 21:59:20 +03:00
|
|
|
|
from datetime import datetime
|
2022-12-14 23:12:26 +03:00
|
|
|
|
import requests
|
|
|
|
|
import shutil
|
2022-12-14 21:38:44 +03:00
|
|
|
|
|
2022-12-14 19:24:12 +03:00
|
|
|
|
streamer = config_python.streamer
|
|
|
|
|
app_id = config_python.appid
|
|
|
|
|
app_secret = config_python.appsecret
|
2022-12-14 18:19:31 +03:00
|
|
|
|
|
2022-12-14 19:24:12 +03:00
|
|
|
|
log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
|
|
|
|
|
log_file = 'output.log'
|
2022-12-14 18:19:31 +03:00
|
|
|
|
|
|
|
|
|
|
2022-12-14 20:39:52 +03:00
|
|
|
|
def stream_message_worker(data: dict):
|
|
|
|
|
# Если это стрим, а не повтор, значит все в порядке
|
|
|
|
|
if data['type'] == 'live':
|
2022-12-14 21:38:44 +03:00
|
|
|
|
# Для удобства выносим в переменные
|
2022-12-14 20:39:52 +03:00
|
|
|
|
game_name = data['game_name']
|
|
|
|
|
stream_title = data['title']
|
|
|
|
|
stream_viewers = str(data['viewer_count'])
|
2022-12-14 21:59:20 +03:00
|
|
|
|
|
2022-12-14 23:12:26 +03:00
|
|
|
|
# Получаем скрин стрима
|
|
|
|
|
stream_pic_url = data['thumbnail_url'].format(width='1920',height='1080')
|
|
|
|
|
picture_file = requests.get(stream_pic_url, stream=True)
|
|
|
|
|
picture_file.raw.decode_content = True
|
|
|
|
|
# Увы, но я не придумал ничего лучше, чем сохранять фотку во временный файл и потом читать его
|
|
|
|
|
with open('tmppic.jpg', 'wb') as f:
|
|
|
|
|
picture_file.raw.decode_content = True
|
|
|
|
|
shutil.copyfileobj(picture_file.raw, f)
|
|
|
|
|
|
2022-12-14 21:59:20 +03:00
|
|
|
|
# Считаем длительность стрима
|
|
|
|
|
# '2022-12-14T17:26:47Z'
|
|
|
|
|
stream_started_at = str(data['started_at'])
|
|
|
|
|
time_started = datetime.strptime(stream_started_at, "%Y-%m-%dT%H:%M:%SZ")
|
|
|
|
|
time_now = datetime.utcnow().replace(microsecond=0)
|
|
|
|
|
stream_duration = time_now - time_started
|
2022-12-14 21:38:44 +03:00
|
|
|
|
# Готовим сообщение
|
2022-12-14 21:49:43 +03:00
|
|
|
|
message = stream_title + '\n' + \
|
2022-12-14 23:12:26 +03:00
|
|
|
|
"Игра: " + game_name + '\n' + \
|
|
|
|
|
"Зрителей: " + stream_viewers + '\n' + \
|
|
|
|
|
"Длительность: " + str(stream_duration)
|
2022-12-14 21:38:44 +03:00
|
|
|
|
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:
|
2022-12-14 23:12:26 +03:00
|
|
|
|
send_message = bot.send_photo(config_python.tgchat, InputFile('tmppic.jpg'), message)
|
2022-12-14 21:38:44 +03:00
|
|
|
|
print(send_message)
|
|
|
|
|
msgid = send_message.message_id
|
|
|
|
|
with open('msgid', 'w') as f:
|
|
|
|
|
f.write(str(msgid))
|
|
|
|
|
else:
|
|
|
|
|
try:
|
2022-12-14 23:12:26 +03:00
|
|
|
|
media = telebot.types.InputMediaPhoto(InputFile('tmppic.jpg'), message)
|
|
|
|
|
edit_msg_media = bot.edit_message_media(media, config_python.tgchat, msgid)
|
2022-12-14 21:38:44 +03:00
|
|
|
|
except telebot.apihelper.ApiTelegramException as e:
|
|
|
|
|
log.error(e)
|
2022-12-14 20:39:52 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Если стрим не идет
|
|
|
|
|
def no_stream_msg_worker():
|
|
|
|
|
log.info(streamer + " Не стримит")
|
2022-12-14 21:38:44 +03:00
|
|
|
|
# Проверяем наличие файла
|
|
|
|
|
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')
|
2022-12-14 20:39:52 +03:00
|
|
|
|
|
|
|
|
|
|
2022-12-14 19:24:12 +03:00
|
|
|
|
def check_alive():
|
|
|
|
|
"""
|
|
|
|
|
1. Проверка на наличие стрима
|
|
|
|
|
1.1 Если нет - удалить lock файл, если он есть
|
|
|
|
|
1.2 Если есть - создать lock файл, запустить записывалку
|
|
|
|
|
"""
|
|
|
|
|
|
2022-12-14 20:39:52 +03:00
|
|
|
|
# Получаем инфо о стримере, если не получается, выходим с ошибкой
|
2022-12-14 19:24:12 +03:00
|
|
|
|
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']
|
2022-12-14 20:39:52 +03:00
|
|
|
|
# Получаем стримы
|
2022-12-14 19:24:12 +03:00
|
|
|
|
user_stream = twitch_client.get_streams(user_id=user_id)
|
2022-12-14 20:39:52 +03:00
|
|
|
|
# Если стрим/повтор идет, то идем дальше
|
2022-12-14 19:24:12 +03:00
|
|
|
|
if user_stream['data']:
|
2022-12-14 20:39:52 +03:00
|
|
|
|
stream_message_worker(user_stream['data'][0])
|
2022-12-14 19:24:12 +03:00
|
|
|
|
else:
|
2022-12-14 20:39:52 +03:00
|
|
|
|
# Если стрим не идет, то запускаем функцию без параметров
|
|
|
|
|
no_stream_msg_worker()
|
2022-12-14 19:24:12 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_console_handler():
|
|
|
|
|
console_handler = logging.StreamHandler(sys.stdout)
|
|
|
|
|
console_handler.setFormatter(log_format)
|
|
|
|
|
return console_handler
|
2022-12-14 18:19:31 +03:00
|
|
|
|
|
|
|
|
|
|
2022-12-14 19:24:12 +03:00
|
|
|
|
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)
|
2022-12-14 21:38:44 +03:00
|
|
|
|
bot = telebot.TeleBot(config_python.tgtoken)
|
|
|
|
|
|
2022-12-14 19:24:12 +03:00
|
|
|
|
while True:
|
|
|
|
|
check_alive()
|
2022-12-14 23:12:26 +03:00
|
|
|
|
time.sleep(config_python.period)
|