NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 17:26:10 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Closes #11851: Add family field to IPAddress queries in GraphQL

Browse Source
This commit is contained in:
rmanyari 2023-03-01 10:20:32 -07:00
parent c9e5a4c996
commit c3565f841a
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
netbox/ipam/graphql/types.py
View File

@ -64,14 +64,30 @@ class FHRPGroupAssignmentType(BaseObjectType):
filterset_class = filtersets.FHRPGroupAssignmentFilterSet filterset_class = filtersets.FHRPGroupAssignmentFilterSet
class IPAddressFamilyType(graphene.ObjectType):
value = graphene.Int()
label = graphene.String()
def __init__(self, value):
self.value = value
self.label = 'IPv4' if value == 4 else 'IPv6'
class IPAddressType(NetBoxObjectType): class IPAddressType(NetBoxObjectType):
assigned_object = graphene.Field('ipam.graphql.gfk_mixins.IPAddressAssignmentType') assigned_object = graphene.Field('ipam.graphql.gfk_mixins.IPAddressAssignmentType')
family = graphene.Field(IPAddressFamilyType)
class Meta: class Meta:
model = models.IPAddress model = models.IPAddress
exclude = ('assigned_object_type', 'assigned_object_id') exclude = ('assigned_object_type', 'assigned_object_id')
filterset_class = filtersets.IPAddressFilterSet filterset_class = filtersets.IPAddressFilterSet
def resolve_family(self, _):
# Note that self, is an instance of models.IPAddress
# thus resolves to the address family value.
return IPAddressFamilyType(self.family)
def resolve_role(self, info): def resolve_role(self, info):
return self.role or None return self.role or None