@@ -198,6 +198,7 @@ def createShaper():
198
198
hardware_name = {}
199
199
access_device_name = {}
200
200
hardware_parent = {}
201
+ hardware_type = {}
201
202
for monitored_device in monitoring :
202
203
hardware_name [monitored_device ['id' ]] = monitored_device ['title' ]
203
204
if 'access_device' in monitored_device :
@@ -214,7 +215,12 @@ def createShaper():
214
215
if hardware_parent [monitored_device ['id' ]] in hardware_name :
215
216
hardware_name_extended [monitored_device ['id' ]] = hardware_name [hardware_parent [monitored_device ['id' ]]] + "_" + monitored_device ['title' ]
216
217
if monitored_device ['id' ] not in hardware_name_extended :
217
- hardware_name_extended [monitored_device ['id' ]] = monitored_device ['title' ]
218
+ hardware_name_extended [monitored_device ['id' ]] = monitored_device ['title' ]
219
+ if 'type' in monitored_device :
220
+ if monitored_device ['type' ] == 5 :
221
+ hardware_type [monitored_device ['id' ]] = 'AP'
222
+ else :
223
+ hardware_type [monitored_device ['id' ]] = 'Site'
218
224
for device_num in hardware_name :
219
225
# Find parent name of hardware
220
226
parent_name = ''
@@ -228,8 +234,13 @@ def createShaper():
228
234
if nodeName in siteBandwidth :
229
235
download = siteBandwidth [nodeName ]["download" ]
230
236
upload = siteBandwidth [nodeName ]["upload" ]
231
- node = NetworkNode (id = device_num , displayName = nodeName , type = NodeType .site ,
232
- parentId = parent_id , download = download , upload = upload , address = None )
237
+ nodeType = hardware_type [device_num ]
238
+ if nodeType == 'AP' :
239
+ node = NetworkNode (id = device_num , displayName = nodeName , type = NodeType .ap ,
240
+ parentId = parent_id , download = download , upload = upload , address = None )
241
+ else :
242
+ node = NetworkNode (id = device_num , displayName = nodeName , type = NodeType .site ,
243
+ parentId = parent_id , download = download , upload = upload , address = None )
233
244
net .addRawNode (node )
234
245
cust_id_to_name = {}
235
246
for customer in customers :
0 commit comments