lulzette/twitch-autosave
1
1
Fork 0
Code Issues 5 Pull Requests Releases 2 Wiki Activity

[#12] Заменил "i" на человекочитаемые переменные

Browse Source
This commit is contained in:
lulzette 2023-06-02 00:18:58 +03:00
parent 537af2c22e
commit d66c0dfc57
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
daemon.py
View File

@ -59,33 +59,33 @@ def check_installed_tools() -> bool:
Проверяет, установлены ли необходимые утилиты Проверяет, установлены ли необходимые утилиты
""" """
tools = ('youtube-dl', 'ffmpeg') tools = ('youtube-dl', 'ffmpeg')
for i in tools: for tool in tools:
if not which(i): if not which(tool):
log.critical(i + " не установлен") log.critical(tool + " не установлен")
return False return False
return True return True
def start_recording(i): def start_recording(streamer):
""" """
Функция, которая запускает в отдельном потоке запись стрима - recorder(i) Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
""" """
th = Thread(target=recorder, args=(i,)) th = Thread(target=recorder, args=(streamer,))
th.start() th.start()
def recorder(i): def recorder(streamer):
""" """
Функция, которая запускает youtube-dl, фактически записывает стрим Функция, которая запускает youtube-dl, фактически записывает стрим
""" """
path = config['app']['path'] + "/" + i path = config['app']['path'] + "/" + streamer
log.info("Записываем стрим %s\n" % i) log.info("Записываем стрим %s\n" % streamer)
# cmdline для запуска youtube-dl # cmdline для запуска youtube-dl
cmdline = ["youtube-dl", "-q", "-o", cmdline = ["youtube-dl", "-q", "-o",
path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s", path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
"https://twitch.tv/" + i] "https://twitch.tv/" + streamer]
subprocess.call(cmdline) subprocess.call(cmdline)
log.info("Запись стрима %s закончена\n" % i) log.info("Запись стрима %s закончена\n" % streamer)
if os.path.exists(path + "/pid"): if os.path.exists(path + "/pid"):
os.remove(path + "/pid") os.remove(path + "/pid")
log.info("lock файл удален") log.info("lock файл удален")
@ -98,22 +98,22 @@ def check_stream():
1.1 Если нет - удалить lock файл, если он есть 1.1 Если нет - удалить lock файл, если он есть
1.2 Если есть - создать lock файл, запустить записывалку 1.2 Если есть - создать lock файл, запустить записывалку
""" """
for i in config['twitch']['streamers'].split(','): for streamer in config['twitch']['streamers'].split(','):
# Путь до диры со стримами # Путь до диры со стримами
path = config['app']['path'] + "/" + i path = config['app']['path'] + "/" + streamer
# Получаем инфо о стримере, если не получается, выходим с ошибкой # Получаем инфо о стримере, если не получается, выходим с ошибкой
# resolved_id = client.users.translate_usernames_to_ids(i) # resolved_id = client.users.translate_usernames_to_ids(i)
resolved_id = twitch_client.get_users(logins=[i]) resolved_id = twitch_client.get_users(logins=[streamer])
if not resolved_id['data']: if not resolved_id['data']:
log.error( log.error(
"Аккаунт " + i + " не найден" "Аккаунт " + streamer + " не найден"
) )
continue continue
# Создаем путь до диры со стримером, если папка не существует # Создаем путь до диры со стримером, если папка не существует
if not (os.path.exists(path)): if not (os.path.exists(path)):
os.makedirs(path) os.makedirs(path)
log.info("Создана директория " + i) log.info("Создана директория " + streamer)
# Достаем ID стримера из инфо # Достаем ID стримера из инфо
user_id = resolved_id['data'][0]['id'] user_id = resolved_id['data'][0]['id']
user_stream = twitch_client.get_streams(user_id=user_id) user_stream = twitch_client.get_streams(user_id=user_id)
@ -121,17 +121,17 @@ def check_stream():
if user_stream['data']: if user_stream['data']:
# Если стрим идет и лок файла нет, то записываем и ставим лок # Если стрим идет и лок файла нет, то записываем и ставим лок
if (user_stream['data'][0]['type'] == 'live') and not ( if (user_stream['data'][0]['type'] == 'live') and not (
os.path.exists(config['app']['path'] + "/" + i + "/pid")): os.path.exists(config['app']['path'] + "/" + streamer + "/pid")):
log.info(i + " стримит") log.info(streamer + " стримит")
start_recording(i) start_recording(streamer)
open(config['app']['path'] + "/pid", 'w').close open(config['app']['path'] + "/pid", 'w').close
else: else:
log.info( log.info(
"Идет запись " + i "Идет запись " + streamer
) )
else: else:
# Если стрим не идет, то пишем об этом и убираем его из залоченных # Если стрим не идет, то пишем об этом и убираем его из залоченных
log.info(i + " Не стримит") log.info(streamer + " Не стримит")
# Если есть лок, то удаляем # Если есть лок, то удаляем
if os.path.exists(path + "/pid"): if os.path.exists(path + "/pid"):
os.remove(path + "/pid") os.remove(path + "/pid")
@ -141,15 +141,15 @@ def remove_old_streams():
# https://clck.ru/WHh32 # https://clck.ru/WHh32
records_path = config['app']['path'] records_path = config['app']['path']
# По каждой папке со стримерами # По каждой папке со стримерами
for i in config['twitch']['streamers']: for streamer in config['twitch']['streamers']:
try: try:
os.chdir(records_path + "/" + i) os.chdir(records_path + "/" + streamer)
# Если файлов в папке со стримами больше чем указано в конфиге # Если файлов в папке со стримами больше чем указано в конфиге
if len(os.listdir(records_path + "/" + i)) > int(config['app']['max_files']): if len(os.listdir(records_path + "/" + streamer)) > int(config['app']['max_files']):
# Получаем список файлов # Получаем список файлов
# и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге # и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
# Если превышает - удаляем старейший # Если превышает - удаляем старейший
oldest = min(os.listdir(records_path + "/" + i), oldest = min(os.listdir(records_path + "/" + streamer),
key=os.path.getctime) key=os.path.getctime)
os.unlink(oldest) os.unlink(oldest)
log.warning("Удален файл: " + oldest) log.warning("Удален файл: " + oldest)