From 6f052614f99b9d98731a86d72c67d5dc4dc3e22d Mon Sep 17 00:00:00 2001 From: Parker Berberian Date: Wed, 20 Dec 2017 13:30:31 -0500 Subject: Allows for manual addition of bookings JIRA: None The added action takes a booking id from the dashboard and manually adds it to the list of bookings it tracks. This allows us to have a much easier transition to using a fully automated system after manually tracking all bookings. Change-Id: I83ebe2a4aabad2ce7d59b3d80659fc6438fda7f4 Signed-off-by: Parker Berberian --- laas-fog/pharoslaas/actions/add_booking.py | 61 ++++++++++++++++++++++++++++ laas-fog/pharoslaas/actions/add_booking.yaml | 26 ++++++++++++ 2 files changed, 87 insertions(+) create mode 100644 laas-fog/pharoslaas/actions/add_booking.py create mode 100644 laas-fog/pharoslaas/actions/add_booking.yaml diff --git a/laas-fog/pharoslaas/actions/add_booking.py b/laas-fog/pharoslaas/actions/add_booking.py new file mode 100644 index 0000000..b34c23d --- /dev/null +++ b/laas-fog/pharoslaas/actions/add_booking.py @@ -0,0 +1,61 @@ +############################################################################## +# Copyright 2017 Parker Berberian and Others # +# # +# Licensed under the Apache License, Version 2.0 (the "License"); # +# you may not use this file except in compliance with the License. # +# You may obtain a copy of the License at # +# # +# http://www.apache.org/licenses/LICENSE-2.0 # +# # +# Unless required by applicable law or agreed to in writing, software # +# distributed under the License is distributed on an "AS IS" BASIS, # +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # +# See the License for the specific language governing permissions and # +# limitations under the License. # +############################################################################## +import json +import requests +import time +import calendar +from st2actions.runners.pythonrunner import Action + + +class Add_Booking_Action(Action): + + def run(self, booking_id): + dashboard = self.action_service.get_value(name="dashboard_url") + url = dashboard + "api/bookings/" + str(booking_id) + try: + booking = requests.get(url).json() + booking['start'] = self.convert_time(booking['start']) + booking['end'] = self.convert_time(booking['end']) + booking['status'] = 1 + + # add booking id to bookings list + bookings = json.loads( + self.action_service.get_value(name="bookings", local=False) + ) + if booking['id'] in bookings: + return + bookings.append(booking['id']) + self.action_service.set_value( + name="bookings", + value=json.dumps(bookings), + local=False + ) + + # add booking to datastore + name = "booking_" + str(booking['id']) + self.action_service.set_value( + name=name, + value=json.dumps(booking), + local=False + ) + + except Exception: + pass + + def convert_time(self, timestr): + time_struct = time.strptime(timestr, '%Y-%m-%dT%H:%M:%SZ') + epoch_time = calendar.timegm(time_struct) + return epoch_time diff --git a/laas-fog/pharoslaas/actions/add_booking.yaml b/laas-fog/pharoslaas/actions/add_booking.yaml new file mode 100644 index 0000000..0851599 --- /dev/null +++ b/laas-fog/pharoslaas/actions/add_booking.yaml @@ -0,0 +1,26 @@ +--- +############################################################################## +# Copyright 2017 Parker Berberian and Others # +# # +# Licensed under the Apache License, Version 2.0 (the "License"); # +# you may not use this file except in compliance with the License. # +# You may obtain a copy of the License at # +# # +# http://www.apache.org/licenses/LICENSE-2.0 # +# # +# Unless required by applicable law or agreed to in writing, software # +# distributed under the License is distributed on an "AS IS" BASIS, # +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # +# See the License for the specific language governing permissions and # +# limitations under the License. # +############################################################################## + +name: add_booking +enabled: true +runner_type: python-script +entry_point: add_booking.py +parameters: + booking: + type: string + required: true + description: "id of booking to add" -- cgit 1.2.3-korg