2Shirt
f1775766e7
* The log file is now moved instead of copied * The new path can now be based on a new dir, name, or both
111 lines
3.1 KiB
Python
111 lines
3.1 KiB
Python
"""WizardKit: Log Functions"""
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import atexit
|
|
import logging
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
import time
|
|
|
|
from wk import cfg
|
|
|
|
|
|
# Functions
|
|
def enable_debug_mode():
|
|
"""Configures logging for better debugging."""
|
|
root_logger = logging.getLogger()
|
|
for handler in root_logger.handlers:
|
|
formatter = logging.Formatter(
|
|
datefmt=cfg.log.DEBUG['datefmt'],
|
|
fmt=cfg.log.DEBUG['format'],
|
|
)
|
|
handler.setFormatter(formatter)
|
|
root_logger.setLevel('DEBUG')
|
|
|
|
|
|
def start(config=None):
|
|
"""Configure and start logging using safe defaults."""
|
|
if os.name == 'nt':
|
|
log_path = '{drive}/{short}/Logs/{date}/{full}/{datetime}.log'.format(
|
|
drive=os.environ.get('SYSTEMDRIVE', 'C:'),
|
|
short=cfg.main.KIT_NAME_SHORT,
|
|
date=time.strftime('%Y-%m-%d'),
|
|
full=cfg.main.KIT_NAME_FULL,
|
|
datetime=time.strftime('%Y-%m-%d_%H%M%S%z'),
|
|
)
|
|
else:
|
|
log_path = '{home}/Logs/{full}_{datetime}.log'.format(
|
|
home=os.path.expanduser('~'),
|
|
full=cfg.main.KIT_NAME_FULL,
|
|
datetime=time.strftime('%Y-%m-%d_%H%M%S%z'),
|
|
)
|
|
log_path = pathlib.Path(log_path).resolve()
|
|
root_logger = logging.getLogger()
|
|
|
|
# Safety checks
|
|
if not config:
|
|
config = cfg.log.DEFAULT
|
|
if root_logger.hasHandlers():
|
|
raise UserWarning('Logging already started, results may be unpredictable.')
|
|
|
|
# Create log_dir
|
|
os.makedirs(log_path.parent, exist_ok=True)
|
|
|
|
# Config logger
|
|
logging.basicConfig(filename=log_path, **config)
|
|
|
|
# Register shutdown to run atexit
|
|
atexit.register(logging.shutdown)
|
|
|
|
|
|
def update_log_path(dest_dir=None, dest_name=None):
|
|
"""Moves current log file to new path and updates the root logger.
|
|
|
|
NOTE: A timestamp and extension will be added to dest_name if provided.
|
|
"""
|
|
root_logger = logging.getLogger()
|
|
cur_handler = root_logger.handlers[0]
|
|
cur_path = pathlib.Path(cur_handler.baseFilename).resolve()
|
|
|
|
# Safety checks
|
|
if not (dest_dir or dest_name):
|
|
raise RuntimeError('Neither a directory nor name specified')
|
|
if len(root_logger.handlers) > 1:
|
|
raise RuntimeError('Multiple handlers not supported')
|
|
if not isinstance(cur_handler, logging.FileHandler):
|
|
raise RuntimeError('Only FileHandlers are supported')
|
|
|
|
# Update dir if specified or use current path
|
|
if dest_dir:
|
|
new_path = pathlib.Path(dest_dir).resolve()
|
|
else:
|
|
new_path = cur_path
|
|
|
|
# Update name if specified
|
|
if dest_name:
|
|
new_path = new_path.with_name(
|
|
f'{dest_name}'
|
|
f'_{time.strftime("%Y-%m-%d_%H%M%S%z")}'
|
|
f'{"".join(cur_path.suffixes)}'
|
|
)
|
|
else:
|
|
new_path = new_path.with_name(cur_path.name)
|
|
|
|
# Copy original log to new location
|
|
if new_path.exists():
|
|
raise FileExistsError(f'Refusing to clobber: {new_path}')
|
|
os.makedirs(new_path.parent, exist_ok=True)
|
|
shutil.move(cur_path, new_path)
|
|
|
|
# Create new cur_handler (preserving formatter settings)
|
|
new_handler = logging.FileHandler(new_path, mode='a')
|
|
new_handler.setFormatter(cur_handler.formatter)
|
|
|
|
# Replace current handler
|
|
root_logger.removeHandler(cur_handler)
|
|
root_logger.addHandler(new_handler)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print("This file is not meant to be called directly.")
|