piStor/logger.py at main · JeffCurless/piStor · 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
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
95
96
#
#
#
import logging

#
# Logger - A class that is used to wrap the logging class... so we can export
#          directly to the console in debug mode, and handle any other change
#          required
#
class Logger:
    '''
    Initialize the logger.
    '''
    def __init__(self,filename=None,level=logging.DEBUG):
        self._logEnabled = True
        try:
            if filename == None:
                self._logEnabled = False
            else:
                logging.basicConfig( filename=filename,
                                     filemode='a',
                                     level=level,
                                     format='%(asctime)s %(process)d [%(levelname)s] %(message)s',
                                     datefmt='%b %d %y %H:%M:%S')
        except Exception as error:
            print( f"Error: Could not log events due to {error}." )
            self._logEnabled = False

    def info( self, message ):
        '''
        Write the message out using the info level

        Parameters:
            message - the message to display to the destination

        '''
        if self._logEnabled:
            logging.info( message )
        else:
            print( f"INFO: {message}" )

    def debug( self, message ):
        '''
        Write the message out using the deug level

        Parameters:
            message - The message to log

        '''
        if self._logEnabled:
            logging.debug( message )
        else:
            print( f"DEBUG: {message}" )

    def warn( self, message ):
        '''
        Write the message out using the warn level

        Parameters:
            message - The message to log

        '''
        if self._logEnabled:
            logging.warning( message )
        else:
            print( f"WARN: {message}" )

    def warning( self, message ):
        self.warn( message )

    def error( self, message ):
        '''
        Write the message out using the error level

        Parameters:
            message - The message to log

        '''
        if self._logEnabled:
            logging.error( message )
        else:
            print( f"(ERROR: {message}" )

    def forceConsole(self):
        '''
        This routine forces all logging messages to the console.

        '''
        self._logEnabled = False
#
#
#
if __name__ == "__main__":
    log = Logger("test.log")
    log.warn( "Testing..." )