diff --git a/hostapps/update.py b/hostapps/update.py index 5c52a92..667515c 100644 --- a/hostapps/update.py +++ b/hostapps/update.py @@ -56,9 +56,10 @@ def main(): schneider = nb.dcim.manufacturers.get(name="Schneider") power_protection = nb.dcim.device_roles.get(name="Power Protection") + power_distribution = nb.dcim.device_roles.get(name="Power Distribution") for device in nb.dcim.devices.filter( manufacturer_id=schneider.id, - role_id=power_protection.id, + role_id=[power_protection.id, power_distribution.id], status="active", has_primary_ip="true", ):