New hybrid method

resources/lib/helper/loghandler.py

@@ -0,0 +1,96 @@
+# -*- coding: utf-8 -*-
+
+##################################################################################################
+
+import logging
+import xbmc
+
+import database
+from . import window
+
+##################################################################################################
+
+
+def config():
+
+    logger = logging.getLogger('EMBY')
+    logger.addHandler(LogHandler())
+    logger.setLevel(logging.DEBUG)
+
+
+class LogHandler(logging.StreamHandler):
+
+    def __init__(self):
+
+        logging.StreamHandler.__init__(self)
+        self.setFormatter(MyFormatter())
+
+        self.sensitive = {'Token': [], 'Server': []}
+
+        for server in database.get_credentials()['Servers']:
+
+            if server.get('AccessToken'):
+                self.sensitive['Token'].append(server['AccessToken'])
+
+            self.sensitive['Server'].append(server['LocalAddress'].split('://')[1])
+            self.sensitive['Server'].append(server['RemoteAddress'].split('://')[1])
+
+            if server.get('ManualAddress'):
+                self.sensitive['Server'].append(server['ManualAddress'].split('://')[1])
+
+
+    def emit(self, record):
+
+        if self._get_log_level(record.levelno):
+            string = self.format(record)
+
+            for server in self.sensitive['Server']:
+                string = string.replace(server or "{server}", "{emby-server}")
+
+            for token in self.sensitive['Token']:
+                string = string.replace(token or "{token}", "{emby-token}")
+            try:
+                xbmc.log(string, level=xbmc.LOGNOTICE)
+            except UnicodeEncodeError:
+                xbmc.log(string.encode('utf-8'), level=xbmc.LOGNOTICE)
+
+    @classmethod
+    def _get_log_level(cls, level):
+
+        levels = {
+            logging.ERROR: 0,
+            logging.WARNING: 0,
+            logging.INFO: 1,
+            logging.DEBUG: 2
+        }
+        try:
+            log_level = int(window('emby_logLevel'))
+        except ValueError:
+            log_level = 0
+
+        return log_level >= levels[level]
+
+
+class MyFormatter(logging.Formatter):
+
+    def __init__(self, fmt="%(name)s -> %(message)s"):
+
+        logging.Formatter.__init__(self, fmt)
+
+    def format(self, record):
+
+        # Save the original format configured by the user
+        # when the logger formatter was instantiated
+        format_orig = self._fmt
+
+        # Replace the original format with one customized by logging level
+        if record.levelno in (logging.DEBUG, logging.ERROR):
+            self._fmt = '%(name)s -> %(levelname)s:: %(message)s'
+
+        # Call the original formatter class to do the grunt work
+        result = logging.Formatter.format(self, record)
+
+        # Restore the original format configured by the user
+        self._fmt = format_orig
+
+        return result