summaryrefslogtreecommitdiff
path: root/service/pixelated/config/logging_setup.py
blob: a15413a0dcd4d0b99a3c1e4fbf65556e24b2906f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#
# Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.

import logging
import socket
import sys
import os
from twisted.python import log
from twisted.python import util


def init_logging(args):
    debug_enabled = args.debug or os.environ.get('DEBUG', False)

    logging.basicConfig(level=logging.DEBUG if debug_enabled else logging.WARNING,
                        format='[%(asctime)s] ' + socket.gethostname() + ' %(name)-12s %(levelname)-8s %(message)s',
                        datefmt='%m-%d %H:%M:%S',
                        filemode='a')

    if debug_enabled:
        init_debugger()

    log.startLogging(sys.stdout)


def init_debugger():
    formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
    console = logging.StreamHandler()
    console.setLevel(logging.DEBUG)
    console.setFormatter(formatter)
    logging.getLogger('').addHandler(console)


class PixelatedLogObserver(log.FileLogObserver):

    """ FileLogObserver with a customized format """
    def emit(self, event):
        text = log.textFromEventDict(event)

        if text is None:
            return

        self.timeFormat = '[%Y-%m-%d %H:%M:%S]'
        time_str = self.formatTime(event['time'])

        fmt_dict = {'text': text.replace('\n', '\n\t')}
        msg_str = log._safeFormat('%(text)s\n', fmt_dict)

        logging.debug(str(event))

        util.untilConcludes(self.write, time_str + ' ' + socket.gethostname() + ' ' + msg_str)
        util.untilConcludes(self.flush)