fix rack comparison while updating location (#84)

Solvik · Sep 12, 2019 · a129781 · a129781
1 parent 06328e7
commit a129781
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/netbox_agent/server.py b/netbox_agent/server.py
@@ -42,15 +42,15 @@ def update_netbox_location(self, server):
         nb_dc = self.get_netbox_datacenter()
 
         update = False
-        if dc and server.site.slug != nb_dc.slug:
+        if dc and server.site and server.site.slug != nb_dc.slug:
             logging.info('Datacenter location has changed from {} to {}, updating'.format(
                 server.site.slug,
                 nb_dc.slug,
             ))
             update = True
             server.site = nb_dc.id
 
-        if rack and server.rack != nb_rack:
+        if rack and server.rack and server.rack.id != nb_rack.id:
             logging.info('Rack location has changed from {} to {}, updating'.format(
                 server.rack,
                 nb_rack,