Changeset 10098 in genesis for tools

Timestamp:

Mar 10, 2012, 9:55:07 PM (13 years ago)

Author:

rick

Message:

Procedure of enabling SNMP seems to work reliable now.

File:

• tools/check-batch-cmd (modified) (5 diffs)

tools/check-batch-cmd

@@ -8 +8 @@
 import argparse
 import gformat
+import getpass
 import netsnmp
 import os
@@ -15 +16 @@
 import time
 
-try:
-  SSHPASS = os.environ['SSHPASS']
-except KeyError:
-  SSHPASS = 'SSH_NOT_SET'
-
+SSHPASS = None
 netsnmp.verbose = 0
 
@@ -49 +46 @@
   retval = sess.get(var_list)
   if sess.ErrorInd < 0:
-    raise CmdError('[%(ErrorInd)s] %(ErrorStr)s (%(DestHost)s)' % vars(sess))
+    raise CmdError('SNMP Failed -- [%(ErrorInd)s] %(ErrorStr)s (%(DestHost)s)' % vars(sess))
   return filter(None, retval)[0]
 
@@ -125 +122 @@
   # create the top-level parser
   parser = argparse.ArgumentParser(prog='Various WL management tools')
-  #parser.add_argument('--foo', action='store_true', help='foo help')
+  parser.add_argument('--ask-pass', dest="ask_pass", action='store_true', help='Ask password if SSHPASS is not found')
   subparsers = parser.add_subparsers(help='sub-command help')
   
@@ -142 +139 @@
   args = parser.parse_args()
 
+  try:
+    SSHPASS = os.environ['SSHPASS']
+  except KeyError:
+    print "#WARN: SSHPASS environ variable not found"
+    if args.ask_pass:
+      SSHPASS = getpass.getpass("WL root password: ")
+
+
   if args.func == 'keys':
     ubnt_keys(args.host)