This commit is contained in:
Arthur 2023-08-25 13:52:24 -07:00
parent 987ab3e1aa
commit 7d2e246a8f

View File

@ -5,6 +5,7 @@ from dcim.graphql.mixins import CabledObjectMixin
from extras.graphql.mixins import CustomFieldsMixin, TagsMixin, ContactsMixin from extras.graphql.mixins import CustomFieldsMixin, TagsMixin, ContactsMixin
from netbox.graphql.types import ObjectType, OrganizationalObjectType, NetBoxObjectType from netbox.graphql.types import ObjectType, OrganizationalObjectType, NetBoxObjectType
from .filters import * from .filters import *
from typing import List
__all__ = ( __all__ = (
'CircuitTerminationType', 'CircuitTerminationType',
@ -16,13 +17,39 @@ __all__ = (
) )
@strawberry.django.type(
models.Provider,
fields='__all__',
filters=ProviderFilter
)
class ProviderType(NetBoxObjectType, ContactsMixin):
pass
@strawberry.django.type(
models.ProviderAccount,
fields='__all__',
filters=ProviderAccountFilter
)
class ProviderAccountType(NetBoxObjectType):
pass
@strawberry.django.type(
models.ProviderNetwork,
fields='__all__',
filters=ProviderNetworkFilter
)
class ProviderNetworkType(NetBoxObjectType):
pass
@strawberry.django.type( @strawberry.django.type(
models.CircuitTermination, models.CircuitTermination,
fields='__all__', fields='__all__',
filters=CircuitTerminationFilter filters=CircuitTerminationFilter
) )
class CircuitTerminationType: class CircuitTerminationType(CustomFieldsMixin, TagsMixin, CabledObjectMixin, ObjectType):
# class CircuitTerminationType(CustomFieldsMixin, TagsMixin, CabledObjectMixin, ObjectType):
pass pass
@ -32,8 +59,7 @@ class CircuitTerminationType:
filters=CircuitFilter filters=CircuitFilter
) )
class CircuitType(NetBoxObjectType, ContactsMixin): class CircuitType(NetBoxObjectType, ContactsMixin):
# class CircuitType(NetBoxObjectType, ContactsMixin): provider: ProviderType
pass
@strawberry.django.type( @strawberry.django.type(
@ -41,36 +67,5 @@ class CircuitType(NetBoxObjectType, ContactsMixin):
fields='__all__', fields='__all__',
filters=CircuitTypeFilter filters=CircuitTypeFilter
) )
class CircuitTypeType: class CircuitTypeType(OrganizationalObjectType):
# class CircuitTypeType(OrganizationalObjectType):
pass
@strawberry.django.type(
models.Provider,
fields='__all__',
filters=ProviderFilter
)
class ProviderType:
# class ProviderType(NetBoxObjectType, ContactsMixin):
pass
@strawberry.django.type(
models.ProviderAccount,
fields='__all__',
filters=ProviderAccountFilter
)
class ProviderAccountType:
# class ProviderAccountType(NetBoxObjectType):
pass
@strawberry.django.type(
models.ProviderNetwork,
fields='__all__',
filters=ProviderNetworkFilter
)
class ProviderNetworkType:
# class ProviderNetworkType(NetBoxObjectType):
pass pass