daemon.py

@@ -5,20 +5,27 @@
 # TODO: Сделать нормальную конфигурацию
 
 import os
+import sys
 from threading import Thread
 import config_python
 import schedule
 from twitch import TwitchClient
 import subprocess
 import time
+import logging
+from logging.handlers import TimedRotatingFileHandler
 streamers = config_python.streamers
 client_id = config_python.twitchid
 ytdl_options = config_python.ytdl_options
+log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
+log_file = 'output.log'
+
 
 def which(command):
     # Пиздец, почему нет нормального аналога which из bash???
     '''
-    Мой аналог which из bash'а, который отдает true или false при наличии или отсутствии утилиты    
+    Мой аналог which из bash'а, который отдает true или false
+    при наличии или отсутствии утилиты
     '''
     for dirs in os.get_exec_path():
         if command in os.listdir(dirs):
@@ -27,6 +34,7 @@ def which(command):
     # Если ничего не нашли во всех дирах, то выходим с False
     return False
 
+
 def checkTools():
     '''
     Проверяет, установлены ли необходимые утилиты
@@ -34,10 +42,11 @@ def checkTools():
     tools = ('youtube-dl', 'ffmpeg')
     for i in tools:
         if not which(i):
-            print(i + " не установлен")
+            log.critical(i + " не установлен")
             return False
     return True
 
+
 def startRecord(i):
     '''
     Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
@@ -45,21 +54,24 @@ def startRecord(i):
     th = Thread(target=recorder, args=(i, ))
     th.start()
 
+
 def recorder(i):
     '''
     Функция, которая запускает youtube-dl, фактически записывает стрим
     '''
     path = config_python.path + "/" + i
-    print("Записываем стрим %s\n" % i)
+    log.info("Записываем стрим %s\n" % i)
     # cmdline для запуска youtube-dl
     cmdline = ["youtube-dl", "-q", ytdl_options,
-        "-o", path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
+               "-o",
+               path+"/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
                "https://twitch.tv/" + i]
     s = subprocess.call(cmdline)
-    print("Запись стрима %s закончена\n" % i)
+    log.info("Запись стрима %s закончена\n" % i)
     if (os.path.exists(path+"/pid")):
         os.system("rm "+path+"/pid")
-        print("lock файл удален")
+        log.info("lock файл удален")
+
 
 def checkAlive():
     '''
@@ -74,19 +86,20 @@ def checkAlive():
         # Создаем путь до диры со стримером, если его нет
         if not (os.path.exists(path)):
             os.makedirs(path)
-        user_id=client.users.translate_usernames_to_ids(i)[0]['id'] # Получить ID по нику
+        # Получить ID по нику
+        user_id = client.users.translate_usernames_to_ids(i)[0]['id']
         # Если стрим идет, то идем дальше
         if client.streams.get_stream_by_user(user_id):
             # Если стрим идет и лок файла нет, то записываем и ставим лок
             if (client.streams.get_stream_by_user(user_id).stream_type == 'live') and not (os.path.exists(config_python.path+"/"+i+"/pid")):
-                print(i+" стримит")
+                log.info(i + " стримит")
                 startRecord(i)
                 os.system("touch "+path+"/pid")
             else:
-                print(i+" Идет запись")
+                log.info(i + " Идет запись")
         else:
             # Если стрим не идет, то пишем об этом и убираем его из залоченных
-            print(i+" Не стримит")
+            log.info(i + " Не стримит")
             # Если есть лок, то удаляем
             if (os.path.exists(path+"/pid")):
                 os.system("rm "+path+"/pid")
@@ -97,6 +110,7 @@ def removeOldStreams():
     records_path = config_python.path
     # По каждой папке со стримерами
     for i in streamers:
+        try:
             # Если файлов в папке со стримами больше чем указано в конфиге
             if len(os.listdir(records_path+"/"+i)) > config_python.max_files:
                 # Получаем список файлов
@@ -104,15 +118,42 @@ def removeOldStreams():
                 # Если превышает - удаляем старейший
                 oldest = min(os.listdir(records_path+"/"+i), key=os.path.getctime)
                 os.unlink(oldest)
-            print("Удален файл: " + oldest)
+                log.warning("Удален файл: " + oldest)
+        except Exception as e:
+            log.error(e)
+
+def get_console_handler():
+    console_handler = logging.StreamHandler(sys.stdout)
+    console_handler.setFormatter(log_format)
+    return console_handler
+
+def get_file_handler():
+    file_handler = TimedRotatingFileHandler(log_file, when='midnight')
+    file_handler.setFormatter(log_format)
+    return file_handler
+
+def get_logger(logger_name):
+    logger = logging.getLogger(logger_name)
+    logger.setLevel(logging.DEBUG)
+    logger.addHandler(get_console_handler())
+    logger.addHandler(get_file_handler())
+    logger.propagate = False
+    return logger
+
 
 if __name__ == "__main__":
     # Проверить, установлены ли нужные утилиты
     if not checkTools(): exit()
+
+    # Log config
+    log = get_logger("main")
+    log.info("Запущен")
+
     # Проверять стримы раз в check_period
     schedule.every(config_python.check_period).seconds.do(checkAlive)
     # Каждый час удалять старые стримы
     schedule.every(1).hours.do(removeOldStreams)
+
     while True:
         schedule.run_pending()
         time.sleep(1)