92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import logging
|
|
import os
|
|
|
|
import fixtures
|
|
|
|
_TRUE_VALUES = ('True', 'true', '1', 'yes')
|
|
_FALSE_VALUES = ('False', 'false', '0', 'no')
|
|
_BASE_LOG_LEVELS = ('DEBUG', 'INFO', 'WARN', 'WARNING', 'ERROR', 'CRITICAL')
|
|
_LOG_LEVELS = dict((n, getattr(logging, n)) for n in _BASE_LOG_LEVELS)
|
|
_LOG_LEVELS.update({
|
|
'TRACE': 5,
|
|
})
|
|
|
|
|
|
def _try_int(value):
|
|
"""Try to make some value into an int."""
|
|
try:
|
|
return int(value)
|
|
except (ValueError, TypeError):
|
|
return None
|
|
|
|
|
|
class ConfigureLogging(fixtures.Fixture):
|
|
"""Configure logging.
|
|
|
|
The behavior is managed through two environment variables. If
|
|
``OS_DEBUG`` is true then the logging level is set to debug. If
|
|
``OS_LOG_CAPTURE`` is true a FakeLogger is configured. Alternatively,
|
|
``OS_DEBUG`` can be set to an explicit log level, such as ``INFO``.
|
|
|
|
"True" values include ``True``, ``true``, ``1`` and ``yes``.
|
|
Valid log levels include ``DEBUG``, ``INFO``, ``WARNING``, ``ERROR``,
|
|
``TRACE`` and ``CRITICAL`` (or any other valid integer logging level).
|
|
|
|
.. py:attribute:: logger
|
|
|
|
The logger fixture, if it is created.
|
|
|
|
.. py:attribute:: level
|
|
|
|
``logging.DEBUG`` if debug logging is enabled, otherwise
|
|
the log level specified by ``OS_DEBUG``, otherwise ``None``.
|
|
|
|
:param format: The logging format string to use.
|
|
|
|
"""
|
|
|
|
DEFAULT_FORMAT = "%(levelname)8s [%(name)s] %(message)s"
|
|
"""Default log format"""
|
|
|
|
def __init__(self, format=DEFAULT_FORMAT):
|
|
super(ConfigureLogging, self).__init__()
|
|
self._format = format
|
|
self.level = None
|
|
_os_debug = os.environ.get('OS_DEBUG')
|
|
_os_level = _try_int(_os_debug)
|
|
if _os_debug in _TRUE_VALUES:
|
|
self.level = logging.DEBUG
|
|
elif _os_level is not None:
|
|
self.level = _os_level
|
|
elif _os_debug in _LOG_LEVELS:
|
|
self.level = _LOG_LEVELS[_os_debug]
|
|
elif _os_debug and _os_debug not in _FALSE_VALUES:
|
|
raise ValueError('OS_DEBUG=%s is invalid.' % (_os_debug))
|
|
self.capture_logs = os.environ.get('OS_LOG_CAPTURE') in _TRUE_VALUES
|
|
self.logger = None
|
|
|
|
def setUp(self):
|
|
super(ConfigureLogging, self).setUp()
|
|
if self.capture_logs:
|
|
self.logger = self.useFixture(
|
|
fixtures.FakeLogger(
|
|
format=self._format,
|
|
level=self.level,
|
|
nuke_handlers=True,
|
|
)
|
|
)
|
|
else:
|
|
logging.basicConfig(format=self._format, level=self.level)
|