File size: 1,561 Bytes
970a7a2 |
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 |
# An Advanced Logger class which writes data in a well formatted manner
# to files based on different priorities.
from enum import Enum
import os
import datetime
import time
class LogPriority(Enum):
"""
Enum class for different log priorities.
"""
LOW = 0
MEDIUM = 1
HIGH = 2
STATS = 3
class AdvancedLogger:
def __init__(self, base_dir):
self.base_dir = base_dir
self.files = []
self.file_names = []
for p in LogPriority:
self.file_names.append(os.path.join(self.base_dir, f'Log_{p.name}' + '.log'))
self.files.append(open(self.file_names[-1], 'w'))
self.last_log_time = -1
def flush(self):
for f in self.files:
f.close()
for i in range(len(self.files)):
self.files[i] = open(self.file_names[i], 'a')
def log(self, *args, priority = LogPriority.LOW):
to_log = ' '.join(map(str, args))
if priority.value <= LogPriority.MEDIUM.value:
# Add current time to to_log
now = datetime.datetime.now()
to_log = f'[{now.strftime("%H:%M:%S")}]: {to_log}'
print(to_log)
for p in range(priority.value+1):
self.files[p].write(to_log + '\n')
# If time - last_log_time is greater than 10s or Priority is HIGH or above close the file and re-open in append mode
if time.time() - self.last_log_time > 10 or priority.value >= LogPriority.HIGH.value:
self.flush()
self.last_log_time = time.time()
|