summaryrefslogtreecommitdiff
path: root/portato/log.py
blob: 9c5ebd242213a20dad4415940c686244ed61a918 (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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# -*- coding: utf-8 -*-
#
# File: portato/log.py
# This file is part of the Portato-Project, a graphical portage-frontend.
#
# Copyright (C) 2008 René 'Necoro' Neumann
# This is free software.  You may redistribute copies of it under the terms of
# the GNU General Public License version 2.
# There is NO WARRANTY, to the extent permitted by law.
#
# Written by René 'Necoro' Neumann <necoro@necoro.net>

from __future__ import absolute_import

import logging
import sys
import os

from .constants import SESSION_DIR

(S_NOT, S_STREAM_ONLY, S_BOTH) = range(3)

started = S_NOT

LOGFILE = os.path.join(SESSION_DIR, "portato.log")

class OutputFormatter (logging.Formatter):

    colors = {
            "blue"    : 34,
            "green"    : 32,
            "red"    : 31,
            "yellow": 33
            }

    def __init__(self, *args, **kwargs):
        logging.Formatter.__init__(self, *args, **kwargs)

        for key, value in self.colors.iteritems():
            self.colors[key] = "\x1b[01;%02dm*\x1b[39;49;00m" % value

    def format (self, record):
        string = logging.Formatter.format(self, record)
        color = None

        if os.isatty(sys.stderr.fileno()):
            if record.levelno <= logging.DEBUG:
                color = self.colors["blue"]
            elif record.levelno <= logging.INFO:
                color = self.colors["green"]
            elif record.levelno <= logging.WARNING:
                color = self.colors["yellow"]
            else:
                color = self.colors["red"]
        else:
            color = "%s:" % record.levelname

        return "%s %s" % (color, string)

def start(file = True):
    global started

    if started == S_BOTH: return

    # logging: root (file)
    if file:
        if not (os.path.exists(SESSION_DIR) and os.path.isdir(SESSION_DIR)):
            os.mkdir(SESSION_DIR)

        formatter = logging.Formatter("%(levelname)-8s: %(message)s (%(filename)s:%(lineno)s)")
        handler = logging.FileHandler(LOGFILE, "w")
        handler.setFormatter(formatter)
        logging.getLogger("portatoLogger").addHandler(handler)
    
    if started == S_NOT:
        logging.getLogger("portatoLogger").setLevel(logging.DEBUG)
        logging.getLogger("portatoLogger").propagate = False

    # logging: stream
    # this logger should be used
    if started == S_NOT:
        formatter = OutputFormatter("%(message)s (%(filename)s:%(lineno)s)")
        handler = logging.StreamHandler()
        handler.setFormatter(formatter)
        logging.getLogger("portatoLogger.stream").addHandler(handler)
        logging.getLogger("portatoLogger.stream").setLevel(logging.DEBUG)

    started = S_BOTH if file else S_STREAM_ONLY
06-25 21:26:05 +0200'>2008-06-25Now load 22 versions when running the correct portageRené 'Necoro' Neumann4-8/+18 2008-06-25Added Package_22 and System_22René 'Necoro' Neumann4-5/+63