Fixes #674: Include more types in trace() method (#684)

Author: arthanson

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
 

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,