mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-01 03:07:56 +00:00
89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
#!env /usr/bin/python
|
|
|
|
# Extend this class to use common set of debug and request features for your script
|
|
import pprint
|
|
from pprint import pprint
|
|
import LANforge.LFUtils
|
|
from LANforge.LFUtils import *
|
|
import traceback
|
|
from traceback import extract_stack
|
|
|
|
class LFCliBase:
|
|
# do not use `super(LFCLiBase,self).__init__(self, host, port, _debugOn)
|
|
# that is py2 era syntax and will force self into the host variable, making you
|
|
# very confused.
|
|
def __init__(self, _lfjson_host, _lfjson_port, _debug=False, _halt_on_error=False):
|
|
self.lfjson_host = _lfjson_host
|
|
self.lfjson_port = _lfjson_port
|
|
self.debugOn = _debug
|
|
self.haltOnError = _halt_on_error
|
|
self.mgr_url = "http://%s:%s/" % (self.lfjson_host, self.lfjson_port)
|
|
|
|
def json_post(self, _req_url, _data):
|
|
json_response = None
|
|
if self.mgr_url.endswith('/') and _req_url.startswith('/'):
|
|
_req_url = _req_url[1:]
|
|
try:
|
|
lf_r = LFRequest.LFRequest(self.mgr_url + _req_url)
|
|
_data['suppress_preexec_cli'] = True
|
|
_data['suppress_preexec_method'] = True
|
|
lf_r.addPostData(_data)
|
|
if (self.debugOn):
|
|
LANforge.LFUtils.debug_printer.pprint(_data)
|
|
json_response = lf_r.jsonPost(self.debugOn)
|
|
except Exception as x:
|
|
if self.debugOn or self.haltOnError:
|
|
print("jsonPost posted to %s" % _req_url)
|
|
pprint(_data)
|
|
print("Exception %s:" % x)
|
|
traceback.print_exception(Exception, x, x.__traceback__, chain=True)
|
|
if self.haltOnError:
|
|
exit(1)
|
|
|
|
return json_response
|
|
|
|
def json_get(self, _req_url):
|
|
json_response = None
|
|
if self.mgr_url.endswith('/') and _req_url.startswith('/'):
|
|
_req_url = _req_url[1:]
|
|
try:
|
|
lf_r = LFRequest.LFRequest(self.mgr_url + _req_url)
|
|
json_response = lf_r.getAsJson(self.debugOn)
|
|
if (json_response is None) and self.debugOn:
|
|
raise ValueError(json_response)
|
|
except ValueError as ve:
|
|
if self.debugOn or self.haltOnError:
|
|
print("jsonGet asked for "+_req_url)
|
|
print("Exception %s:" % ve)
|
|
traceback.print_exception(ValueError, ve, ve.__traceback__, chain=True)
|
|
if self.haltOnError:
|
|
sys.exit(1)
|
|
|
|
return json_response
|
|
|
|
def error(self, exception):
|
|
#print("lfcli_base error: %s" % exception)
|
|
pprint.pprint(exception)
|
|
traceback.print_exception(Exception, exception, exception.__traceback__, chain=True)
|
|
if self.haltOnError:
|
|
print("halting on error")
|
|
sys.exit(1)
|
|
#else:
|
|
# print("continuing...")
|
|
|
|
def check_connect(self):
|
|
print("Checking for LANforge GUI connection: %s" % self.mgr_url)
|
|
response = self.json_get("/")
|
|
duration = 0
|
|
while (response is None) and (duration < 300):
|
|
print("LANforge GUI connection not found sleeping 5 seconds, tried: %s" % self.mgr_url)
|
|
duration += 2
|
|
time.sleep(2)
|
|
response = self.json_get("")
|
|
|
|
if duration >= 300:
|
|
print("Could not connect to LANforge GUI")
|
|
sys.exit(1)
|
|
|
|
# ~class
|