scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-11-03 20:27:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

lf_cisco_power.py: added switches for exit on failure and exit on error

Browse Source
This commit is contained in:
Chuck SmileyRekiere
2020-10-30 15:28:02 -06:00
parent 098855b2fa
commit 9bf180f070
1 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lf_cisco_power.py
View File

@@ -161,6 +161,12 @@ class FileAdapter(object):
def flush(self): def flush(self):
pass # leave it to logging to flush properly pass # leave it to logging to flush properly
def test_exit(workbook):
workbook.close()
sleep(0.5)
exit(1)
def main(): def main():
global lfmgr global lfmgr
global lfstation global lfstation
@@ -217,6 +223,8 @@ def main():
parser.add_argument("--cleanup", help="--cleanup , Clean up stations after test completes ", action='store_true') parser.add_argument("--cleanup", help="--cleanup , Clean up stations after test completes ", action='store_true')
parser.add_argument("--vht160", help="--vht160 , Enable VHT160 in lanforge ", action='store_true') parser.add_argument("--vht160", help="--vht160 , Enable VHT160 in lanforge ", action='store_true')
parser.add_argument("--verbose", help="--verbose , switch present will have verbose logging", action='store_true') parser.add_argument("--verbose", help="--verbose , switch present will have verbose logging", action='store_true')
parser.add_argument("--exit_on_fail", help="--exit_on_fail, exit on test failure", action='store_true')
parser.add_argument("--exit_on_error", help="--exit_on_error, exit on test error, test mechanics failed", action='store_true')
#current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + "{:.3f}".format(time.time() - (math.floor(time.time())))[1:] #current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + "{:.3f}".format(time.time() - (math.floor(time.time())))[1:]
#print(current_time) #print(current_time)
@@ -278,9 +286,9 @@ def main():
filehandler = None filehandler = None
except Exception as e: except Exception as e:
logging.exception(e); logging.exception(e)
usage() usage()
exit(2); exit(2)
console_handler = logging.StreamHandler() console_handler = logging.StreamHandler()
formatter = logging.Formatter(FORMAT) formatter = logging.Formatter(FORMAT)
@@ -1208,10 +1216,9 @@ def main():
print("_nss {} allowed_per_path (AP should be transmitting at) {}".format(_nss, allowed_per_path)) print("_nss {} allowed_per_path (AP should be transmitting at) {}".format(_nss, allowed_per_path))
if (pf == 0): if (pf == 0):
pfs = "FAIL" pfs = "FAIL"
time_stamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + "{:.3f}".format(time.time() - (math.floor(time.time())))[1:] time_stamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + "{:.3f}".format(time.time() - (math.floor(time.time())))[1:]
ln = "%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s"%( ln = "%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s"%(
@@ -1308,6 +1315,20 @@ def main():
csvs.write("\n"); csvs.write("\n");
csvs.flush() csvs.flush()
# write out the data and exit on failure
if (pf == 0):
if(args.exit_on_fail):
print("EXITING ON FAILURE, exit_on_fail set ")
workbook.close()
sleep(0.5)
exit(1)
if (e_tot != ""):
if(args.exit_on_error):
print("EXITING ON ERROR, exit_on_error set ")
workbook.close()
sleep(0.5)
exit(1)
workbook.close() workbook.close()
# check if keeping the existing state # check if keeping the existing state