From ce29408579581bb6c90eee7d9178678eda0a2442 Mon Sep 17 00:00:00 2001
From: Chuck SmileyRekiere <chuck.smileyrekiere@candelatech.com>
Date: Fri, 22 Jan 2021 07:42:20 -0700
Subject: [PATCH] lf_ap_power.py: LCS-45 Read AP for Power values Updated login
 to help with multiple logins lf_cisco_power.py: comments update

---
 cisco_ap_ctl.py   | 83 +++++++++++++++++++++++++++++++----------------
 lf_cisco_power.py |  5 +--
 2 files changed, 56 insertions(+), 32 deletions(-)

diff --git a/cisco_ap_ctl.py b/cisco_ap_ctl.py
index 059b4a16..654e7150 100644
--- a/cisco_ap_ctl.py
+++ b/cisco_ap_ctl.py
@@ -20,12 +20,12 @@ if sys.version_info[0] != 3:
     print("This script requires Python 3")
     exit()
 
-import re
+#import re
 import logging
 import time
 from time import sleep
-import pprint
-import telnetlib
+#import pprint
+#import telnetlib
 import argparse
 import pexpect
 
@@ -95,10 +95,8 @@ def main():
         scheme = args.scheme
         port = (default_ports[scheme], args.port)[args.port != None]
         user = args.user
-        passwd = args.passwd
         if (args.log != None):
             logfile = args.log
-        filehandler = None
     except Exception as e:
         logging.exception(e)
         usage()
@@ -159,31 +157,60 @@ def main():
     except Exception as e:
         logging.exception(e)
     
-    ap_prompt = "{}>".format(args.prompt)
-    ap_hash = "{}#".format(args.prompt)
-    egg.sendline()
+    AP_PROMPT       = "{}>".format(args.prompt)
+    AP_HASH         = "{}#".format(args.prompt)
+    AP_ESCAPE       = "Escape character is '^]'."
+    AP_USERNAME     = "Username:"
+    AP_PASSWORD     = "Password:"
+    AP_EN           = "en"
+    AP_MORE         = "--More--"
+    AP_EXIT         = "exit"
+    CR = "\r\n"
+    time.sleep(0.1)
+    logged_in  = False
+    loop_count = 0
+    while (loop_count <= 8 and logged_in == False):
+        loop_count += 1
+        i = egg.expect_exact([AP_ESCAPE,AP_PROMPT,AP_HASH,AP_USERNAME,AP_PASSWORD,pexpect.TIMEOUT],timeout=5)
+        if i == 0:
+            logg.info("Expect: {} i: {} before: {} after: {}".format(AP_ESCAPE,i,egg.before,egg.after))
+            egg.sendline(CR) # Needed after Escape or should just do timeout and then a CR?
+            sleep(0.2)
+        if i == 1:
+            logg.info("Expect: {} i: {} before: {} after: {}".format(AP_PROMPT,i,egg.before,egg.after))
+            egg.sendline(AP_EN) 
+            sleep(0.2)
+        if i == 2:
+            logg.info("Expect: {} i: {} before: {} after: {}".format(AP_HASH,i,egg.before,egg.after))
+            logged_in = True 
+            sleep(0.2)
+        if i == 3:
+            logg.info("Expect: {} i: {} before: {} after: {}".format(AP_USERNAME,i,egg.before,egg.after))
+            egg.sendline(args.user) 
+            sleep(0.2)
+        if i == 4:
+            logg.info("Expect: {} i: {} before: {} after: {}".format(AP_PASSWORD,i,egg.before,egg.after))
+            egg.sendline(args.passwd) 
+            sleep(0.2)
+        if i == 5:
+            logg.info("Expect: {} i: {} before: {} after: {}".format("Timeout",i,egg.before,egg.after))
+            egg.sendline(CR) 
+            sleep(0.2)
 
-    egg.sendline()
-    time.sleep(0.1)
-    egg.expect('Username:', timeout=3)
-    time.sleep(0.1)
-    egg.sendline(args.user)
-    time.sleep(0.1)
-    egg.expect('Password:')
-    egg.sendline(args.passwd)
-    egg.expect(ap_prompt)
-    egg.sendline("en")
-    egg.expect("Password:")
-    egg.sendline(args.passwd)
-    egg.expect("#")
-    egg.sendline('show controllers dot11Radio 1 powercfg | g T1')
-    egg.expect("--More--")
-    egg.sendcontrol('c')
-    egg.expect("#",timeout=5)
-    egg.sendline("exit")
-    egg.expect(">",timeout=5)
-    egg.sendline("exit")
+    if (args.action == "powercfg"):
+        egg.sendline('show controllers dot11Radio 1 powercfg | g T1')
+        egg.expect_exact(AP_MORE,timeout=5)
+        egg.sendcontrol('c')
 
+    else: # no other command at this time so send the same power command
+        egg.sendline('show controllers dot11Radio 1 powercfg | g T1')
+        egg.expect_exact(AP_MORE,timeout=5)
+        egg.sendcontrol('c')
+
+    egg.expect_exact(AP_HASH,timeout=5)
+    egg.sendline(AP_EXIT)
+    egg.expect_exact(AP_PROMPT,timeout=5)
+    egg.sendline(AP_EXIT)
     #             ctlr.execute(cn_cmd)
 if __name__ == '__main__':
     main()
diff --git a/lf_cisco_power.py b/lf_cisco_power.py
index 1a0dfdfe..543248e5 100755
--- a/lf_cisco_power.py
+++ b/lf_cisco_power.py
@@ -1599,7 +1599,7 @@ def main():
                                logg.info("####################################################################################################") 
                       
                                logg.info("####################################################################################################") 
-                               logg.info("# Unable to commicate to AP or unable to communicate to controller error code: {} output {}".format(process_error.returncode, process_error.output)) 
+                               logg.info("# Unable to commicate to AP  error code: {} output {}".format(process_error.returncode, process_error.output)) 
                                logg.info("####################################################################################################") 
                                #exit_test(workbook)
                                exit(1)
@@ -1626,9 +1626,6 @@ def main():
                                 else:
                                     logg.info("AP Check regular expression!!!")
                                     exit(1)
-
-
-
                        #
                        #  The controller may adjust the number of spatial streams to allow for the 
                        #  best power values