9856 update types

This commit is contained in:
Arthur 2024-02-23 15:56:21 -08:00
parent 69134dbb50
commit 6bb9d68f60
2 changed files with 18 additions and 3 deletions

View File

@ -1,4 +1,4 @@
from typing import List from typing import TYPE_CHECKING, Annotated, List, Union
import strawberry import strawberry
import strawberry_django import strawberry_django
@ -108,7 +108,13 @@ class FHRPGroupType(NetBoxObjectType):
) )
class FHRPGroupAssignmentType(BaseObjectType): class FHRPGroupAssignmentType(BaseObjectType):
# interface = graphene.Field('ipam.graphql.gfk_mixins.FHRPGroupInterfaceType') # interface = graphene.Field('ipam.graphql.gfk_mixins.FHRPGroupInterfaceType')
pass
@strawberry_django.field
def interface(self) -> Annotated[Union[
Annotated["InterfaceType", strawberry.lazy('dcim.graphql.types')],
Annotated["VMInterfaceType", strawberry.lazy('virtualization.graphql.types')],
], strawberry.union("FHRPGroupInterfaceType")]:
return self.interface
@strawberry_django.type( @strawberry_django.type(

View File

@ -1,3 +1,5 @@
from typing import Annotated, List
import strawberry import strawberry
import strawberry_django import strawberry_django
@ -62,7 +64,14 @@ class IKEProposalType(OrganizationalObjectType):
filters=IKEPolicyFilter filters=IKEPolicyFilter
) )
class IKEPolicyType(OrganizationalObjectType): class IKEPolicyType(OrganizationalObjectType):
pass
@strawberry_django.field
def proposals(self) -> List[Annotated["IKEProposalType", strawberry.lazy('vpn.graphql.types')]]:
return self.proposals.all()
@strawberry_django.field
def ipsec_profiles(self) -> List[Annotated["IPSecProposalType", strawberry.lazy('vpn.graphql.types')]]:
return self.ipsec_profiles.all()
@strawberry_django.type( @strawberry_django.type(