Closes #1368: Render reservations in rack elevations view

This commit is contained in:
Jeremy Stretch
2017-08-02 13:33:44 -04:00
parent 52e7f670f3
commit b6690e0012
4 changed files with 14 additions and 9 deletions

View File

@@ -357,6 +357,16 @@ class Rack(CreatedUpdatedModel, CustomFieldModel):
return list(reversed(available_units))
def get_reserved_units(self):
"""
Return a dictionary mapping all reserved units within the rack to their reservation.
"""
reserved_units = {}
for r in self.reservations.all():
for u in r.units:
reserved_units[u] = r
return reserved_units
def get_0u_devices(self):
return self.devices.filter(position=0)