log: add log to file and flush logger functionalities

1 files changed, 41 insertions, 0 deletions

diff --git a/rumba/log.py b/rumba/log.py
index 893f5f7..6c9cd10 100644
--- a/rumba/log.py
+++ b/rumba/log.py
@@ -27,8 +27,11 @@
 import logging
 import logging.handlers
 import multiprocessing
+import os
 import sys
 
+import time
+
 
 DEBUG = logging.DEBUG
 INFO = logging.INFO
@@ -277,6 +280,10 @@ def reset_logging_level():
     set_logging_level(logging.ERROR, '')
 
 
+def flush_log():
+    time.sleep(0.1)
+
+
 def flush_and_kill_logging():
     """
     Flushes all queued log messages and stops the logging facility.
@@ -286,3 +293,37 @@ def flush_and_kill_logging():
     """
     global logging_listener
     logging_listener.stop()
+
+
+class LogOptions(object):
+
+    @staticmethod
+    def _get_handlers():
+        global logging_listener
+        return tuple(logging_listener.handlers)
+
+    @staticmethod
+    def _set_handlers(*handlers):
+        global logging_listener
+        logging_listener.handlers = handlers
+
+    def _add_handler(self, handler):
+        handler.setFormatter(RumbaFormatter())
+        handlers = self._get_handlers() + (handler,)
+        self._set_handlers(*handlers)
+        return self
+
+    def log_to_file(self, path='rumba.log'):
+        new_handler = logging.handlers.RotatingFileHandler(path)
+        return self._add_handler(new_handler)
+
+    def reset_logging(self):
+        self._set_handlers(*tuple())
+        return self
+
+    def log_to_console(self):
+        new_handler = logging.StreamHandler(sys.stdout)
+        return self._add_handler(new_handler)
+
+
+options = LogOptions()  # module instance used for configuration