Updated methods to fit PEP8 standards

This commit is contained in:
Logan Lipke
2020-10-15 15:36:10 -07:00
parent b0ffdb4e56
commit 24e510a848
2 changed files with 50 additions and 3 deletions

View File

@@ -46,6 +46,9 @@ class LFRequest:
# request first url on stack
def formPost(self, show_error=True, debug=False, die_on_error_=False):
return self.form_post(show_error=show_error, debug=debug, die_on_error_=die_on_error_)
def form_post(self, show_error=True, debug=False, die_on_error_=False):
if self.die_on_error:
die_on_error_ = True
if (debug == False) and (self.debug == True):
@@ -100,6 +103,9 @@ class LFRequest:
return None
def jsonPost(self, show_error=True, debug=False, die_on_error_=False, response_json_list_=None):
return self.json_post(show_error=show_error, debug=debug, die_on_error_=die_on_error_, response_json_list_=response_json_list_)
def json_post(self, show_error=True, debug=False, die_on_error_=False, response_json_list_=None):
if (debug == False) and (self.debug == True):
debug = True
if self.die_on_error:
@@ -215,6 +221,9 @@ class LFRequest:
return None
def getAsJson(self, die_on_error_=False, debug_=False):
return self.get_as_json(die_on_error_=die_on_error_, debug_=debug_)
def get_as_json(self, die_on_error_=False, debug_=False):
responses = []
j = self.get(debug=debug_, die_on_error_=die_on_error_)
responses.append(j)
@@ -228,6 +237,9 @@ class LFRequest:
return json_data
def addPostData(self, data):
self.add_post_data(data=data)
def add_post_data(self, data):
self.post_data = data

View File

@@ -196,6 +196,9 @@ def port_reset_request(resource_id, port_name, debug_on=False):
def generateMac(parent_mac, random_octet, debug=False):
return generate_mac(parent_mac=parent_mac, random_octet=random_octet, debug=debug)
def generate_mac(parent_mac, random_octet, debug=False):
if debug:
print("************ random_octet: %s **************" % (random_octet))
my_oct = random_octet
@@ -248,8 +251,12 @@ def port_name_series(prefix="sta", start_id=0, end_id=1, padding_number=10000, r
return name_list
# generate random hex if you need it for mac addresses
def generateRandomHex():
return generate_random_hex()
# generate random hex if you need it for mac addresses
def generate_random_hex():
# generate a few random numbers and convert them into hex:
random_hex = []
for rn in range(0, 254):
@@ -270,11 +277,15 @@ def generateRandomHex():
# },
# { ... }
def portListToAliasMap(json_list, debug_=False):
return port_list_to_alias_map(json_list=json_list, debug_=debug_)
def port_list_to_alias_map(json_list, debug_=False):
reverse_map = {}
if (json_list is None) or (len(json_list) < 1):
if debug_:
print("portListToAliasMap: no json_list provided")
raise ValueError("portListToAliasMap: no json_list provided")
print("port_list_to_alias_map: no json_list provided")
raise ValueError("port_list_to_alias_map: no json_list provided")
return reverse_map
json_interfaces = json_list
@@ -301,6 +312,10 @@ def portListToAliasMap(json_list, debug_=False):
def findPortEids(resource_id=1, base_url="http://localhost:8080", port_names=(), debug=False):
return find_port_eids(resource_id=resource_id, base_url=base_url, port_names=port_names, debug=debug)
def find_port_eids(resource_id=1, base_url="http://localhost:8080", port_names=(), debug=False):
port_eids = []
if len(port_names) < 0:
return []
@@ -319,6 +334,10 @@ def findPortEids(resource_id=1, base_url="http://localhost:8080", port_names=(),
def waitUntilPortsAdminDown(resource_id=1, base_url="http://localhost:8080", port_list=()):
return wait_until_ports_admin_down(resource_id=resource_id, base_url=base_url, port_list=port_list)
def wait_until_ports_admin_down(resource_id=1, base_url="http://localhost:8080", port_list=()):
print("Waiting until ports appear admin-down...")
up_stations = port_list.copy()
sleep(1)
@@ -505,6 +524,10 @@ def wait_until_endps(base_url="http://localhost:8080", endp_list=(), debug=False
def removePort(resource, port_name, baseurl="http://localhost:8080/", debug=False):
remove_port(resource=resource, port_name=port_name, baseurl=baseurl, debug=debug)
def remove_port(resource, port_name, baseurl="http://localhost:8080/", debug=False):
if debug:
print("Removing port %d.%s" % (resource, port_name))
url = "/cli-json/rm_vlan"
@@ -518,6 +541,10 @@ def removePort(resource, port_name, baseurl="http://localhost:8080/", debug=Fals
def removeCX(baseurl, cx_names, debug=False):
remove_cx(baseurl=baseurl, cx_names=cx_names, debug=debug)
def remove_cx(baseurl, cx_names, debug=False):
if debug:
print("Removing cx %s" % ", ".join(cx_names))
url = "/cli-json/rm_cx"
@@ -532,6 +559,10 @@ def removeCX(baseurl, cx_names, debug=False):
def removeEndps(baseurl, endp_names, debug=False):
remove_endps(baseurl=baseurl, endp_names=endp_names, debug=debug)
def remove_endps(baseurl, endp_names, debug=False):
if debug:
print("Removing endp %s" % ", ".join(endp_names))
url = "/cli-json/rm_endp"
@@ -545,6 +576,10 @@ def removeEndps(baseurl, endp_names, debug=False):
def execWrap(cmd):
exec_wrap(cmd=cmd)
def exec_wrap(cmd):
if os.system(cmd) != 0:
print("\nError with '" + cmd + "', bye\n")
exit(1)