Compare commits

..

3 Commits

View File

@ -1,8 +1,9 @@
#!/usr/bin/python3
# TODO: Сделать проверку на наличие стримера
# FIXME: не создавать папки для несуществующих стримеров
# TODO: Сделать нормальную конфигурацию
# TODO: Автоматически удалять старые стримы
# TODO: сделать возможность добавлять свои параметры в cmdline к команде записи
import os
from threading import Thread
@ -11,6 +12,8 @@ import schedule
from twitch import TwitchClient
import subprocess
import time
streamers = config_python.streamers
client_id = config_python.twitchid
def which(command):
@ -50,7 +53,6 @@ def recorder(i):
path = config_python.path + "/"+ i
print("Записываем стрим %s\n" % i)
# cmdline для запуска youtube-dl
# TODO: сделать возможность добавлять свои параметры
cmdline = ["youtube-dl","https://twitch.tv/"+i]
s = subprocess.call(cmdline, stdout=subprocess.DEVNULL)
print("Запись стрима %s закончена\n" % i)
@ -59,8 +61,6 @@ def recorder(i):
print("lock файл удален")
def checkAlive():
streamers = config_python.streamers
client_id = config_python.twitchid
'''
1. Проверка на наличие стрима
1.1 Если нет - удалить lock файл, если он есть
@ -71,9 +71,8 @@ def checkAlive():
# Путь до диры со стримами
path = config_python.path + "/"+ i
# Создаем путь до диры со стримером, если его нет
if not (os.path.exists(config_python.path+"/"+i)):
if not (os.path.exists(path)):
os.makedirs(path)
# TODO: Сделать проверку на наличие стримера
user_id=client.users.translate_usernames_to_ids(i)[0]['id'] # Получить ID по нику
# Если стрим идет, то идем дальше
if client.streams.get_stream_by_user(user_id):
@ -94,11 +93,24 @@ def checkAlive():
def removeOldStreams():
# https://clck.ru/WHh32
pass
records_path = config_python.path
# По каждой папке со стримерами
for i in streamers:
if len(os.listdir(records_path+i)) > config_python.max_files:
# Получаем список файлов
# и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
# Если превышает - удаляем старейший
oldest = min(os.listdir(records_path+i), key=os.path.getctime)
os.unlink(oldest)
print("Удален файл: " + oldest)
if __name__ == "__main__":
# Проверить, установлены ли нужные утилиты
if not checkTools(): exit()
schedule.every(config_python.period).minutes.do(checkAlive)
# Проверять стримы раз в check_period
schedule.every(config_python.check_period).minutes.do(checkAlive)
# Каждый час удалять старые стримы
schedule.every(1).hours.do(removeOldStreams)
while True:
schedule.run_pending()
time.sleep(1)