1 files changed, 53 insertions, 0 deletions
diff --git a/scripts/rebuild-endpoints.py b/scripts/rebuild-endpoints.py
new file mode 100644
index 00000000..37ac37d8
--- /dev/null
+++ b/scripts/rebuild-endpoints.py
@@ -0,0 +1,53 @@
+import json
+from pyquery import PyQuery as pq
+import requests
+
+
+class FetchError(Exception):
+    pass
+
+
+def fetch_endpoints():
+    # We utilize what the Java SDK publishes as a baseline.
+    resp = requests.get('https://raw2.github.com/aws/aws-sdk-java/master/src/main/resources/etc/regions.xml')
+
+    if int(resp.status_code) != 200:
+        raise FetchError("Failed to fetch the endpoints. Got {0}: {1}".format(
+            resp.status,
+            resp.body
+        ))
+
+    return resp.text
+
+
+def parse_xml(raw_xml):
+    return pq(raw_xml, parser='xml')
+
+
+def build_data(doc):
+    data = {}
+
+    # Run through all the regions. These have all the data we need.
+    for region_elem in doc('Regions').find('Region'):
+        region = pq(region_elem, parser='xml')
+        region_name = region.find('Name').text()
+
+        for endp in region.find('Endpoint'):
+            service_name = endp.find('ServiceName').text
+            endpoint = endp.find('Hostname').text
+
+            data.setdefault(service_name, {})
+            data[service_name][region_name] = endpoint
+
+    return data
+
+
+def main():
+    raw_xml = fetch_endpoints()
+    doc = parse_xml(raw_xml)
+    data = build_data(doc)
+    print(json.dumps(data, indent=4, sort_keys=True))
+
+
+if __name__ == '__main__':
+    main()