corrected type of loglevel and timeout

ravchandra · ravchandra · commit d04e446b58fe · 2017-11-07T12:29:17.000+01:00
diff --git a/arguments.py b/arguments.py
@@ -22,7 +22,7 @@
         default=5)
 parser.add_argument("logfile", help="provide log file path", nargs="?",
         default="runshell.log")
-parser.add_argument("loglevel", help="provide log level", nargs="?",
+parser.add_argument("loglevel", help="provide log level as number", nargs="?",
         default=logging.INFO)
 if len(sys.argv) < 2:
     parser.print_help()
diff --git a/shell.py b/shell.py
@@ -1,3 +1,4 @@
+import logging
 import pexpect
 import subprocess
 from pexpect import pxssh
@@ -8,8 +9,7 @@
 import arguments
 
 args = arguments.args
-
-set_logger(logfile=args.logfile, loglevel=args.loglevel)
+set_logger(logfile=args.logfile, loglevel=int(args.loglevel))
 
 # Setting command type to interactive if given command is found in cmd_list
 interactive = False
@@ -29,7 +29,7 @@ def __init__(self, command, host, username, password, timeout):
         self.host = host
         self.username = username
         self.password = password
-        self.timeout = timeout
+        self.timeout = int(timeout)
 
     def remote(self):
         """
