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())
|
Возвращает объект конфига (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)
|
||||||
|
|||||||
Reference in New Issue
Block a user