Skip to content

Commit 2c7c3eb

Browse files
Noah418afoster
authored andcommitted
cleaned up some more
1 parent 9a70da1 commit 2c7c3eb

3 files changed

Lines changed: 19 additions & 15 deletions

File tree

module/netbox/object_classes.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -659,7 +659,6 @@ def update(self, data=None, read_from_netbox=False, source=None):
659659
max_len=self.data_model.get("slug"))
660660

661661
# update all data items
662-
log.debug2(f"Updating {self.name} '{display_name}' with data: {parsed_data.items()}")
663662
data_updated = False
664663
for key, new_value in parsed_data.items():
665664

@@ -751,8 +750,7 @@ def update(self, data=None, read_from_netbox=False, source=None):
751750
new_value_str = new_value_str.replace("\n", " ")
752751
log.info(f"{self.name.capitalize()} '{display_name}' attribute '{key}' changed from "
753752
f"'{current_value_str}' to '{new_value_str}'")
754-
log.debug2(f"Updating {self.name} '{display_name}' attribute '{key}' from "
755-
f"'{current_value_str}' to '{new_value_str}'")
753+
756754
self.data[key] = new_value
757755
self.updated_items.append(key)
758756
data_updated = True
@@ -1873,6 +1871,7 @@ class NBCluster(NetBoxObject):
18731871

18741872
def __init__(self, *args, **kwargs):
18751873
self.mapping = NetBoxMappings()
1874+
# scope types allowed for clusters
18761875
self.scopes = [
18771876
NBSite, NBSiteGroup, NBLocation, NBRegion
18781877
]
@@ -1883,7 +1882,9 @@ def __init__(self, *args, **kwargs):
18831882
"tenant": NBTenant,
18841883
"group": NBClusterGroup,
18851884
"scope_type": self.mapping.scopes_object_types(self.scopes),
1885+
# supports scoped clusters
18861886
"scope_id": NetBoxObject,
1887+
# supports pre4.2.0 clusters with site
18871888
"site": NBSite,
18881889
"tags": NBTagList
18891890
}

module/sources/vmware/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def __init__(self):
163163
key: defines a cluster name as regex
164164
value: defines the NetBox scope id (use quotes if name contains commas)
165165
""",
166-
config_example="Cluster_NYC = New York, Cluster_FFM.* = Data Centers, Cluster_BER = Building 1"),
166+
config_example="Cluster_NYC = 1, Cluster_FFM.* = 2, Cluster_BER = 7"),
167167
ConfigOption("cluster_tenant_relation",
168168
str,
169169
description="""\

module/sources/vmware/connection.py

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ def get_site_name(self, object_type, object_name, cluster_name=""):
469469
if object_type not in [NBCluster, NBDevice]:
470470
raise ValueError(f"Object must be a '{NBCluster.name}' or '{NBDevice.name}'.")
471471

472-
log.debug(f"Trying to find site name for {object_type.name} '{object_name}'")
472+
log.debug2(f"Trying to find site name for {object_type.name} '{object_name}'")
473473

474474
# check if site was provided in config
475475
relation_name = "host_site_relation" if object_type == NBDevice else "cluster_site_relation"
@@ -525,24 +525,28 @@ def get_scope_type(self, object_type, object_name):
525525
if object_type != NBCluster:
526526
raise ValueError(f"Object type must be '{NBCluster.name}'.")
527527

528+
# get scope type from relation config
528529
relation_name = "cluster_scope_type_relation"
529530
scope_type = self.get_object_relation(object_name, relation_name)
530531
log.debug(f"Retrieved scope type '{scope_type}' for {object_type.name} '{object_name}' from relation '{relation_name}'.")
531532

533+
# if the scope_type is a list, use the first element
532534
if scope_type is not None and type(scope_type) is list:
533535
scope_type_list = scope_type
534536
scope_type = scope_type_list[0] if len(scope_type_list) > 0 else None
535537
log.debug(f"Scope type for {object_type.name} '{object_name}' is a list, using first element: '{scope_type}'")
536538

539+
# if scope_type is not a str, return None
537540
if type(scope_type) is not str:
538541
log.debug(f"scope_type is type: {type(scope_type)}, not str")
539542
return None
540543

544+
# set scope_type to None if it is configured as "<NONE>"
541545
if scope_type == "<NONE>":
542546
log.debug(f"Scope type for {object_type.name} '{object_name}' is set to None")
543547
return None
544548

545-
log.debug(f"Returning scope type '{scope_type}' for {object_type.name} '{object_name}'. End of method.")
549+
log.debug2(f"Returning scope type '{scope_type}' for {object_type.name} '{object_name}'.")
546550
return scope_type
547551

548552
def get_scope_id(self, object_type, object_name):
@@ -567,21 +571,19 @@ def get_scope_id(self, object_type, object_name):
567571
if object_type != NBCluster:
568572
raise ValueError(f"Object type must be '{NBCluster.name}'.")
569573

570-
574+
# get scope id from relation config
571575
relation_name = "cluster_scope_id_relation"
572-
573576
scope_id = self.get_object_relation(object_name, relation_name)
574577

578+
# return None if scope_id is None or not a string
575579
if scope_id is None:
576-
scope_id = object_name
577-
if scope_id is None:
578-
log.debug(f"No scope id found for {object_name}.")
579-
return None
580+
log.debug(f"No scope id found for {object_name}.")
581+
return None
580582
if type(scope_id) is not str:
581583
log.debug(f"scope_id is type: {type(scope_id)}, not str")
582584
return None
583585

584-
log.debug(f"Retrieved scope id '{scope_id}' for {object_type.name} '{object_name}' from relation '{relation_name}'. End of method.")
586+
log.debug2(f"Retrieved scope id '{scope_id}' for {object_type.name} '{object_name}' from relation '{relation_name}'. End of method.")
585587

586588
return scope_id
587589

@@ -1472,8 +1474,9 @@ def add_cluster(self, obj):
14721474
self.settings.cluster_include_filter,
14731475
self.settings.cluster_exclude_filter) is False:
14741476
return
1475-
log.debug(f"Cluster '{name}' passes include and exclude filters. Continuing.")
1477+
log.debug2(f"Cluster '{name}' passes include and exclude filters. Continuing.")
14761478

1479+
# get scope type and id, or site name
14771480
scope_type = self.get_scope_type(NBCluster, full_cluster_name)
14781481
if scope_type is None:
14791482
scope_type = self.get_scope_type(NBCluster, name)
@@ -1504,7 +1507,7 @@ def add_cluster(self, obj):
15041507
else:
15051508
log.debug(f"Cluster '{full_cluster_name}' has no scope type or scope id.")
15061509
else:
1507-
# set site_name in the pre-4.2.0 NetBox versions is one is found
1510+
# set site_name in the pre-4.2.0 NetBox versions if one is found
15081511
if site_name is not None:
15091512
data["site"] = {"name": site_name}
15101513

0 commit comments

Comments
 (0)