logging.py 2.67 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
# -*- coding:utf-8 -*-
#
# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
#
# This file is part of Poezio.
#
# Poezio 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 of the License.
#
# Poezio 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Poezio.  If not, see <http://www.gnu.org/licenses/>.

import sys
20
from os import environ, makedirs
21
import os
22
from datetime import datetime
23
from config import config
24

25
DATA_HOME = config.get('log_dir', os.path.join(environ.get('XDG_DATA_HOME') or os.path.join(environ.get('HOME'), '.local', 'share'), 'poezio'))
26

27 28 29 30 31
class Logger(object):
    """
    Appends things to files. Error/information/warning logs
    and also log the conversations to logfiles
    """
32 33 34 35 36 37 38 39 40 41 42
    def __init__(self):# , logfile, loglevel):
        self.logfile = config.get('logfile', 'logs')
        self.loglevel = config.get('loglevel', 3)
        # self.logfile = logfile
        # self.loglevel = loglevel

    def info(self, msg):
        if self.logfile and self.loglevel >= 3:
            fd = open(self.logfile, 'a')
            fd.write(datetime.now().strftime("%H:%M:%S") + ' Info [' + msg + ']\n')
            fd.close()
43 44

    def warning(self, msg):
45
        if self.logfile and self.loglevel >= 2:
46
            fd = open(self.logfile, 'a')
47
            fd.write(datetime.now().strftime("%H:%M:%S") + ' Warning [' + msg + ']\n')
48 49 50
            fd.close()

    def error(self, msg):
51
        if self.logfile and self.loglevel >= 1:
52
            fd = open(self.logfile, 'a')
53
            fd.write(datetime.now().strftime("%H:%M:%S") + ' Error [' + msg + ']\n')
54
            fd.close()
55 56
        sys.exit(-1)

57 58 59 60 61 62
    def message(self, room, nick, msg):
        """
        log the message in the appropriate room
        """
        if config.get('use_log', 'false') == 'false':
            return
63
        dir = DATA_PATH+'logs/'
64 65
        try:
            makedirs(dir)
66 67 68 69 70 71
        except OSError:
            pass
        try:
            fd = open(dir+room, 'a')
        except IOError:
            return
72 73 74 75
        try:
            msg = msg.encode('utf-8')
        except:
            pass
76
        if nick:
77
            fd.write(datetime.now().strftime('%d-%m-%y [%H:%M:%S] ')+nick.encode('utf-8')+': '+msg+'\n')
78
        else:
79
            fd.write(datetime.now().strftime('%d-%m-%y [%H:%M:%S] ')+'* '+msg+'\n')
80 81
        fd.close()

82
logger = Logger()