128 lines
3.2 KiB
Python
128 lines
3.2 KiB
Python
import machine
|
|
import os
|
|
import gc
|
|
|
|
from . import util
|
|
|
|
log_file = 'log.txt'
|
|
|
|
LOG_INFO = 0b00001
|
|
LOG_WARNING = 0b00010
|
|
LOG_ERROR = 0b00100
|
|
LOG_DEBUG = 0b01000
|
|
LOG_EXCEPTION = 0b10000
|
|
LOG_ALL = LOG_INFO | LOG_WARNING | LOG_ERROR | LOG_DEBUG | LOG_EXCEPTION
|
|
|
|
_logging_types = LOG_ALL
|
|
|
|
# the log file will be truncated if it exceeds _log_truncate_at bytes in
|
|
# size. the defaults values are designed to limit the log to at most
|
|
# three blocks on the Pico
|
|
_log_truncate_at = 11 * 1024
|
|
_log_truncate_to = 8 * 1024
|
|
|
|
|
|
def file_size(file):
|
|
try:
|
|
return os.stat(file)[6]
|
|
except OSError:
|
|
return None
|
|
|
|
|
|
def set_truncate_thresholds(truncate_at, truncate_to):
|
|
global _log_truncate_at
|
|
global _log_truncate_to
|
|
_log_truncate_at = truncate_at
|
|
_log_truncate_to = truncate_to
|
|
|
|
|
|
def enable_logging_types(types):
|
|
global _logging_types
|
|
_logging_types = _logging_types | types
|
|
|
|
|
|
def disable_logging_types(types):
|
|
global _logging_types
|
|
_logging_types = _logging_types & ~types
|
|
|
|
|
|
# truncates the log file down to a target size while maintaining
|
|
# clean line breaks
|
|
def truncate(file, target_size):
|
|
# get the current size of the log file
|
|
size = file_size(file)
|
|
|
|
# calculate how many bytes we're aiming to discard
|
|
discard = size - target_size
|
|
if discard <= 0:
|
|
return
|
|
|
|
with open(file, 'rb') as infile:
|
|
with open(file + '.tmp', 'wb') as outfile:
|
|
# skip a bunch of the input file until we've discarded
|
|
# at least enough
|
|
while discard > 0:
|
|
chunk = infile.read(1024)
|
|
discard -= len(chunk)
|
|
|
|
# try to find a line break nearby to split first chunk on
|
|
break_position = max(
|
|
chunk.find(b'\n', -discard), # search forward
|
|
chunk.rfind(b'\n', 0, -discard) # search backwards
|
|
)
|
|
if break_position != -1: # if we found a line break..
|
|
outfile.write(chunk[break_position + 1:])
|
|
|
|
# now copy the rest of the file
|
|
while True:
|
|
chunk = infile.read(1024)
|
|
if not chunk:
|
|
break
|
|
outfile.write(chunk)
|
|
|
|
# delete the old file and replace with the new
|
|
os.remove(file)
|
|
os.rename(file + '.tmp', file)
|
|
|
|
|
|
def log(level, text):
|
|
log_entry = '{0} [{1:8} /{2:>4}kB] {3}'.format(
|
|
util.datetime(),
|
|
level,
|
|
round(gc.mem_free() / 1024),
|
|
text
|
|
)
|
|
|
|
print(log_entry)
|
|
|
|
with open(log_file, 'a') as logfile:
|
|
logfile.write(log_entry + '\n')
|
|
|
|
if _log_truncate_at and file_size(log_file) > _log_truncate_at:
|
|
truncate(log_file, _log_truncate_to)
|
|
|
|
|
|
def info(*items):
|
|
if _logging_types & LOG_INFO:
|
|
log('info', ' '.join(map(str, items)))
|
|
|
|
|
|
def warn(*items):
|
|
if _logging_types & LOG_WARNING:
|
|
log('warning', ' '.join(map(str, items)))
|
|
|
|
|
|
def error(*items):
|
|
if _logging_types & LOG_ERROR:
|
|
log('error', ' '.join(map(str, items)))
|
|
|
|
|
|
def debug(*items):
|
|
if _logging_types & LOG_DEBUG:
|
|
log('debug', ' '.join(map(str, items)))
|
|
|
|
|
|
def exception(*items):
|
|
if _logging_types & LOG_EXCEPTION:
|
|
log('exception', ' '.join(map(str, items)))
|