2021-07-18 03:06:08 +03:00
|
|
|
|
#!/usr/bin/python3
|
|
|
|
|
|
|
|
|
|
# база со стримерами в json файле
|
|
|
|
|
|
2021-07-18 05:39:06 +03:00
|
|
|
|
import os
|
2021-07-18 03:06:08 +03:00
|
|
|
|
from threading import Thread
|
|
|
|
|
import json
|
|
|
|
|
import config_python
|
|
|
|
|
|
2021-07-18 05:39:06 +03:00
|
|
|
|
locked_streams = list()
|
2021-07-18 03:06:08 +03:00
|
|
|
|
|
|
|
|
|
def startRecord(i):
|
|
|
|
|
'''
|
|
|
|
|
Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
|
|
|
|
|
'''
|
|
|
|
|
th = Thread(target=recorder, args=(i, ))
|
|
|
|
|
th.start()
|
|
|
|
|
|
|
|
|
|
def recorder(i):
|
|
|
|
|
'''
|
|
|
|
|
Функция, которая запускает youtube-dl, фактически записывает стрим
|
|
|
|
|
'''
|
|
|
|
|
print("Записываем стрим %s\n" % i)
|
|
|
|
|
# FIXME: пофиксить абсолютный путь
|
2021-07-18 05:39:06 +03:00
|
|
|
|
cmdline = ["/home/losted/.local/bin/youtube-dl","https://twitch.tv/"+i]
|
2021-07-18 03:06:08 +03:00
|
|
|
|
import subprocess
|
2021-07-18 04:40:48 +03:00
|
|
|
|
s = subprocess.call(cmdline, stdout=subprocess.DEVNULL)
|
2021-07-18 03:06:08 +03:00
|
|
|
|
print("Запись стрима %s закончена\n" % i)
|
2021-07-18 05:39:06 +03:00
|
|
|
|
os.system("rm "+config_python.path + "/"+i+"/pid")
|
|
|
|
|
print("lock файл удален")
|
2021-07-18 03:06:08 +03:00
|
|
|
|
|
|
|
|
|
def checkAlive(streamers, client_id):
|
|
|
|
|
'''
|
|
|
|
|
1. Проверка на наличие стрима
|
|
|
|
|
1.1 Если нет - удалить lock файл, если он есть
|
|
|
|
|
1.2 Если есть - создать lock файл, запустить записывалку
|
|
|
|
|
'''
|
|
|
|
|
from twitch import TwitchClient
|
|
|
|
|
client = TwitchClient(client_id=client_id)
|
|
|
|
|
for i in streamers:
|
2021-07-18 05:39:06 +03:00
|
|
|
|
# Путь до диры со стримами
|
|
|
|
|
path = config_python.path + "/"+ i
|
|
|
|
|
# Создаем путь, если его нет
|
|
|
|
|
if not (os.path.exists(config_python.path+"/"+i)):
|
|
|
|
|
os.makedirs(path)
|
|
|
|
|
# TODO: Сделать проверку на наличие стримера
|
|
|
|
|
user_id=client.users.translate_usernames_to_ids(i)[0]['id'] # Получить ID по нику
|
|
|
|
|
# Если стрим идет, то идем дальше
|
2021-07-18 03:06:08 +03:00
|
|
|
|
if client.streams.get_stream_by_user(user_id):
|
2021-07-18 05:39:06 +03:00
|
|
|
|
# Если стрим идет и лок файла нет, то записываем и ставим лок
|
|
|
|
|
if (client.streams.get_stream_by_user(user_id).stream_type == 'live') and not (os.path.exists(config_python.path+"/"+i+"/pid")):
|
|
|
|
|
print(i+" стримит")
|
2021-07-18 03:06:08 +03:00
|
|
|
|
startRecord(i)
|
2021-07-18 05:39:06 +03:00
|
|
|
|
os.system("touch "+path+"/pid")
|
|
|
|
|
else:
|
|
|
|
|
print(i+" Уже стримит")
|
2021-07-18 03:06:08 +03:00
|
|
|
|
else:
|
2021-07-18 05:39:06 +03:00
|
|
|
|
# Если стрим не идет, то пишем об этом и убираем его из залоченных
|
2021-07-18 03:06:08 +03:00
|
|
|
|
print(i+" Не стримит")
|
2021-07-18 05:39:06 +03:00
|
|
|
|
os.system("rm "+path+"/pid")
|
|
|
|
|
|
2021-07-18 03:06:08 +03:00
|
|
|
|
|
2021-07-18 03:32:01 +03:00
|
|
|
|
def removeOldStreams():
|
|
|
|
|
pass
|
2021-07-18 03:06:08 +03:00
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
checkAlive(config_python.streamers, config_python.twitchid)
|