Skip to content

Commit 5967548

Browse files
authored
Adding pynetbox check and wireless endpoints for NetBox v3.1 (#673)
* adding pynetbox check and wireless endpoints.
1 parent 7b2bdb4 commit 5967548

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

plugins/lookup/nb_lookup.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,27 @@ def get_endpoint(netbox, term):
256256
"webhooks": {"endpoint": netbox.extras.webhooks},
257257
}
258258

259+
major, minor, patch = map(int, pynetbox.__version__.split("."))
260+
261+
if major >= 6 and minor >= 4 and patch >= 0:
262+
netbox_endpoint_map["wireless-lan-groups"] = {
263+
"endpoint": netbox.wireless.wireless_lan_groups
264+
}
265+
netbox_endpoint_map["wireless-lan-groups"] = {
266+
"endpoint": netbox.wireless.wireless_lan_groups
267+
}
268+
netbox_endpoint_map["wireless-lan"] = {"endpoint": netbox.wireless.wireless_lan}
269+
netbox_endpoint_map["wireless-links"] = {
270+
"endpoint": netbox.wireless.wireless_links
271+
}
272+
273+
else:
274+
if "wireless" in term:
275+
Display().v(
276+
"pynetbox version %d.%d.%d does not support wireless app; please update to v6.4.0 or newer."
277+
% (major, minor, patch)
278+
)
279+
259280
return netbox_endpoint_map[term]["endpoint"]
260281

261282

0 commit comments

Comments
 (0)