You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.0 KiB
66 lines
2.0 KiB
import logging
|
|
import re
|
|
import traceback
|
|
|
|
class CPLog(object):
|
|
|
|
context = ''
|
|
replace_private = ['api', 'apikey', 'api_key', 'password', 'username', 'h', 'uid', 'key']
|
|
|
|
def __init__(self, context = ''):
|
|
if context.endswith('.main'):
|
|
context = context[:-5]
|
|
|
|
self.context = context
|
|
self.logger = logging.getLogger()
|
|
|
|
def info(self, msg, replace_tuple = ()):
|
|
self.logger.info(self.addContext(msg, replace_tuple))
|
|
|
|
def debug(self, msg, replace_tuple = ()):
|
|
self.logger.debug(self.addContext(msg, replace_tuple))
|
|
|
|
def error(self, msg, replace_tuple = ()):
|
|
self.logger.error(self.addContext(msg, replace_tuple))
|
|
|
|
def warning(self, msg, replace_tuple = ()):
|
|
self.logger.warning(self.addContext(msg, replace_tuple))
|
|
|
|
def critical(self, msg, replace_tuple = ()):
|
|
self.logger.critical(self.addContext(msg, replace_tuple), exc_info = 1)
|
|
|
|
def addContext(self, msg, replace_tuple = ()):
|
|
return '[%+25.25s] %s' % (self.context[-25:], self.safeMessage(msg, replace_tuple))
|
|
|
|
def safeMessage(self, msg, replace_tuple = ()):
|
|
|
|
from couchpotato.environment import Env
|
|
from couchpotato.core.helpers.encoding import ss
|
|
|
|
msg = ss(msg)
|
|
|
|
try:
|
|
msg = msg % replace_tuple
|
|
except:
|
|
try:
|
|
if isinstance(replace_tuple, tuple):
|
|
msg = msg % tuple([ss(x) for x in list(replace_tuple)])
|
|
else:
|
|
msg = msg % ss(replace_tuple)
|
|
except:
|
|
self.logger.error(u'Failed encoding stuff to log: %s' % traceback.format_exc())
|
|
|
|
if not Env.get('dev'):
|
|
|
|
for replace in self.replace_private:
|
|
msg = re.sub('(%s=)[^\&]+' % replace, '%s=xxx' % replace, msg)
|
|
|
|
# Replace api key
|
|
try:
|
|
api_key = Env.setting('api_key')
|
|
if api_key:
|
|
msg = msg.replace(api_key, 'API_KEY')
|
|
except:
|
|
pass
|
|
|
|
return msg
|
|
|