added locations service, added all availabilities endpoint,

use proper url query params

Author: tamirse

datacrunch/constants.py

@@ -56,8 +56,8 @@ def __init__(self):
 
 
 class Locations:
-    FIN_01 = "FIN-01"
-    ICE_01 = "ICE-01"
+    FIN_01: str = "FIN-01"
+    ICE_01: str = "ICE-01"
 
     def __init__(self):
         return

datacrunch/datacrunch.py

@@ -9,6 +9,7 @@
 from datacrunch.volume_types.volume_types import VolumeTypesService
 from datacrunch.volumes.volumes import VolumesService
 from datacrunch.constants import Constants
+from datacrunch.locations.locations import LocationsService
 from datacrunch.__version__ import VERSION
 
 
@@ -62,3 +63,7 @@ def __init__(self, client_id: str, client_secret: str, base_url: str = "https://
 
         self.volumes: VolumesService = VolumesService(self._http_client)
         """Volume service. Create, attach, detach, get, rename, delete volumes"""
+
+        self.locations: LocationsService = LocationsService(
+            self._http_client)
+        """Locations service. Get locations"""

datacrunch/http_client/http_client.py

@@ -32,7 +32,7 @@ def __init__(self, auth_service, base_url: str) -> None:
         self._auth_service = auth_service
         self._auth_service.authenticate()
 
-    def post(self, url: str, json: dict = None, **kwargs) -> requests.Response:
+    def post(self, url: str, json: dict = None, params: dict = None, **kwargs) -> requests.Response:
         """Sends a POST request.
 
         A wrapper for the requests.post method.
@@ -43,6 +43,8 @@ def post(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         :type url: str
         :param json: A JSON serializable Python object to send in the body of the Request, defaults to None
         :type json: dict, optional
+        :param params: Dictionary of querystring data to attach to the Request, defaults to None
+        :type params: dict, optional
 
         :raises APIException: an api exception with message and error type code
 
@@ -54,12 +56,13 @@ def post(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         url = self._add_base_url(url)
         headers = self._generate_headers()
 
-        response = requests.post(url, json=json, headers=headers, **kwargs)
+        response = requests.post(
+            url, json=json, headers=headers, params=params, **kwargs)
         handle_error(response)
 
         return response
 
-    def put(self, url: str, json: dict = None, **kwargs) -> requests.Response:
+    def put(self, url: str, json: dict = None, params: dict = None, **kwargs) -> requests.Response:
         """Sends a PUT request.
 
         A wrapper for the requests.put method.
@@ -70,6 +73,8 @@ def put(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         :type url: str
         :param json: A JSON serializable Python object to send in the body of the Request, defaults to None
         :type json: dict, optional
+        :param params: Dictionary of querystring data to attach to the Request, defaults to None
+        :type params: dict, optional
 
         :raises APIException: an api exception with message and error type code
 
@@ -81,7 +86,8 @@ def put(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         url = self._add_base_url(url)
         headers = self._generate_headers()
 
-        response = requests.put(url, json=json, headers=headers, **kwargs)
+        response = requests.put(
+            url, json=json, headers=headers, params=params, **kwargs)
         handle_error(response)
 
         return response
@@ -95,7 +101,7 @@ def get(self, url: str, params: dict = None, **kwargs) -> requests.Response:
 
         :param url: relative url of the API endpoint
         :type url: str
-        :param params: Dictionary, list of tuples or bytes to send in the query string for the Request. defaults to None
+        :param params: Dictionary of querystring data to attach to the Request, defaults to None
         :type params: dict, optional
 
         :raises APIException: an api exception with message and error type code
@@ -113,7 +119,7 @@ def get(self, url: str, params: dict = None, **kwargs) -> requests.Response:
 
         return response
 
-    def delete(self, url: str, json: dict = None, **kwargs) -> requests.Response:
+    def delete(self, url: str, json: dict = None, params: dict = None, **kwargs) -> requests.Response:
         """Sends a DELETE request.
 
         A wrapper for the requests.delete method.
@@ -124,6 +130,8 @@ def delete(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         :type url: str
         :param json: A JSON serializable Python object to send in the body of the Request, defaults to None
         :type json: dict, optional
+        :param params: Dictionary of querystring data to attach to the Request, defaults to None
+        :type params: dict, optional
 
         :raises APIException: an api exception with message and error type code
 
@@ -135,7 +143,8 @@ def delete(self, url: str, json: dict = None, **kwargs) -> requests.Response:
         url = self._add_base_url(url)
         headers = self._generate_headers()
 
-        response = requests.delete(url, headers=headers, json=json, **kwargs)
+        response = requests.delete(
+            url, headers=headers, json=json, params=params, **kwargs)
         handle_error(response)
 
         return response

datacrunch/instances/instances.py

@@ -426,16 +426,33 @@ def action(self, id_list: Union[List[str], str], action: str, volume_ids: Option
         self._http_client.put(INSTANCES_ENDPOINT, json=payload)
         return
 
-    def is_available(self, instance_type: str, is_spot: bool = None) -> bool:
+    # TODO: use enum/const for location_code
+    def is_available(self, instance_type: str, is_spot: bool = None, location_code: str = Locations.FIN_01) -> bool:
         """Returns True if a specific instance type is now available for deployment
 
         :param instance_type: instance type
         :type instance_type: str
         :param is_spot: Is spot instance
         :type is_spot: bool, optional
+        :param location_code: datacenter location, defaults to "FIN-01"
+        :type location_code: str, optional
         :return: True if available to deploy, False otherwise
         :rtype: bool
         """
-        query_param = '?isSpot=true' if is_spot else ''
-        url = f'/instance-availability/{instance_type}{query_param}'
-        return self._http_client.get(url).json()
+        query_params = {'isSpot': is_spot, 'location_code': location_code}
+        url = f'/instance-availability/{instance_type}'
+        return self._http_client.get(url, query_params).json()
+
+    # TODO: use enum/const for location_code
+    def get_availabilities(self, is_spot: bool = None, location_code: str = Locations.FIN_01) -> bool:
+        """Returns a list of available instance types
+
+        :param is_spot: Is spot instance
+        :type is_spot: bool, optional
+        :param location_code: datacenter location, defaults to "FIN-01"
+        :type location_code: str, optional
+        :return: list of available instance types in every location
+        :rtype: list
+        """
+        query_params = {'isSpot': is_spot, 'location_code': location_code}
+        return self._http_client.get('/instance-availability', params=query_params).json()

datacrunch/locations/__init__.py

@@ -0,0 +1,16 @@
+from typing import List
+
+LOCATIONS_ENDPOINT = '/locations'
+
+
+class LocationsService:
+    """A service for interacting with the locations endpoint"""
+
+    def __init__(self, http_client) -> None:
+        self._http_client = http_client
+
+    def get(self) -> List[dict]:
+        """Get all locations
+        """
+        locations = self._http_client.get(LOCATIONS_ENDPOINT).json()
+        return locations