1 files changed, 56 insertions, 0 deletions
diff --git a/app/discover/fetchers/api/api_fetch_availability_zones.py b/app/discover/fetchers/api/api_fetch_availability_zones.py
new file mode 100644
index 0000000..196893b
--- /dev/null
+++ b/app/discover/fetchers/api/api_fetch_availability_zones.py
@@ -0,0 +1,56 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# 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                                  #
+###############################################################################
+from discover.fetchers.api.api_access import ApiAccess
+
+
+class ApiFetchAvailabilityZones(ApiAccess):
+    def __init__(self):
+        super(ApiFetchAvailabilityZones, self).__init__()
+
+    def get(self, project_id):
+        token = self.v2_auth_pwd(project_id)
+        if not token:
+            return []
+        ret = []
+        for region in self.regions:
+            ret.extend(self.get_for_region(project_id, region, token))
+        return ret
+
+    def get_for_region(self, project, region, token):
+        # we use os-availability-zone/detail rather than os-availability-zone,
+        # because the later does not inclde the "internal" zone in the results
+        endpoint = self.get_region_url_nover(region, "nova")
+        req_url = endpoint + "/v2/" + token["tenant"]["id"] + \
+                  "/os-availability-zone/detail"
+        headers = {
+            "X-Auth-Project-Id": project,
+            "X-Auth-Token": token["id"]
+        }
+        response = self.get_url(req_url, headers)
+        if "status" in response and int(response["status"]) != 200:
+            return []
+        ret = []
+        if "availabilityZoneInfo" not in response:
+            return []
+        azs = response["availabilityZoneInfo"]
+        if not azs:
+            return []
+        for doc in azs:
+            doc["id"] = doc["zoneName"]
+            doc["name"] = doc.pop("zoneName")
+            doc["master_parent_type"] = "region"
+            doc["master_parent_id"] = region
+            doc["parent_type"] = "availability_zones_folder"
+            doc["parent_id"] = region + "-availability_zones"
+            doc["parent_text"] = "Availability Zones"
+            doc["available"] = doc["zoneState"]["available"]
+            doc.pop("zoneState")
+            ret.append(doc)
+        return ret