added log level option

diff --git a/src/logalert.py b/src/logalert.py
index e2f02ccbd40f53c8a763f6d2435778374f9566f0..c8eed192577ef7bba95265168eef10ca6eeadac7 100755 (executable)
--- a/src/logalert.py
+++ b/src/logalert.py
@@ -9,10 +9,6 @@ import misc
 MAX_LINES = 10
 ALERT_INTERVAL = 86400
 
-logging.basicConfig(format='[%(asctime)s] %(levelname)s: %(message)s',
-                    level=logging.INFO,
-                    datefmt='%m/%d/%Y %H:%M:%S')
-
 parser = argparse.ArgumentParser(description='Alert on excessive number of error log lines.')
 parser.add_argument('logfile', type=str, help='logfile to be watched')
 parser.add_argument('handler', type=str,
@@ -21,9 +17,15 @@ parser.add_argument('-s', '--interval-size', type=int, default=600, dest='interv
                     help='sample interval size in seconds (default: 600)')
 parser.add_argument('-n', '--num-intervals', type=int, default=6, dest='num_intervals',
                     help='number of intervals to keep in history (default: 6)')
+parser.add_argument('-l', '--log-level', type=str, default='INFO', dest='log_lvl',
+                    choices=['DEBUG', 'INFO', 'WARNING'], help='select log level')
 
 args = parser.parse_args()
 
+logging.basicConfig(format='[%(asctime)s] %(levelname)s: %(message)s',
+                    level=logging.getLevelName(args.log_lvl),
+                    datefmt='%m/%d/%Y %H:%M:%S')
+
 kept_times = []
 lines = []
 last_slot_time = None