From 4cb0be4df3cb01f4708c9ccc85f7a9ace69cde82 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Fri, 13 Oct 2017 10:42:45 -0400
Subject: [PATCH] Fixes #1582: Add virtual_machine attribute to IPAddress

---
 netbox/ipam/models.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/netbox/ipam/models.py b/netbox/ipam/models.py
index 791ba1b1a..73a522d63 100644
--- a/netbox/ipam/models.py
+++ b/netbox/ipam/models.py
@@ -440,7 +440,7 @@ class IPAddress(CreatedUpdatedModel, CustomFieldModel):
             self.get_status_display(),
             self.get_role_display(),
             self.device.identifier if self.device else None,
-            self.virtual_machine.name if self.device else None,
+            self.virtual_machine.name if self.virtual_machine else None,
             self.interface.name if self.interface else None,
             is_primary,
             self.description,
@@ -452,6 +452,12 @@ class IPAddress(CreatedUpdatedModel, CustomFieldModel):
             return self.interface.device
         return None
 
+    @property
+    def virtual_machine(self):
+        if self.interface:
+            return self.interface.virtual_machine
+        return None
+
     def get_status_class(self):
         return STATUS_CHOICE_CLASSES[self.status]