Add GraphQL for DCIM

2024-05-10 07:54:54 +00:00 · 2021-06-25 16:17:36 -04:00
parent 605b7c5b3e
commit 14e27363e0
6 changed files with 441 additions and 29 deletions
--- a/netbox/dcim/graphql/schema.py
+++ b/netbox/dcim/graphql/schema.py
@@ -0,0 +1,106 @@
+import graphene
+
+from netbox.graphql.fields import ObjectField, ObjectListField
+from .types import *
+
+
+class DCIMQuery(graphene.ObjectType):
+    cable = ObjectField(CableType)
+    cables = ObjectListField(CableType)
+
+    console_port = ObjectField(ConsolePortType)
+    console_ports = ObjectListField(ConsolePortType)
+
+    console_port_template = ObjectField(ConsolePortTemplateType)
+    console_port_templates = ObjectListField(ConsolePortTemplateType)
+
+    console_server_port = ObjectField(ConsoleServerPortType)
+    console_server_ports = ObjectListField(ConsoleServerPortType)
+
+    console_server_port_template = ObjectField(ConsoleServerPortTemplateType)
+    console_server_port_templates = ObjectListField(ConsoleServerPortTemplateType)
+
+    device = ObjectField(DeviceType)
+    devices = ObjectListField(DeviceType)
+
+    device_bay = ObjectField(DeviceBayType)
+    device_bays = ObjectListField(DeviceBayType)
+
+    device_bay_template = ObjectField(DeviceBayTemplateType)
+    device_bay_templates = ObjectListField(DeviceBayTemplateType)
+
+    device_role = ObjectField(DeviceRoleType)
+    device_roles = ObjectListField(DeviceRoleType)
+
+    device_type = ObjectField(DeviceTypeType)
+    device_types = ObjectListField(DeviceTypeType)
+
+    front_port = ObjectField(FrontPortType)
+    front_ports = ObjectListField(FrontPortType)
+
+    front_port_template = ObjectField(FrontPortTemplateType)
+    front_port_templates = ObjectListField(FrontPortTemplateType)
+
+    interface = ObjectField(InterfaceType)
+    interfaces = ObjectListField(InterfaceType)
+
+    interface_template = ObjectField(InterfaceTemplateType)
+    interface_templates = ObjectListField(InterfaceTemplateType)
+
+    inventory_item = ObjectField(InventoryItemType)
+    inventory_items = ObjectListField(InventoryItemType)
+
+    location = ObjectField(LocationType)
+    locations = ObjectListField(LocationType)
+
+    manufacturer = ObjectField(ManufacturerType)
+    manufacturers = ObjectListField(ManufacturerType)
+
+    platform = ObjectField(PlatformType)
+    platforms = ObjectListField(PlatformType)
+
+    power_feed = ObjectField(PowerFeedType)
+    power_feeds = ObjectListField(PowerFeedType)
+
+    power_outlet = ObjectField(PowerOutletType)
+    power_outlets = ObjectListField(PowerOutletType)
+
+    power_outlet_template = ObjectField(PowerOutletTemplateType)
+    power_outlet_templates = ObjectListField(PowerOutletTemplateType)
+
+    power_panel = ObjectField(PowerPanelType)
+    power_panels = ObjectListField(PowerPanelType)
+
+    power_port = ObjectField(PowerPortType)
+    power_ports = ObjectListField(PowerPortType)
+
+    power_port_template = ObjectField(PowerPortTemplateType)
+    power_port_templates = ObjectListField(PowerPortTemplateType)
+
+    rack = ObjectField(RackType)
+    racks = ObjectListField(RackType)
+
+    rack_reservation = ObjectField(RackReservationType)
+    rack_reservations = ObjectListField(RackReservationType)
+
+    rack_role = ObjectField(RackRoleType)
+    rack_roles = ObjectListField(RackRoleType)
+
+    rear_port = ObjectField(RearPortType)
+    rear_ports = ObjectListField(RearPortType)
+
+    rear_port_template = ObjectField(RearPortTemplateType)
+    rear_port_templates = ObjectListField(RearPortTemplateType)
+
+    region = ObjectField(RegionType)
+    regions = ObjectListField(RegionType)
+
+    site = ObjectField(SiteType)
+    sites = ObjectListField(SiteType)
+
+    site_group = ObjectField(SiteGroupType)
+    site_groups = ObjectListField(SiteGroupType)
+
+    virtual_chassis = ObjectField(VirtualChassisType)
+    # TODO: Rectify list field name
+    virtual_chassis_list = ObjectListField(VirtualChassisType)