Compare commits

..

3 Commits

View File

@@ -76,6 +76,7 @@ def set_config():
Возвращает объект конфига (configparser.ConfigParser()) Возвращает объект конфига (configparser.ConfigParser())
""" """
config = configparser.ConfigParser() config = configparser.ConfigParser()
# Читаем конфиг, если пустой - заполняем
if not config.read('cfg_file.ini'): if not config.read('cfg_file.ini'):
config["app"] = { config["app"] = {
"path": "", "path": "",
@@ -85,10 +86,17 @@ def set_config():
config["twitch"] = { config["twitch"] = {
"app_id": "", "app_id": "",
"app_secret": "", "app_secret": "",
"streamers": "asdf,qqqqq" "streamers": "t2x2,arcadia_online,252mart,the_viox"
} }
with open('cfg_file.ini', 'w') as cfg_file: with open('cfg_file.ini', 'w') as cfg_file:
config.write(cfg_file) config.write(cfg_file)
# Проверка конфига
if config['twitch']['app_id'] == "" or config['twitch']['app_secret'] == "":
log.critical("Параметры app_id или app_secret пусты. Необходимо заполнить эти параметры в конфиге. "
"Читай README.md")
exit(1)
return config return config
@@ -123,13 +131,13 @@ def recorder(streamer):
Функция, которая запускает youtube-dl, фактически записывает стрим Функция, которая запускает youtube-dl, фактически записывает стрим
""" """
streamer_path = os.path.join(config['app']['path'], streamer) streamer_path = os.path.join(config['app']['path'], streamer)
log.info("Записываем стрим {}\n".format(streamer)) log.info("Записываем стрим {}".format(streamer))
# cmdline для запуска youtube-dl # cmdline для запуска youtube-dl
cmdline = ["youtube-dl", "-q", "-o", cmdline = ["youtube-dl", "-q", "-o",
streamer_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("Запись стрима {} закончена".format(streamer))
if os.path.exists(os.path.join(streamer_path, "pid")): if os.path.exists(os.path.join(streamer_path, "pid")):
os.remove(os.path.join(streamer_path, "pid")) os.remove(os.path.join(streamer_path, "pid"))
log.info("lock файл удален") log.info("lock файл удален")
@@ -169,7 +177,7 @@ def record_streamer(user_stream, streamer):
) )
else: else:
# Если стрим не идет, то пишем об этом и убираем его из залоченных # Если стрим не идет, то пишем об этом и убираем его из залоченных
log.info("{} не стримит".format(streamer)) log.debug("{} не стримит".format(streamer))
# Если есть лок, то удаляем # Если есть лок, то удаляем
if os.path.exists(os.path.join(streamer_path, "pid")): if os.path.exists(os.path.join(streamer_path, "pid")):
os.remove(os.path.join(streamer_path, "pid")) os.remove(os.path.join(streamer_path, "pid"))
@@ -215,6 +223,9 @@ def remove_old_streams():
if __name__ == "__main__": if __name__ == "__main__":
# Log config
log = get_logger("main")
# Проверить, установлены ли нужные утилиты # Проверить, установлены ли нужные утилиты
if not check_installed_tools(): if not check_installed_tools():
exit() exit()
@@ -222,9 +233,6 @@ if __name__ == "__main__":
# Set config # Set config
config = set_config() config = set_config()
# Log config
log = get_logger("main")
# Проверять стримы раз в check_period # Проверять стримы раз в check_period
# Каждый час удалять старые стримы # Каждый час удалять старые стримы
schedule.every(int(config['app']['check_period'])).seconds.do(streamers_loop) schedule.every(int(config['app']['check_period'])).seconds.do(streamers_loop)