eryn/dotfiles
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

A little more progress on shell-log

Browse source
This commit is contained in:
Eryn Wells 2018-03-23 14:50:16 -07:00
parent 2ac09bc6fa
commit 621b85bc18
1 changed files with 7 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

16
bin/shell-log Normal file → Executable file
View file

@ -6,8 +6,8 @@ Logging for my shells.
import argparse import argparse
import logging import logging
import os
import os.path import os.path
import sys
LOGFILE = os.path.expanduser('~/.shell.log') LOGFILE = os.path.expanduser('~/.shell.log')
@ -15,7 +15,7 @@ def parse_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-b', '--bold', action='store_true') parser.add_argument('-b', '--bold', action='store_true')
parser.add_argument('-f', '--file', default=LOGFILE) parser.add_argument('-f', '--file', default=LOGFILE)
parser.add_argument('-l', '--level', type=int, default=0) parser.add_argument('-l', '--level', type=int, default=logging.INFO)
parser.add_argument('-n', '--noeol', action='store_true') parser.add_argument('-n', '--noeol', action='store_true')
parser.add_argument('msg') parser.add_argument('msg')
parser.add_argument('args', nargs=argparse.REMAINDER) parser.add_argument('args', nargs=argparse.REMAINDER)
@ -26,21 +26,19 @@ def logger(args):
logger = logging.getLogger('') logger = logging.getLogger('')
stdout_handler = logging.StreamHandler() stdout_handler = logging.StreamHandler()
# TODO: Set formatter correctly. formatter = logging.Formatter('\033[34m==>\033[39m {msg}', style='{')
stdout_handler.setFormatter(formatter)
logger.addHandler(stdout_handler) logger.addHandler(stdout_handler)
file_handler = logging.FileHandler(args.file)
logger.addHandler(file_handler)
return logger return logger
def main(): def main():
# Set up. # Set up.
args = parse_args() args = parse_args()
log = logger() log = logger(args)
# Set the level from the shell. # Set the level from the shell.
level = sys.environ.get('NOISY', logging.INFO) level = int(os.environ.get('NOISY', logging.INFO))
log.setLevel(level) log.setLevel(level)
# Log it. # Log it.