#!/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()