From 1fb06c96e32d8a9b76f153cf8a9313346d42daa9 Mon Sep 17 00:00:00 2001
From: Marco Ceppi <marco@ceppi.net>
Date: Wed, 18 Dec 2019 17:35:23 -0500
Subject: [PATCH] Make sure racks get a lookup from rack name and site

---
 docker/startup_scripts/110_devices.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docker/startup_scripts/110_devices.py b/docker/startup_scripts/110_devices.py
index 16c6f9ee6..584b5c255 100644
--- a/docker/startup_scripts/110_devices.py
+++ b/docker/startup_scripts/110_devices.py
@@ -26,7 +26,6 @@ with file.open('r') as stream:
   optional_assocs = {
     'tenant': (Tenant, 'name'),
     'platform': (Platform, 'name'),
-    'rack': (Rack, 'name'),
     'cluster': (Cluster, 'name'),
     'primary_ip4': (IPAddress, 'address'),
     'primary_ip6': (IPAddress, 'address')
@@ -49,6 +48,9 @@ with file.open('r') as stream:
 
           params[assoc] = model.objects.get(**query)
 
+      if 'rack' in params:
+        params['rack'] = Rack.objects.get(name=params.pop('rack'), site=params['site'].id)
+
       if 'face' in params:
         for rack_face in RACK_FACE_CHOICES:
           if params['face'] in rack_face: