61 lines
1.5 KiB
Python
Executable file
61 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Eryn Wells <eryn@erynwells.me>
|
|
'''
|
|
Logging for my shells.
|
|
'''
|
|
|
|
import argparse
|
|
import logging
|
|
import os
|
|
import os.path
|
|
|
|
_LOGFILE = os.path.expanduser('~/.shell.log')
|
|
_LOGGER = None
|
|
|
|
LOG_LEVELS = {
|
|
'crit': logging.CRITICAL,
|
|
'error': logging.ERROR,
|
|
'warn': logging.WARNING,
|
|
'info': logging.INFO,
|
|
'debug': logging.DEBUG,
|
|
}
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-b', '--bold', action='store_true')
|
|
parser.add_argument('-f', '--file', default=_LOGFILE)
|
|
parser.add_argument('-l', '--level', default='info', choices=list(LOG_LEVELS.keys()))
|
|
parser.add_argument('-n', '--noeol', action='store_true')
|
|
parser.add_argument('msg')
|
|
parser.add_argument('args', nargs=argparse.REMAINDER)
|
|
args = parser.parse_args()
|
|
return args
|
|
|
|
def logger(args):
|
|
logger = logging.getLogger('')
|
|
|
|
stdout_handler = logging.StreamHandler()
|
|
formatter = logging.Formatter('\033[34m==>\033[39m {msg}', style='{')
|
|
stdout_handler.setFormatter(formatter)
|
|
logger.addHandler(stdout_handler)
|
|
|
|
# TODO: Add file handler
|
|
|
|
return logger
|
|
|
|
def main():
|
|
args = parse_args()
|
|
log = logger(args)
|
|
|
|
level = LOG_LEVELS[os.environ.get('SHELL_LOG_LEVEL', 'info')]
|
|
log.setLevel(level)
|
|
|
|
try:
|
|
message_log_level = LOG_LEVELS[args.level]
|
|
except KeyError:
|
|
message_log_level = logging.INFO
|
|
|
|
log.log(message_log_level, args.msg, *args.args)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|