mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-08 09:01:47 -05:00
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
# Copyright (c) 2017 Ansible by Red Hat
|
|
# All Rights Reserved.
|
|
|
|
# Python
|
|
import logging
|
|
import os.path
|
|
|
|
# Django
|
|
from django.conf import settings
|
|
|
|
|
|
class RSysLogHandler(logging.handlers.SysLogHandler):
|
|
|
|
append_nul = False
|
|
|
|
def _connect_unixsocket(self, address):
|
|
super(RSysLogHandler, self)._connect_unixsocket(address)
|
|
self.socket.setblocking(False)
|
|
|
|
def emit(self, msg):
|
|
if not settings.LOG_AGGREGATOR_ENABLED:
|
|
return
|
|
if not os.path.exists(settings.LOGGING['handlers']['external_logger']['address']):
|
|
return
|
|
try:
|
|
return super(RSysLogHandler, self).emit(msg)
|
|
except ConnectionRefusedError:
|
|
# rsyslogd has gone to lunch; this generally means that it's just
|
|
# been restarted (due to a configuration change)
|
|
# unfortunately, we can't log that because...rsyslogd is down (and
|
|
# would just us back ddown this code path)
|
|
pass
|
|
except BlockingIOError:
|
|
# for <some reason>, rsyslogd is no longer reading from the domain socket, and
|
|
# we're unable to write any more to it without blocking (we've seen this behavior
|
|
# from time to time when logging is totally misconfigured;
|
|
# in this scenario, it also makes more sense to just drop the messages,
|
|
# because the alternative is blocking the socket.send() in the
|
|
# Python process, which we definitely don't want to do)
|
|
pass
|
|
|
|
|
|
ColorHandler = logging.StreamHandler
|
|
|
|
if settings.COLOR_LOGS is True:
|
|
try:
|
|
from logutils.colorize import ColorizingStreamHandler
|
|
|
|
class ColorHandler(ColorizingStreamHandler):
|
|
|
|
def format(self, record):
|
|
message = logging.StreamHandler.format(self, record)
|
|
return '\n'.join([
|
|
self.colorize(line, record)
|
|
for line in message.splitlines()
|
|
])
|
|
|
|
level_map = {
|
|
logging.DEBUG: (None, 'green', True),
|
|
logging.INFO: (None, None, True),
|
|
logging.WARNING: (None, 'yellow', True),
|
|
logging.ERROR: (None, 'red', True),
|
|
logging.CRITICAL: (None, 'red', True),
|
|
}
|
|
except ImportError:
|
|
# logutils is only used for colored logs in the dev environment
|
|
pass
|