-
Notifications
You must be signed in to change notification settings - Fork 1
/
logger.py
66 lines (51 loc) · 2.12 KB
/
logger.py
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
from os import path, makedirs
from datetime import datetime
from enum import Enum
class LogLevel(Enum):
OFF = 1
MINIMUM = 2
NORMAL = 3
DEBUG = 4
class Logger(object):
def __init__(self, full_name, log_level=LogLevel.DEBUG):
module_name = path.splitext(path.basename(full_name))[0]
self.log_name = module_name + '.log'
logs_folder = 'logs'
if not path.exists(logs_folder):
makedirs(logs_folder, exist_ok = True)
self.log = path.join(logs_folder, self.log_name)
self.create_log()
self.logging_level = log_level
def create_log(self):
with open(self.log, mode='w', encoding='utf-8') as log_file:
log_file.write(self.get_date_time() + '\t\t*** Starting Log ***\n')
log_file.close()
def get_date_time(self):
return datetime.now().strftime("%m-%d-%Y %H:%M:%S")
def set_logging_level(self, level):
self.logging_level = level
def write_to_log(self, msg='', log_level=LogLevel.DEBUG):
if log_level.value > self.logging_level.value:
return
with open(self.log, mode='a', encoding='utf-8') as log_file:
if msg.startswith('\n'):
msg = msg[1:]
log_file.write(self.get_date_time() + '\n')
msg = f'{log_level.name}: {msg}'
if msg.endswith('\n'):
log_file.write(self.get_date_time() + '\t\t' + msg)
log_file.write(self.get_date_time() + '\n')
else:
log_file.write(self.get_date_time() + '\t\t' + msg + '\n')
log_file.close()
'''if __name__== '__main__':
logger = Logger(__file__, log_level=LogLevel.NORMAL)
logger.write_to_log('test regular', log_level=LogLevel.MINIMUM)
logger.write_to_log('\ntest with newline leading', log_level=LogLevel.NORMAL)
logger.write_to_log('ending with \n', log_level=LogLevel.NORMAL)
logger.set_logging_level(LogLevel.DEBUG)
logger.write_to_log('Final line', log_level=LogLevel.DEBUG)
full_path, ext = path.splitext(__file__)
new_full_path = full_path + '-OTHER-' + ext
another_logger = Logger(new_full_path, log_level=LogLevel.MINIMUM)
another_logger.write_to_log('Other module message', log_level=LogLevel.MINIMUM)'''