summaryrefslogtreecommitdiff
path: root/pykolab/logger.py
blob: 8f2e5f7337af1322f44673d4554e1d70264cddee (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
89
90
91
92
93
94
# -*- coding: utf-8 -*-
# Copyright 2010-2011 Kolab Systems AG (http://www.kolabsys.com)
#
# Jeroen van Meeuwen (Kolab Systems) <vanmeeuwen a kolabsys.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 3 or, at your option, any later version
#
# This program 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 Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#

import inspect
import logging
import logging.handlers
import sys
import time

from pykolab.translate import _

class Logger(logging.Logger):
    debuglevel = 0
    loglevel = logging.CRITICAL

    for arg in sys.argv:
        if debuglevel == -1:
            debuglevel = int(arg)
            #print debuglevel
            loglevel = logging.DEBUG
            break
        if '-d' == arg:
            debuglevel = -1
            continue

        if '-l' == arg:
            loglevel = -1
            continue
        if loglevel == -1:
            if hasattr(logging,arg.upper()):
                loglevel = getattr(logging,arg.upper())
            else:
                loglevel = logging.DEBUG

    def __init__(self, *args, **kw):
        if kw.has_key('name'):
            name = kw['name']
        elif len(args) == 1:
            name = args[0]
        else:
            name = 'pykolab'

        logging.Logger.__init__(self, name)

        plaintextformatter = logging.Formatter("%(asctime)s %(name)s %(levelname)s %(message)s")

        self.console_stdout = logging.StreamHandler(sys.stdout)
        self.console_stdout.setFormatter(plaintextformatter)

        self.addHandler(self.console_stdout)

        if kw.has_key('logfile'):
            self.logfile = kw['logfile']
        else:
            self.logfile = '/var/log/kolab/pykolab.log'

        try:
            filelog_handler = logging.FileHandler(filename=self.logfile)
            filelog_handler.setFormatter(plaintextformatter)
        except IOError, e:
            print >> sys.stderr, _("Cannot log to file %s: %s") % (self.logfile, e)

        if not len(self.handlers) > 1:
            try:
                self.addHandler(filelog_handler)
            except:
                pass

    def remove_stdout_handler(self):
        self.removeHandler(self.console_stdout)

    def debug(self, msg, level=1):
        self.setLevel(self.loglevel)
        if level <= self.debuglevel:
            # TODO: Not the way it's supposed to work!
            self.log(logging.DEBUG, msg)

logging.setLoggerClass(Logger)