diff --git a/pynetbox/models/dcim.py b/pynetbox/models/dcim.py index 02533c0..a0bb397 100644 --- a/pynetbox/models/dcim.py +++ b/pynetbox/models/dcim.py @@ -19,16 +19,22 @@ from pynetbox.core.endpoint import DetailEndpoint, RODetailEndpoint from pynetbox.core.query import Request from pynetbox.core.response import JsonField, Record -from pynetbox.models.circuits import Circuits +from pynetbox.models.circuits import Circuits, CircuitTerminations from pynetbox.models.ipam import IpAddresses class TraceableRecord(Record): def _get_obj_class(self, url): uri_to_obj_class_map = { + "circuits/circuit-terminations": CircuitTerminations, "dcim/cables": Cables, + "dcim/console-ports": ConsolePorts, + "dcim/console-server-ports": ConsoleServerPorts, "dcim/front-ports": FrontPorts, "dcim/interfaces": Interfaces, + "dcim/power-feeds": PowerFeeds, + "dcim/power-outlets": PowerOutlets, + "dcim/power-ports": PowerPorts, "dcim/rear-ports": RearPorts, } @@ -154,6 +160,10 @@ class Interfaces(TraceableRecord): interface_connection = InterfaceConnection +class PowerFeeds(TraceableRecord): + pass + + class PowerOutlets(TraceableRecord): device = Devices diff --git a/pynetbox/models/mapper.py b/pynetbox/models/mapper.py index 1dcfc99..01d9f5a 100644 --- a/pynetbox/models/mapper.py +++ b/pynetbox/models/mapper.py @@ -7,6 +7,7 @@ DeviceTypes, FrontPorts, Interfaces, + PowerFeeds, PowerOutlets, PowerPorts, RackReservations, @@ -48,7 +49,7 @@ "dcim.modulebaytemplate": None, "dcim.moduletype": None, "dcim.platform": None, - "dcim.powerfeed": None, + "dcim.powerfeed": PowerFeeds, "dcim.poweroutlet": PowerOutlets, "dcim.poweroutlettemplate": None, "dcim.powerpanel": None,