mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-01 19:28:00 +00:00
create_wanlink: this rewritten version supercedes py-json/create_wanlink.py; does not trigger half-created rdd interfaces
Signed-off-by: Jed Reynolds <jed@candelatech.com>
This commit is contained in:
Jed Reynolds
308
py-scripts/create_wanlink.py
Executable file
308
py-scripts/create_wanlink.py
Executable file
@@ -0,0 +1,308 @@
|
||||
#!/usr/bin/python3
|
||||
# Create and modify WAN Links Using LANforge JSON AP : http://www.candelatech.com/cookbook.php?vol=cli&book=JSON:+Managing+WANlinks+using+JSON+and+Python
|
||||
# Written by Candela Technologies Inc.
|
||||
# Updated by: Erin Grimes
|
||||
import sys
|
||||
|
||||
if sys.version_info[0] != 3:
|
||||
print("This script requires Python 3")
|
||||
exit(1)
|
||||
|
||||
|
||||
|
||||
|
||||
from time import sleep
|
||||
import urllib
|
||||
from urllib import error
|
||||
import pprint
|
||||
|
||||
sys.path.append("../py-json")
|
||||
from LANforge import LFRequest
|
||||
from LANforge import LFUtils
|
||||
|
||||
|
||||
j_printer = pprint.PrettyPrinter(indent=2)
|
||||
# todo: this needs to change
|
||||
resource_id = 1
|
||||
|
||||
|
||||
def main(base_url, args=()):
|
||||
print(base_url)
|
||||
json_post = ""
|
||||
json_response = ""
|
||||
num_wanlinks = -1
|
||||
|
||||
# force a refresh on the ports and wanlinks
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/nc_show_ports", debug_=True)
|
||||
lf_r.addPostData({
|
||||
"shelf": 1,
|
||||
"resource": 1,
|
||||
"port": "all",
|
||||
})
|
||||
json_response = lf_r.jsonPost(debug=True)
|
||||
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/nc_show_endpoints", debug_=True)
|
||||
lf_r.addPostData({
|
||||
"endpoint": "all"
|
||||
})
|
||||
json_response = lf_r.jsonPost(debug=True)
|
||||
|
||||
sleep(1)
|
||||
|
||||
# see if there are old wanlinks to remove
|
||||
lf_r = LFRequest.LFRequest(base_url+"/wl_ep/list", debug_=True)
|
||||
json_reponse = lf_r.get_as_json()
|
||||
|
||||
endpA = args['name']+"-A"
|
||||
endpB = args['name']+"-B"
|
||||
|
||||
# count the number of wanlink endpoints
|
||||
if "endpoint" in json_response:
|
||||
endpoint_map = LFUtils.list_to_alias_map(json_list=json_reponse, from_element="endpoint")
|
||||
if endpA in endpoint_map:
|
||||
num_wanlinks += 1
|
||||
if endpB in endpoint_map:
|
||||
num_wanlinks += 1
|
||||
|
||||
# remove old wanlinks
|
||||
if (num_wanlinks > 0):
|
||||
print("Removing old wanlinks...")
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/rm_cx", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'test_mgr': 'all',
|
||||
'cx_name': args['name']
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/rm_endp", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'endp_name': endpA
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/rm_endp", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'endp_name': endpB
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
sleep(1)
|
||||
|
||||
# check to see if we have ports
|
||||
lf_r = LFRequest.LFRequest(base_url+"/ports/1/1/list", debug_=True)
|
||||
port_response = lf_r.getAsJson()
|
||||
|
||||
if "interfaces" not in port_response:
|
||||
print("No interfaces in port_response!")
|
||||
pprint.pprint(port_response)
|
||||
exit(1)
|
||||
|
||||
if "interfaces" in port_response:
|
||||
port_map = LFUtils.list_to_alias_map(json_list=port_response, from_element="interfaces")
|
||||
ports_created = 0
|
||||
if args["port_A"] not in port_map:
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/add_rdd", debug_=True)
|
||||
lf_r.addPostData({
|
||||
"shelf": 1,
|
||||
"resource": 1,
|
||||
"port": args["port_A"],
|
||||
"peer_ifname": args["port_A"]+"b",
|
||||
})
|
||||
json_reponse = lf_r.jsonPost(debug=True)
|
||||
if not json_response:
|
||||
print("could not create port "+args["port_A"])
|
||||
exit(1)
|
||||
sleep(0.1)
|
||||
ports_created += 1
|
||||
if args["port_B"] not in port_map:
|
||||
lf_r.addPostData({
|
||||
"shelf": 1,
|
||||
"resource": 1,
|
||||
"port": args["port_B"],
|
||||
"peer_ifname": args["port_B"]+"b",
|
||||
})
|
||||
json_reponse = lf_r.jsonPost(debug=True)
|
||||
if not json_response:
|
||||
print("could not create port " + args["port_B"])
|
||||
exit(1)
|
||||
ports_created += 1
|
||||
sleep(0.1)
|
||||
if ports_created > 0:
|
||||
LFUtils.wait_until_ports_appear(base_url=base_url,
|
||||
port_list=(args["port_A"], args["port_B"]),
|
||||
debug=True)
|
||||
print("Created {} ports".format(ports_created))
|
||||
|
||||
# create wanlink endpoint A
|
||||
print("Adding WL Endpoints...", end='')
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/add_wl_endp", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'alias': endpA,
|
||||
'shelf': 1,
|
||||
'resource': '1',
|
||||
'port': args['port_A'],
|
||||
'latency': args['latency_A'],
|
||||
'max_rate': args['rate_A'],
|
||||
})
|
||||
json_response = lf_r.jsonPost(debug=True)
|
||||
if not json_response:
|
||||
print("Unable to create "+endpA)
|
||||
else:
|
||||
print("A, ", end='')
|
||||
# create wanlink endpoint B
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/add_wl_endp", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'alias': endpB,
|
||||
'shelf': 1,
|
||||
'resource': '1',
|
||||
'port': args['port_B'],
|
||||
'latency': args['latency_B'],
|
||||
'max_rate': args['rate_B'],
|
||||
})
|
||||
json_response = lf_r.jsonPost()
|
||||
if not json_response:
|
||||
print("Unable to create "+endpB)
|
||||
else:
|
||||
print("B")
|
||||
sleep(1)
|
||||
|
||||
# create cx
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/add_cx", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'alias': args['name'],
|
||||
'test_mgr': 'default_tm',
|
||||
'tx_endp': endpA,
|
||||
'rx_endp': endpB
|
||||
})
|
||||
lf_r.jsonPost(debug=True)
|
||||
sleep(0.5)
|
||||
|
||||
# modify wanlink endpoint A
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/set_wanlink_info", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'name': endpA,
|
||||
'max_jitter': args['jitter_A'],
|
||||
'jitter_freq': args['jitter_freq_A'],
|
||||
'drop_freq': args['drop_A']
|
||||
})
|
||||
lf_r.jsonPost(debug=True)
|
||||
|
||||
# modify wanlink endpoint B
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/set_wanlink_info", debug_=True)
|
||||
lf_r.addPostData({
|
||||
'name': endpB,
|
||||
'max_jitter': args['jitter_B'],
|
||||
'jitter_freq': args['jitter_freq_B'],
|
||||
'drop_freq': args['drop_B']
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
|
||||
# start wanlink once we see it
|
||||
seen = 0
|
||||
print("Looking for {} and {}: ".format(endpA, endpB), end='')
|
||||
while (seen < 2):
|
||||
sleep(1)
|
||||
lf_r = LFRequest.LFRequest(base_url+"/wl_ep/list?fields=name,eid")
|
||||
try:
|
||||
json_response = lf_r.getAsJson()
|
||||
if json_response is None:
|
||||
print(".", end="")
|
||||
continue
|
||||
LFUtils.debug_printer.pprint(json_response)
|
||||
if "endpoint" not in json_response:
|
||||
print("-", end="")
|
||||
continue
|
||||
|
||||
endpoint_map = LFUtils.list_to_alias_map(json_list=json_response["endpoint"],
|
||||
from_element="endpoint")
|
||||
if endpA in endpoint_map:
|
||||
seen += 1
|
||||
print("+", end="")
|
||||
if endpB in endpoint_map:
|
||||
seen += 1
|
||||
print("+", end="")
|
||||
|
||||
except urllib.error.HTTPError as error:
|
||||
print("Error code {}".format(error.code))
|
||||
continue
|
||||
print("")
|
||||
print("Starting wanlink:")
|
||||
# print("the latency is {laten}".format(laten=latency))
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/set_cx_state")
|
||||
lf_r.addPostData({
|
||||
'test_mgr': 'all',
|
||||
'cx_name': args['name'],
|
||||
'cx_state': 'RUNNING'
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
|
||||
running = 0
|
||||
while (running < 1):
|
||||
sleep(1)
|
||||
lf_r = LFRequest.LFRequest(base_url+"/wl/"+args['name']+"?fields=name,state,_links")
|
||||
try:
|
||||
json_response = lf_r.getAsJson()
|
||||
if (json_response is None):
|
||||
continue
|
||||
for key, value in json_response.items():
|
||||
if (isinstance(value, dict)):
|
||||
if ("_links" in value):
|
||||
if (value["name"] == args['name']):
|
||||
if (value["state"].startswith("Run")):
|
||||
LFUtils.debug_printer.pprint(json_response)
|
||||
running = 1
|
||||
|
||||
except urllib.error.HTTPError as error:
|
||||
print("Error code {}".format(error.code))
|
||||
continue
|
||||
|
||||
print("Wanlink is running")
|
||||
|
||||
# stop wanlink
|
||||
lf_r = LFRequest.LFRequest(base_url+"/cli-json/set_cx_state")
|
||||
lf_r.addPostData({
|
||||
'test_mgr': 'all',
|
||||
'cx_name': args['name'],
|
||||
'cx_state': 'STOPPED'
|
||||
})
|
||||
lf_r.jsonPost()
|
||||
running = 1
|
||||
while (running > 0):
|
||||
sleep(1)
|
||||
lf_r = LFRequest.LFRequest(base_url+"/wl/"+args['name']+"?fields=name,eid,state,_links")
|
||||
LFUtils.debug_printer.pprint(json_response)
|
||||
try:
|
||||
json_response = lf_r.getAsJson()
|
||||
if (json_response is None):
|
||||
continue
|
||||
for key, value in json_response.items():
|
||||
if (isinstance(value, dict)):
|
||||
if ("_links" in value):
|
||||
if (value["name"] == args['name']):
|
||||
if (value["state"].startswith("Stop")):
|
||||
LFUtils.debug_printer.pprint(json_response)
|
||||
running = 0
|
||||
|
||||
except urllib.error.HTTPError as error:
|
||||
print("Error code {}".format(error.code))
|
||||
continue
|
||||
|
||||
print("Wanlink is stopped.")
|
||||
|
||||
# print("Wanlink info:")
|
||||
# lf_r = LFRequest.LFRequest(base_url+"/wl/wl_eg1")
|
||||
# json_response = lf_r.getAsJson()
|
||||
# LFUtils.debug_printer.pprint(json_response)
|
||||
|
||||
# lf_r = LFRequest.LFRequest(base_url+"/wl_ep/wl_eg1-A")
|
||||
# json_response = lf_r.getAsJson()
|
||||
# LFUtils.debug_printer.pprint(json_response)
|
||||
|
||||
# lf_r = LFRequest.LFRequest(base_url+"/wl_ep/wl_eg1-B")
|
||||
# json_response = lf_r.getAsJson()
|
||||
# LFUtils.debug_printer.pprint(json_response)
|
||||
|
||||
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(base_url="http://localhost:8080")
|
||||
Reference in New Issue
Block a user