From 621b85bc1867be25bbc982c7e5808ddc3499ca6a Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Fri, 23 Mar 2018 14:50:16 -0700
Subject: [PATCH] A little more progress on shell-log

---
 bin/shell-log | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)
 mode change 100644 => 100755 bin/shell-log

diff --git a/bin/shell-log b/bin/shell-log
old mode 100644
new mode 100755
index e8df22d..b3f267e
--- a/bin/shell-log
+++ b/bin/shell-log
@@ -6,8 +6,8 @@ Logging for my shells.
 
 import argparse
 import logging
+import os
 import os.path
-import sys
 
 LOGFILE = os.path.expanduser('~/.shell.log')
 
@@ -15,7 +15,7 @@ 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', 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('msg')
     parser.add_argument('args', nargs=argparse.REMAINDER)
@@ -26,21 +26,19 @@ def logger(args):
     logger = logging.getLogger('')
 
     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)
 
-    file_handler = logging.FileHandler(args.file)
-    logger.addHandler(file_handler)
-
     return logger
 
 def main():
     # Set up.
     args = parse_args()
-    log = logger()
-    
+    log = logger(args)
+
     # 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 it.