mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-20 20:32:25 -06:00
30 lines
647 B
Python
30 lines
647 B
Python
import graphene
|
|
from graphene.types.generic import GenericScalar
|
|
|
|
__all__ = (
|
|
'CustomFieldsMixin',
|
|
'ImageAttachmentsMixin',
|
|
'TagsMixin',
|
|
)
|
|
|
|
|
|
class CustomFieldsMixin:
|
|
custom_fields = GenericScalar()
|
|
|
|
def resolve_custom_fields(self, info):
|
|
return self.custom_field_data
|
|
|
|
|
|
class ImageAttachmentsMixin:
|
|
image_attachments = graphene.List('extras.graphql.types.ImageAttachmentType')
|
|
|
|
def resolve_image_attachments(self, info):
|
|
return self.images.restrict(info.context.user, 'view')
|
|
|
|
|
|
class TagsMixin:
|
|
tags = graphene.List(graphene.String)
|
|
|
|
def resolve_tags(self, info):
|
|
return self.tags.all()
|