Compare commits
3 Commits
97fb4c8637
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| 227910a14d | |||
| f3d2f2e705 | |||
| 6ef60de9f4 |
22
daemon.py
22
daemon.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user