diff options
Diffstat (limited to 'src/liblaas/views.py')
| -rw-r--r-- | src/liblaas/views.py | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/src/liblaas/views.py b/src/liblaas/views.py new file mode 100644 index 0000000..5edc727 --- /dev/null +++ b/src/liblaas/views.py @@ -0,0 +1,205 @@ +############################################################################## +# Copyright (c) 2018 Sawyer Bergeron and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +# Unauthenticated requests to liblaas. If a call makes it to here, it is assumed to be authenticated + +import os +import requests +import json + +base = os.environ.get("LIBLAAS_BASE_URL") +post_headers = {'Content-Type': 'application/json'} + +def liblaas_docs(): + endpoint = f'docs' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +### BOOKING + +# DELETE +def booking_end_booking(agg_id: str) -> requests.Response: + endpoint = f'booking/{agg_id}/end' + url = f'{base}{endpoint}' + try: + response = requests.delete(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# GET +def booking_booking_status(agg_id: str) -> requests.Response: + endpoint = f'booking/{agg_id}/status' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# POST +def booking_create_booking(booking_blob: dict) -> requests.Response: + endpoint = f'booking/create' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(booking_blob), headers=post_headers) + return response.json() + except: + print(f"Error at {url}") + return None + +### FLAVOR + +# GET +def flavor_list_flavors(project: str) -> requests.Response: + endpoint = f'flavor' #todo - add project to url once liblaas supports it + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# GET +def flavor_get_flavor_by_id(flavor_id: str) -> requests.Response: + endpoint = f'flavor/name/{flavor_id}/' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# GET +def flavor_list_hosts(project: str) -> requests.Response: + endpoint = f'flavor/hosts/{project}' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +### TEMPLATE + +# GET +def template_list_templates(uid: str) -> requests.Response: + endpoint = f'template/list/{uid}' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# DELETE +def template_delete_template(template_id: str) -> requests.Response: + endpoint = f'template/{template_id}' + url = f'{base}{endpoint}' + try: + response = requests.delete(url) + return response.json() + except: + print(f"Error at {url}") + return None + +#POST +def template_make_template(template_blob: dict) -> requests.Response: + endpoint = f'template/create' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(template_blob), headers=post_headers) + return response.json() + except: + print(f"Error at {url}") + return None + +### USER + +# GET +def user_get_user(uid: str) -> requests.Response: + endpoint = f'user/{uid}' + url = f'{base}{endpoint}' + try: + response = requests.get(url) + return response.json() + except: + print(f"Error at {url}") + return None + +# POST +def user_create_user(user_blob: dict) -> requests.Response: + endpoint = f'user/create' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(user_blob), headers=post_headers) + # LibLaaS is not returning anything here regardless, so we need some way of determining if it was successful + # return response.json() + + if response.status_code == 200: + return user_blob["uid"] + + return response.json() + except: + print(f"Error at {url}") + return None + +# POST +def user_set_ssh(uid: str, keys: list) -> requests.Response: + endpoint = f'user/{uid}/ssh' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(clean_ssh_keys(keys)), headers=post_headers) + # return response.json() + return response.status_code == 200 + except: + print(f"Error at {url}") + return None + +# POST +def user_set_company(uid: str, company: str): + endpoint = f'user/{uid}/company' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(company), headers=post_headers) + # return response.json() + return response.status_code == 200 + except: + print(f"Error at {url}") + return None + +# POST +def user_set_email(uid: str, email: str): + endpoint = f'user/{uid}/email' + url = f'{base}{endpoint}' + try: + response = requests.post(url, data=json.dumps(email), headers=post_headers) + # return response.json() + return response.status_code == 200 + except: + print(f"Error at {url}") + return None + +# utils +def clean_ssh_keys(ssh_key_list): + cleaned = [] + for key in ssh_key_list: + cleaned.append(key.strip()) + return cleaned
\ No newline at end of file |