From e00d336aeb6da542d8a54f157ab2e285cd4a3e57 Mon Sep 17 00:00:00 2001 From: Parker Berberian Date: Tue, 12 Mar 2019 10:28:48 -0400 Subject: Allow labs to retirieve and update some host information in the api Change-Id: Ib0682141351f7789c50d98a992ab166c2f033f4b Signed-off-by: Parker Berberian --- src/api/models.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'src/api/models.py') diff --git a/src/api/models.py b/src/api/models.py index 7109bbe..b35adf2 100644 --- a/src/api/models.py +++ b/src/api/models.py @@ -116,6 +116,22 @@ class LabManager(object): inventory['host_types'] = self.serialize_host_profiles(profiles) return inventory + def get_host(self, hostname): + host = get_object_or_404(Host, labid=hostname, lab=self.lab) + return { + "booked": host.booked, + "working": host.working, + "type": host.profile.name + } + + def update_host(self, hostname, data): + host = get_object_or_404(Host, labid=hostname, lab=self.lab) + if "working" in data: + working = data['working'] == "true" + host.working = working + host.save() + return self.get_host(hostname) + def get_status(self): return {"status": self.lab.status} -- cgit 1.2.3-korg