| #!/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 |
| import _env # pylint: disable=unused-import |
| |
| _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') |