mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-19 22:41:49 -05:00
b0db2b7bec
when rsyslogd restarts due to config changes, there's a brief moment where the socket will refuse connections on teardown; exception handling is needed here to deal with that
56 lines
1.7 KiB
Python
56 lines
1.7 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 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
|
|
|
|
|
|
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
|