Фикс check_installed_tools() и проверка конфига

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
 
 
@@ -118,75 +126,79 @@ def check_installed_tools() -> bool:
     return True
 
 
-def start_recording(streamer):
-    """
-    Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
-    """
-    th = Thread(target=recorder, args=(streamer,))
-    th.start()
-
-
 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 файл удален")
 
 
-def check_stream():
+def get_streamer_id(streamer):
-    # FIXME: Распилить на более мелкие функции
+    """Получаем id стримера, при неудаче отдаем None"""
-    """
+    resolved_id = twitch_client.get_users(logins=[streamer])
-    1. Проверка на наличие стрима
+    if resolved_id['data']:
-    1.1 Если нет - удалить lock файл, если он есть
+        return resolved_id['data'][0]['id']
-    1.2 Если есть - создать lock файл, запустить записывалку
+    else:
-    """
+        log.error(
-    for streamer in config['twitch']['streamers'].split(','):
+            "Аккаунт {} не найден".format(streamer)
-        # Путь до диры со стримами
+        )
-        streamer_path = os.path.join(config['app']['path'], streamer)
+        return None
-        # Получаем инфо о стримере, если не получается, выходим с ошибкой
 
-        # resolved_id = client.users.translate_usernames_to_ids(i)
+
-        resolved_id = twitch_client.get_users(logins=[streamer])
+def record_streamer(user_stream, streamer):
-        if not resolved_id['data']:
+    """Проверяем, идет ли стрим. Если идет - записываем. Если не идет - удаляем pid файл"""
-            log.error(
+    streamer_path = os.path.join(config['app']['path'], streamer)
-                "Аккаунт {} не найден".format(streamer)
+    if user_stream['data']:
-            )
-            continue
         # Создаем путь до диры со стримером, если папка не существует
         if not (os.path.exists(streamer_path)):
             os.makedirs(streamer_path)
             log.info("Создана директория {}".format(streamer_path))
-        # Достаем ID стримера из инфо
+
-        user_id = resolved_id['data'][0]['id']
+        # Если стрим идет и лок файла нет, то записываем и ставим лок
-        user_stream = twitch_client.get_streams(user_id=user_id)
+        if (user_stream['data'][0]['type'] == 'live') and not (
-        # Если стрим идет, то идем дальше
+                os.path.exists(os.path.join(streamer_path, "pid"))):
-        if user_stream['data']:
+            log.info("{} стримит".format(streamer))
-            # Если стрим идет и лок файла нет, то записываем и ставим лок
+            th = Thread(target=recorder, args=(streamer,))
-            if (user_stream['data'][0]['type'] == 'live') and not (
+            th.start()
-                    os.path.exists(os.path.join(streamer_path, "pid"))):
+            os.mknod(os.path.join(streamer_path, "pid"))
-                log.info("{} стримит".format(streamer))
-                start_recording(streamer)
-                os.mknod(os.path.join(streamer_path, "pid"))
-            else:
-                log.info(
-                    "Идет запись {}".format(streamer)
-                )
         else:
-            # Если стрим не идет, то пишем об этом и убираем его из залоченных
+            log.info(
-            log.info("{} не стримит".format(streamer))
+                "Идет запись {}".format(streamer)
-            # Если есть лок, то удаляем
+            )
-            if os.path.exists(os.path.join(streamer_path, "pid")):
+    else:
-                os.remove(os.path.join(streamer_path, "pid"))
+        # Если стрим не идет, то пишем об этом и убираем его из залоченных
+        log.debug("{} не стримит".format(streamer))
+        # Если есть лок, то удаляем
+        if os.path.exists(os.path.join(streamer_path, "pid")):
+            os.remove(os.path.join(streamer_path, "pid"))
+
+
+def streamers_loop():
+    """
+    Цикл по стримерам
+    Проходится по каждому логину, достает ID стримера,
+    достает инфу о стримах, запускает функцию для записи
+    """
+    for streamer in config['twitch']['streamers'].split(','):
+
+        # Достаем ID стримера, если пустой - пропускаем цикл
+        user_id = get_streamer_id(streamer)
+        if user_id is None:
+            continue
+        # Получаем данные о стриме
+        user_stream = twitch_client.get_streams(user_id=user_id)
+        # Запускаем запись
+        record_streamer(user_stream, streamer)
 
 
 def remove_old_streams():
@@ -211,6 +223,9 @@ def remove_old_streams():
 
 
 if __name__ == "__main__":
+    # Log config
+    log = get_logger("main")
+
     # Проверить, установлены ли нужные утилиты
     if not check_installed_tools():
         exit()
@@ -218,16 +233,14 @@ if __name__ == "__main__":
     # Set config
     config = set_config()
 
-    # Log config
-    log = get_logger("main")
-    log.info("Запущен")
-
     # Проверять стримы раз в check_period
-    schedule.every(int(config['app']['check_period'])).seconds.do(check_stream)
     # Каждый час удалять старые стримы
+    schedule.every(int(config['app']['check_period'])).seconds.do(streamers_loop)
     schedule.every(1).hours.do(remove_old_streams)
 
+    # Инициализируем клиент твича
     twitch_client = Twitch(config['twitch']['app_id'], config['twitch']['app_secret'])
+    log.info("Запущен")
     while True:
         schedule.run_pending()
         time.sleep(1)

requirements.txt

@@ -1,4 +1,3 @@
-youtube-dl==2021.4.17
+youtube-dl==2021.12.17
 twitchAPI==2.5.7.1
-schedule
+schedule==1.2.0
-termcolor