From 6f9feaea10c7e3e5bb0dde82148e0763a473b239 Mon Sep 17 00:00:00 2001 From: Kyler Date: Thu, 28 May 2020 17:38:15 -0500 Subject: [PATCH] :recycle: Use site name in query for racks when building devices (#29) --- docker/startup_scripts/110_devices.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/docker/startup_scripts/110_devices.py b/docker/startup_scripts/110_devices.py index 4217549d7..6acabb146 100644 --- a/docker/startup_scripts/110_devices.py +++ b/docker/startup_scripts/110_devices.py @@ -44,7 +44,14 @@ with file.open('r') as stream: for assoc, details in optional_assocs.items(): if assoc in params: model, field = details - query = { field: params.pop(assoc) } + + if assoc == 'rack': # Special handling for rack query to reference rack name and site + query = { + 'site': params.get('site'), + field: params.pop(assoc), + } + else: + query = { field: params.pop(assoc) } params[assoc] = model.objects.get(**query)