diff --git a/netbox/ipam/graphql/filters.py b/netbox/ipam/graphql/filters.py index 2d366379a..1b0e0133b 100644 --- a/netbox/ipam/graphql/filters.py +++ b/netbox/ipam/graphql/filters.py @@ -20,6 +20,7 @@ __all__ = ( 'VLANFilter', 'VLANGroupFilter', 'VLANTranslationPolicyFilter', + 'VLANTranslationRuleFilter', 'VRFFilter', ) @@ -120,6 +121,12 @@ class VLANTranslationPolicyFilter(BaseFilterMixin): pass +@strawberry_django.filter(models.VLANTranslationRule, lookups=True) +@autotype_decorator(filtersets.VLANTranslationRuleFilterSet) +class VLANTranslationRuleFilter(BaseFilterMixin): + pass + + @strawberry_django.filter(models.VRF, lookups=True) @autotype_decorator(filtersets.VRFFilterSet) class VRFFilter(BaseFilterMixin): diff --git a/netbox/ipam/graphql/schema.py b/netbox/ipam/graphql/schema.py index d991d68f7..5fcf78ea9 100644 --- a/netbox/ipam/graphql/schema.py +++ b/netbox/ipam/graphql/schema.py @@ -56,5 +56,8 @@ class IPAMQuery: vlan_translation_policy: VLANTranslationPolicyType = strawberry_django.field() vlan_translation_policy_list: List[VLANTranslationPolicyType] = strawberry_django.field() + vlan_translation_rule: VLANTranslationRuleType = strawberry_django.field() + vlan_translation_rule_list: List[VLANTranslationRuleType] = strawberry_django.field() + vrf: VRFType = strawberry_django.field() vrf_list: List[VRFType] = strawberry_django.field() diff --git a/netbox/ipam/graphql/types.py b/netbox/ipam/graphql/types.py index 055cc8a2e..98fa9f973 100644 --- a/netbox/ipam/graphql/types.py +++ b/netbox/ipam/graphql/types.py @@ -28,6 +28,7 @@ __all__ = ( 'VLANType', 'VLANGroupType', 'VLANTranslationPolicyType', + 'VLANTranslationRuleType', 'VRFType', ) @@ -277,13 +278,22 @@ class VLANGroupType(OrganizationalObjectType): @strawberry_django.type( models.VLANTranslationPolicy, - exclude=('scope_type', 'scope_id'), + fields='__all__', filters=VLANTranslationPolicyFilter ) class VLANTranslationPolicyType(NetBoxObjectType): pass +@strawberry_django.type( + models.VLANTranslationRule, + fields='__all__', + filters=VLANTranslationRuleFilter +) +class VLANTranslationRuleType(NetBoxObjectType): + pass + + @strawberry_django.type( models.VRF, fields='__all__',