Compare commits

..

3 Commits

View File

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