From 0a76d962ce87b716a648a0154646bdd03fcf322b Mon Sep 17 00:00:00 2001 From: Daniel Sheppard Date: Fri, 17 May 2024 11:12:08 -0500 Subject: [PATCH] Move duplicated dicts into inheritable meta class --- netbox/dcim/tables/devices.py | 62 ++++++++--------------------------- 1 file changed, 14 insertions(+), 48 deletions(-) diff --git a/netbox/dcim/tables/devices.py b/netbox/dcim/tables/devices.py index ad9caa03c..5ebb0f715 100644 --- a/netbox/dcim/tables/devices.py +++ b/netbox/dcim/tables/devices.py @@ -43,12 +43,14 @@ MODULEBAY_STATUS = """ """ -def get_cabletermination_row_class(record): - if record.mark_connected: - return 'success' - elif record.cable: - return record.cable.get_status_color() - return '' +class ConnectionDeviceComponentTableMixin: + class Meta: + row_attrs = { + 'data-name': lambda record: record.name, + 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", + 'data-cable-status': lambda record: record.cable.status if record.cable else "", + 'data-type': lambda record: record.type + } # @@ -382,19 +384,13 @@ class DeviceConsolePortTable(ConsolePortTable): extra_buttons=CONSOLEPORT_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.ConsolePort fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'speed', 'description', 'mark_connected', 'cable', 'cable_color', 'link_peer', 'connection', 'tags', 'actions' ) default_columns = ('pk', 'name', 'label', 'type', 'speed', 'description', 'cable', 'connection') - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class ConsoleServerPortTable(ModularDeviceComponentTable, PathEndpointTable): @@ -430,19 +426,13 @@ class DeviceConsoleServerPortTable(ConsoleServerPortTable): extra_buttons=CONSOLESERVERPORT_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.ConsoleServerPort fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'speed', 'description', 'mark_connected', 'cable', 'cable_color', 'link_peer', 'connection', 'tags', 'actions', ) default_columns = ('pk', 'name', 'label', 'type', 'speed', 'description', 'cable', 'connection') - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class PowerPortTable(ModularDeviceComponentTable, PathEndpointTable): @@ -485,7 +475,7 @@ class DevicePowerPortTable(PowerPortTable): extra_buttons=POWERPORT_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.PowerPort fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'maximum_draw', 'allocated_draw', @@ -494,12 +484,6 @@ class DevicePowerPortTable(PowerPortTable): default_columns = ( 'pk', 'name', 'label', 'type', 'maximum_draw', 'allocated_draw', 'description', 'cable', 'connection', ) - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class PowerOutletTable(ModularDeviceComponentTable, PathEndpointTable): @@ -539,7 +523,7 @@ class DevicePowerOutletTable(PowerOutletTable): extra_buttons=POWEROUTLET_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.PowerOutlet fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'power_port', 'feed_leg', 'description', @@ -548,12 +532,6 @@ class DevicePowerOutletTable(PowerOutletTable): default_columns = ( 'pk', 'name', 'label', 'type', 'power_port', 'feed_leg', 'description', 'cable', 'connection', ) - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class BaseInterfaceTable(NetBoxTable): @@ -745,7 +723,7 @@ class DeviceFrontPortTable(FrontPortTable): extra_buttons=FRONTPORT_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.FrontPort fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'rear_port', 'rear_port_position', @@ -754,12 +732,6 @@ class DeviceFrontPortTable(FrontPortTable): default_columns = ( 'pk', 'name', 'label', 'type', 'rear_port', 'rear_port_position', 'description', 'cable', 'link_peer', ) - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class RearPortTable(ModularDeviceComponentTable, CableTerminationTable): @@ -798,7 +770,7 @@ class DeviceRearPortTable(RearPortTable): extra_buttons=REARPORT_BUTTONS ) - class Meta(DeviceComponentTable.Meta): + class Meta(ConnectionDeviceComponentTableMixin.Meta, DeviceComponentTable.Meta): model = models.RearPort fields = ( 'pk', 'id', 'name', 'module_bay', 'module', 'label', 'type', 'positions', 'description', 'mark_connected', @@ -807,12 +779,6 @@ class DeviceRearPortTable(RearPortTable): default_columns = ( 'pk', 'name', 'label', 'type', 'positions', 'description', 'cable', 'link_peer', ) - row_attrs = { - 'data-name': lambda record: record.name, - 'data-mark-connected': lambda record: "true" if record.mark_connected else "false", - 'data-cable-status': lambda record: record.cable.status if record.cable else "", - 'data-type': lambda record: record.type - } class DeviceBayTable(DeviceComponentTable):