mirror of
https://github.com/LibreQoE/LibreQoS.git
synced 2024-11-25 01:30:17 -06:00
127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
import requests
|
|
from ispConfig import excludeSites, findIPv6usingMikrotik, bandwidthOverheadFactor, exceptionCPEs, splynx_api_key, splynx_api_secret, splynx_api_url
|
|
from integrationCommon import isIpv4Permitted
|
|
import base64
|
|
from requests.auth import HTTPBasicAuth
|
|
if findIPv6usingMikrotik == True:
|
|
from mikrotikFindIPv6 import pullMikrotikIPv6
|
|
from integrationCommon import NetworkGraph, NetworkNode, NodeType
|
|
|
|
def buildHeaders():
|
|
credentials = splynx_api_key + ':' + splynx_api_secret
|
|
credentials = base64.b64encode(credentials.encode()).decode()
|
|
return {'Authorization' : "Basic %s" % credentials}
|
|
|
|
def spylnxRequest(target, headers):
|
|
# Sends a REST GET request to Spylnx and returns the
|
|
# result in JSON
|
|
url = splynx_api_url + "/api/2.0/" + target
|
|
r = requests.get(url, headers=headers, timeout=10)
|
|
return r.json()
|
|
|
|
def getTariffs(headers):
|
|
data = spylnxRequest("admin/tariffs/internet", headers)
|
|
tariff = []
|
|
downloadForTariffID = {}
|
|
uploadForTariffID = {}
|
|
for tariff in data:
|
|
tariffID = tariff['id']
|
|
speed_download = round((int(tariff['speed_download']) / 1000))
|
|
speed_upload = round((int(tariff['speed_upload']) / 1000))
|
|
downloadForTariffID[tariffID] = speed_download
|
|
uploadForTariffID[tariffID] = speed_upload
|
|
return (tariff, downloadForTariffID, uploadForTariffID)
|
|
|
|
def getCustomers(headers):
|
|
data = spylnxRequest("admin/customers/customer", headers)
|
|
#addressForCustomerID = {}
|
|
#customerIDs = []
|
|
#for customer in data:
|
|
# customerIDs.append(customer['id'])
|
|
# addressForCustomerID[customer['id']] = customer['street_1']
|
|
return data
|
|
|
|
def getRouters(headers):
|
|
data = spylnxRequest("admin/networking/routers", headers)
|
|
ipForRouter = {}
|
|
for router in data:
|
|
routerID = router['id']
|
|
ipForRouter[routerID] = router['ip']
|
|
return ipForRouter
|
|
|
|
def combineAddress(json):
|
|
# Combines address fields into a single string
|
|
# The API docs seem to indicate that there isn't a "state" field?
|
|
if json["street_1"]=="" and json["city"]=="" and json["zip_code"]=="":
|
|
return json["id"] + "/" + json["name"]
|
|
else:
|
|
return json["street_1"] + " " + json["city"] + " " + json["zip_code"]
|
|
|
|
def createShaper():
|
|
net = NetworkGraph()
|
|
|
|
print("Fetching data from Spylnx")
|
|
headers = buildHeaders()
|
|
tariff, downloadForTariffID, uploadForTariffID = getTariffs(headers)
|
|
customers = getCustomers(headers)
|
|
ipForRouter = getRouters(headers)
|
|
|
|
# It's not very clear how a service is meant to handle multiple
|
|
# devices on a shared tariff. Creating each service as a combined
|
|
# entity including the customer, to be on the safe side.
|
|
for customerJson in customers:
|
|
if customerJson['status'] == 'active':
|
|
services = spylnxRequest("admin/customers/customer/" + customerJson["id"] + "/internet-services", headers)
|
|
for serviceJson in services:
|
|
combinedId = "c_" + str(customerJson["id"]) + "_s_" + str(serviceJson["id"])
|
|
tariff_id = serviceJson['tariff_id']
|
|
customer = NetworkNode(
|
|
type=NodeType.client,
|
|
id=combinedId,
|
|
displayName=customerJson["name"],
|
|
address=combineAddress(customerJson),
|
|
download=downloadForTariffID[tariff_id],
|
|
upload=uploadForTariffID[tariff_id],
|
|
)
|
|
net.addRawNode(customer)
|
|
|
|
ipv4 = ''
|
|
ipv6 = ''
|
|
routerID = serviceJson['router_id']
|
|
# If not "Taking IPv4" (Router will assign IP), then use router's set IP
|
|
if serviceJson['taking_ipv4'] == 0:
|
|
ipv4 = ipForRouter[routerID]
|
|
elif serviceJson['taking_ipv4'] == 1:
|
|
ipv4 = serviceJson['ipv4']
|
|
# If not "Taking IPv6" (Router will assign IP), then use router's set IP
|
|
if serviceJson['taking_ipv6'] == 0:
|
|
ipv6 = ''
|
|
elif serviceJson['taking_ipv6'] == 1:
|
|
ipv6 = serviceJson['ipv6']
|
|
|
|
device = NetworkNode(
|
|
id=combinedId+"_d" + str(serviceJson["id"]),
|
|
displayName=serviceJson["description"],
|
|
type=NodeType.device,
|
|
parentId=combinedId,
|
|
mac=serviceJson["mac"],
|
|
ipv4=[ipv4],
|
|
ipv6=[ipv6]
|
|
)
|
|
net.addRawNode(device)
|
|
|
|
net.prepareTree()
|
|
net.plotNetworkGraph(False)
|
|
if net.doesNetworkJsonExist():
|
|
print("network.json already exists. Leaving in-place.")
|
|
else:
|
|
net.createNetworkJson()
|
|
net.createShapedDevices()
|
|
|
|
def importFromSplynx():
|
|
#createNetworkJSON()
|
|
createShaper()
|
|
|
|
if __name__ == '__main__':
|
|
importFromSplynx()
|