NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-23 04:22:01 -06:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add image_attachments to Device, Location, Rack, Site

Browse Source
This commit is contained in:
jeremystretch 2021-08-03 11:38:18 -04:00
parent 0df67dbc12
commit 8ad958708f
2 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
netbox/dcim/graphql/types.py
View File

@ -1,4 +1,5 @@
from dcim import filtersets, models from dcim import filtersets, models
from extras.graphql.mixins import ImageAttachmentsMixin
from ipam.graphql.mixins import IPAddressesMixin from ipam.graphql.mixins import IPAddressesMixin
from netbox.graphql.types import BaseObjectType, ObjectType, TaggedObjectType from netbox.graphql.types import BaseObjectType, ObjectType, TaggedObjectType
@ -97,7 +98,7 @@ class ConsoleServerPortTemplateType(BaseObjectType):
return self.type or None return self.type or None
class DeviceType(TaggedObjectType): class DeviceType(ImageAttachmentsMixin, TaggedObjectType):
class Meta: class Meta:
model = models.Device model = models.Device
@ -186,7 +187,7 @@ class InventoryItemType(TaggedObjectType):
filterset_class = filtersets.InventoryItemFilterSet filterset_class = filtersets.InventoryItemFilterSet
class LocationType(ObjectType): class LocationType(ImageAttachmentsMixin, ObjectType):
class Meta: class Meta:
model = models.Location model = models.Location
@ -276,7 +277,7 @@ class PowerPortTemplateType(BaseObjectType):
return self.type or None return self.type or None
class RackType(TaggedObjectType): class RackType(ImageAttachmentsMixin, TaggedObjectType):
class Meta: class Meta:
model = models.Rack model = models.Rack
@ -330,7 +331,7 @@ class RegionType(ObjectType):
filterset_class = filtersets.RegionFilterSet filterset_class = filtersets.RegionFilterSet
class SiteType(TaggedObjectType): class SiteType(ImageAttachmentsMixin, TaggedObjectType):
class Meta: class Meta:
model = models.Site model = models.Site

12
netbox/extras/graphql/mixins.py Normal file
View File

@ -0,0 +1,12 @@
import graphene
__all__ = (
'ImageAttachmentsMixin',
)
class ImageAttachmentsMixin:
image_attachments = graphene.List('extras.graphql.types.ImageAttachmentType')
def resolve_image_attachments(self, info):
return self.images.restrict(info.context.user, 'view')