| """ |
| SPDX-FileCopyrightText: 2020-present Open Networking Foundation <info@opennetworking.org> |
| SPDX-License-Identifier: LicenseRef-ONF-Member-1.01 |
| """ |
| |
| import requests |
| import json |
| from requests.structures import CaseInsensitiveDict |
| |
| |
| URL = "https://roc.menlo.aetherproject.org/aether-roc-api/aether/v4.0.0/connectivity-service-v4/" |
| #URL = "https://roc.staging.aether.onlab.us/aether-roc-api/aether/v4.0.0/connectivity-service-v4/" |
| |
| |
| def headers(key): |
| h = CaseInsensitiveDict() |
| h["Content-Type"] = "application/json" |
| h["Authorization"] = "Bearer " + key |
| return h |
| |
| def get_mbr(key, device_group): |
| url = URL + "vcs/vcs/vcs-{}/slice/mbr".format(device_group) |
| response = requests.get(url, headers=headers(key)) |
| if response.status_code != 200: |
| print("Failed to get mbr, status_code: {}".format(response.status_code)) |
| return None |
| mbr = json.loads(response.text) |
| return mbr |
| |
| def set_mbr(key, device_group, mbr): |
| print("Set {} mbr to {}".format(device_group, mbr)) |
| m = {'uplink' : mbr} |
| url = URL + "vcs/vcs/vcs-{}/slice/mbr".format(device_group) |
| response = requests.post(url, headers=headers(key), json=m) |
| assert response.status_code == 201, "Failed to set mbr" |
| if response.status_code != 201: |
| print("Failed to set mbr, device_group:{}, mbr:{}, status_code: {}".format(device_group, mbr, response.status_code)) |
| |
| if __name__ == '__main__': |
| key = "" |
| cameras = "menlo-4g-cameras" |
| #cameras = "cameras-4g" |
| |
| mbr = get_mbr(key, cameras) |
| print("uplink mbr:{}, downlink mbr: {}".format(mbr["uplink"], mbr["downlink"])) |
| set_mbr(key, cameras, 5000000) |
| mbr = get_mbr(key, cameras) |
| print("uplink mbr:{}, downlink mbr: {}".format(mbr["uplink"], mbr["downlink"])) |
| set_mbr(key, cameras, 10000000) |
| mbr = get_mbr(key, cameras) |
| print("uplink mbr:{}, downlink mbr: {}".format(mbr["uplink"], mbr["downlink"])) |