Add files to get UISP data and process nn

supportbot/supportbot/utils/uisp_data.py

@@ -0,0 +1,36 @@
+import requests
+import json
+from dotenv import load_dotenv
+import os
+import re
+
+load_dotenv()
+
+def get_uisp_devices():
+ response = requests.get("https://uisp.mesh.nycmesh.net/nms/api/v2.1/devices", headers={'x-auth-token': os.environ.get('NYCMESH_TOOL_AUTH_TOKEN')}, verify=False)
+
+ devices = json.loads(response.content)
+
+ if devices == []:
+ raise ValueError('Problem downloading UISP devices.')
+
+ return devices
+
+def nn_from_uisp_name(uisp_name):
+ matches = re.findall("(?:^|-)(\d{3,})(?:-|$)", uisp_name)
+ if not matches:
+ return None
+
+ return int(matches[0])
+
+def filter_devices_by_nn(devices, nn):
+ filtered_devices = []
+ for device in devices:
+ try:
+ name = device['identification']['displayName']
+ nn = nn_from_uisp_name(name)
+ if nn:
+ filtered_devices.append(device)
+ except:
+ pass
+ return filtered_devices

tests/test_uisp_data.py

@@ -0,0 +1,16 @@
+import os
+from dotenv import load_dotenv
+from supportbot.utils.uisp_data import nn_from_uisp_name, get_uisp_devices
+
+load_dotenv()
+
+def test_get_uisp_devices():
+ assert len(get_uisp_devices()) > 0
+
+def test_nn_from_uisp_name():
+ assert nn_from_uisp_name("lbe-5134-diy-455") == 5134
+ assert nn_from_uisp_name("lbe-5134") == 5134
+ assert nn_from_uisp_name("5134-lbe") == 5134
+
+if __name__ == "__main__":
+ test_nn_from_uisp_name()