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

[#12] Сделал os.path там где это требуется

Browse Source
This commit is contained in:
lulzette 2023-06-02 00:37:43 +03:00
parent ded7eb2158
commit caa9e3264a
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
daemon.py
View File

@ -78,16 +78,16 @@ def recorder(streamer):
""" """
Функция, которая запускает youtube-dl, фактически записывает стрим Функция, которая запускает youtube-dl, фактически записывает стрим
""" """
path = config['app']['path'] + "/" + streamer streamer_path = os.path.join(config['app']['path'], streamer)
log.info("Записываем стрим {}\n".format(streamer)) log.info("Записываем стрим {}\n".format(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", streamer_path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
"https://twitch.tv/{}".format(streamer)] "https://twitch.tv/{}".format(streamer)]
subprocess.call(cmdline) subprocess.call(cmdline)
log.info("Запись стрима {} закончена\n".format(streamer)) log.info("Запись стрима {} закончена\n".format(streamer))
if os.path.exists(path + "/pid"): if os.path.exists(os.path.join(streamer_path, "pid")):
os.remove(path + "/pid") os.remove(os.path.join(streamer_path, "pid"))
log.info("lock файл удален") log.info("lock файл удален")
@ -100,7 +100,7 @@ def check_stream():
""" """
for streamer in config['twitch']['streamers'].split(','): for streamer in config['twitch']['streamers'].split(','):
# Путь до диры со стримами # Путь до диры со стримами
path = config['app']['path'] + "/" + streamer streamer_path = os.path.join(config['app']['path'], streamer)
# Получаем инфо о стримере, если не получается, выходим с ошибкой # Получаем инфо о стримере, если не получается, выходим с ошибкой
# resolved_id = client.users.translate_usernames_to_ids(i) # resolved_id = client.users.translate_usernames_to_ids(i)
@ -111,9 +111,9 @@ def check_stream():
) )
continue continue
# Создаем путь до диры со стримером, если папка не существует # Создаем путь до диры со стримером, если папка не существует
if not (os.path.exists(path)): if not (os.path.exists(streamer_path)):
os.makedirs(path) os.makedirs(streamer_path)
log.info("Создана директория {}".format(path)) log.info("Создана директория {}".format(streamer_path))
# Достаем 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,10 +121,10 @@ 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'] + "/" + streamer + "/pid")): os.path.exists(os.path.join(streamer_path, "pid"))):
log.info("{} стримит".format(streamer)) log.info("{} стримит".format(streamer))
start_recording(streamer) start_recording(streamer)
open(config['app']['path'] + "/pid", 'w').close open(os.path.join(streamer_path, "pid"), 'w').close
else: else:
log.info( log.info(
"Идет запись {}".format(streamer) "Идет запись {}".format(streamer)
@ -133,8 +133,8 @@ def check_stream():
# Если стрим не идет, то пишем об этом и убираем его из залоченных # Если стрим не идет, то пишем об этом и убираем его из залоченных
log.info("{} не стримит".format(streamer)) log.info("{} не стримит".format(streamer))
# Если есть лок, то удаляем # Если есть лок, то удаляем
if os.path.exists(path + "/pid"): if os.path.exists(os.path.join(streamer_path, "pid")):
os.remove(path + "/pid") os.remove(os.path.join(streamer_path, "pid"))
def remove_old_streams(): def remove_old_streams():
@ -143,13 +143,14 @@ def remove_old_streams():
# По каждой папке со стримерами # По каждой папке со стримерами
for streamer in config['twitch']['streamers']: for streamer in config['twitch']['streamers']:
try: try:
os.chdir(records_path + "/" + streamer) streamer_dir_path = os.path.join(records_path, streamer)
os.chdir(streamer_dir_path)
# Если файлов в папке со стримами больше чем указано в конфиге # Если файлов в папке со стримами больше чем указано в конфиге
if len(os.listdir(records_path + "/" + streamer)) > int(config['app']['max_files']): if len(os.listdir(streamer_dir_path)) > int(config['app']['max_files']):
# Получаем список файлов # Получаем список файлов
# и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге # и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
# Если превышает - удаляем старейший # Если превышает - удаляем старейший
oldest = min(os.listdir(records_path + "/" + streamer), oldest = min(os.listdir(streamer_dir_path),
key=os.path.getctime) key=os.path.getctime)
os.unlink(oldest) os.unlink(oldest)
log.warning("Удален файл: {}".format(oldest)) log.warning("Удален файл: {}".format(oldest))