backup.py/activitylog.py at master · drmad/backup.py · GitHub

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
from datetime import datetime
import sys

class ActivityLog:
    DEBUG = 3
    INFO = 2
    WARNING = 1
    ERROR = 0

    LEVEL_LABEL = {
        DEBUG : 'debug',
        INFO : 'info',
        WARNING : 'warning',
        ERROR : 'error',
    }

    log_fd = None
    log_level = 2 # Solo errores

    def set_log_level(self, log_level):
        self.log_level = log_level
        return self

    def set_log_file(self, log_file):
        self.log_fd = open(log_file, "a", 0)
        return self

    def log(self, level, message, set_date = False):
        # Añadimos la fecha, de ser necesario
        message ="{} {}".format(datetime.now(), message)

        # Siempre guardamos todos los mensajes en el fichero de registro
        if self.log_fd:
            self.log_fd.write(message + "\n")

        if level <= self.log_level:
            print(message)

    def debug(self, message, set_date = False):
        self.log(self.DEBUG, message, set_date)

    def info(self, message, set_date = False):
        self.log(self.INFO, message, set_date)

    def warning(self, message, set_date = False):
        self.log(self.WARNING, message, set_date)

    def error(self, message, set_date = False):
        self.log(self.ERROR, message, set_date)

    def fail(self, message, set_date = False):
        ''' Igual que 'error', pero acaba el programa '''
        self.error(message, set_date)
        sys.exit(1)