| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- #!/usr/bin/env python
- # Licensed to Cloudera, Inc. under one
- # or more contributor license agreements. See the NOTICE file
- # distributed with this work for additional information
- # regarding copyright ownership. Cloudera, Inc. licenses this file
- # to you 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
- from pytz import UnknownTimeZoneError, datetime, timezone
- class Formatter(logging.Formatter):
- def formatTime(self, record, datefmt=None):
- try:
- tz = timezone(os.environ['TZ'])
- except (KeyError, UnknownTimeZoneError):
- tz = None
- try:
- ct = datetime.datetime.fromtimestamp(record.created, tz=tz)
- except (OverflowError, TypeError, ValueError):
- # Fallback to original.
- return super(Formatter, self).formatTime(record, datefmt=datefmt)
- if datefmt:
- s = ct.strftime(datefmt)
- else:
- t = ct.strftime("%Y-%m-%d %H:%M:%S")
- s = "%s,%03d" % (t, record.msecs)
- return s
- class MessageOnlyFormatter(logging.Formatter):
- def formatTime(self, record, datefmt=None):
- return ''
- def format(self, record):
- return record.getMessage()
|