Updates for diffs

This commit is contained in:
Josh VanDeraa 2021-03-13 01:14:17 +00:00
parent 3b79b7e44c
commit 97cd37c452

View File

@ -613,7 +613,7 @@ class Prefix(ChangeLoggedModel, CustomFieldModel):
Returns:
UtilizationData (namedtuple): (numerator, denominator)
"""
if self.status == Prefix.STATUS_CONTAINER:
if self.status == PrefixStatusChoices.STATUS_CONTAINER:
queryset = Prefix.objects.filter(prefix__net_contained=str(self.prefix), vrf=self.vrf)
child_prefixes = netaddr.IPSet([p.prefix for p in queryset])
return UtilizationData(numerator=child_prefixes.size, denominator=self.prefix.size)