blob: 67f5c83ae3271914689c55fc8b10c95c78e805d1 [file] [log] [blame]
#!/usr/bin/python
#
# Copyright 2020 The Cobalt Authors. All Rights Reserved.
#
# 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.
"""Provides a consistent mechanism for the initialize of logging."""
import logging
_NAME_TO_LEVEL = {
'info': logging.INFO,
'debug': logging.DEBUG,
'warning': logging.WARNING,
'error': logging.ERROR,
'critical': logging.CRITICAL,
}
def InitializeLogging(args):
"""Parses the provided argparse.Namespace to determine a logging level."""
log_level = logging.INFO
if args:
# Allow a 'verbose' flag to force |logging.DEBUG|.
if 'verbose' in args and args.verbose:
log_level = logging.DEBUG
elif 'log_level' in args:
log_level = _NAME_TO_LEVEL[args.log_level]
InitializeLoggingWithLevel(log_level)
def InitializeLoggingWithLevel(log_level):
logging.basicConfig(
level=log_level,
format=('[%(process)d:%(asctime)s.%(msecs)03d...:'
'%(levelname)s:%(filename)s(%(lineno)s)] %(message)s'),
datefmt='%m-%d %H:%M')