diff --git a/netbox_awx_plugin/serializers.py b/netbox_awx_plugin/serializers.py index 7e2b7c9d8d20e524673bdf2a468c4669fb6e1612..f21e5c1858026734b0284f1a852d9c87f17cd3ca 100644 --- a/netbox_awx_plugin/serializers.py +++ b/netbox_awx_plugin/serializers.py @@ -85,7 +85,10 @@ class IPAddressSerializer(serializers.BaseSerializer): class DeviceSerializer(serializers.BaseSerializer): def to_representation(self, instance): - variables = {} + variables = { + "netbox_model_label": instance._meta.label, + "netbox_id": instance.id, + } variables["netbox_interfaces"] = [] for interface in instance.interfaces.all(): serializer = InterfaceSerializer(interface) @@ -112,6 +115,8 @@ class VMInterfaceSerializer(serializers.BaseSerializer): class VMSerializer(serializers.BaseSerializer): def to_representation(self, instance): variables = { + "netbox_model_label": instance._meta.label, + "netbox_id": instance.id, "netbox_virtualmachine_name": instance.name, "netbox_virtualmachine_vcpus": float(instance.vcpus) if instance.vcpus is not None else 0.0, "netbox_virtualmachine_memory": instance.memory or 0, diff --git a/netbox_awx_plugin/tests/test_serializers.py b/netbox_awx_plugin/tests/test_serializers.py index 315372c6e6de185ab80bcef250f46cd5f9161afb..df1580cb2d132401396749b55202c03506cc7189 100644 --- a/netbox_awx_plugin/tests/test_serializers.py +++ b/netbox_awx_plugin/tests/test_serializers.py @@ -225,6 +225,8 @@ class DeviceSerializerTest(TestCase): "description": 'Test Device Description', "enabled": True, "variables": json.dumps({ + "netbox_model_label": "dcim.Device", + "netbox_id": 2, "netbox_interfaces": [ { "name": 'eth0', @@ -309,6 +311,8 @@ class VMSerializerTest(TestCase): "description": 'Test VM Description', "enabled": True, "variables": json.dumps({ + "netbox_model_label": "virtualization.VirtualMachine", + "netbox_id": 3, "netbox_virtualmachine_name": 'Test VM', "netbox_virtualmachine_vcpus": 4.0, "netbox_virtualmachine_memory": 8192, @@ -421,6 +425,8 @@ class DeviceSerializerNoPrimaryIPTest(TestCase): "description": 'Test Device Description', "enabled": True, "variables": json.dumps({ + "netbox_model_label": "dcim.Device", + "netbox_id": 1, "netbox_interfaces": [ { "name": 'eth0', @@ -471,6 +477,8 @@ class VMSerializerNoPrimaryIPTest(TestCase): "description": 'Test VM Description', "enabled": True, "variables": json.dumps({ + "netbox_model_label": "virtualization.VirtualMachine", + "netbox_id": 2, "netbox_virtualmachine_name": 'Test VM', "netbox_virtualmachine_vcpus": 4.0, "netbox_virtualmachine_memory": 8192,