1 files changed, 102 insertions, 0 deletions
diff --git a/mod/distributorapi/distributor/runtime_client.py b/mod/distributorapi/distributor/runtime_client.py
new file mode 100644
index 0000000..7cd06ae
--- /dev/null
+++ b/mod/distributorapi/distributor/runtime_client.py
@@ -0,0 +1,102 @@
+# ============LICENSE_START=======================================================
+# Copyright (c) 2019 AT&T Intellectual Property. All rights reserved.
+# ================================================================================
+# 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.
+# ============LICENSE_END=========================================================
+"""Runtime API client"""
+
+import requests as reqs
+from distributor import errors
+from distributor.utils import urljoin, get_json
+
+
+def get_graph(runtime_url, graph_id):
+    # REVIEW: There's only support for one graph right now..
+    url = urljoin(runtime_url, "api/graph/main")
+    return get_json(url)
+
+
+def create_graph(runtime_url, graph_id, graph_name):
+    url = urljoin(runtime_url, "api/graph/main")
+
+    resp = reqs.post(url, json={"name": graph_name, "id": graph_id
+        , "description": "", "main": True})
+
+    try:
+        resp.raise_for_status()
+    except Exception as e:
+        raise errors.DistributorAPIError(e)
+
+
+def delete_graph(runtime_url):
+    url = urljoin(runtime_url, "api/graph/main")
+
+    try:
+        reqs.delete(url).raise_for_status()
+    except Exception as e:
+        raise errors.DistributorAPIError(e)
+
+
+def post_graph(runtime_url, graph_id, actions):
+    url = urljoin(runtime_url, "api/graph", graph_id, "distribute")
+    graph_request = {"actions": actions}
+
+    resp = reqs.post(url, json=graph_request)
+
+    try:
+        resp.raise_for_status()
+        # REVIEW: Should be blueprint
+        return resp.json()
+    except Exception as e:
+        with open("runtime-request-failed.json", "w+") as f:
+            import json
+            json.dump(graph_request, f)
+        raise errors.DistributorAPIError(e)
+
+
+def ensure_graph(runtime_url, pg_id, pg_name, max_attempts=6):
+    """Ensures the graph with the specified id will exist"""
+    # TODO: Remove this when runtime API more mature
+    # Added this attempted delete call here to make sure repeated calls to post
+    # flows works by making sure the runtime API main graph is always empty
+    try:
+        delete_graph(runtime_url)
+    except:
+        # Probably a 404, doesn't matter
+        pass
+
+    # The attempts are not *really* attempts because attempts equates to looping
+    # twice
+    for i in range(0, max_attempts):
+        resp = None
+
+        try:
+            resp = get_graph(runtime_url, pg_id)
+        except Exception as e:
+            # Assuming you are here because graph needs to be created
+            create_graph(runtime_url, pg_id, pg_name)
+
+        # TODO: Runtime api only supports 1 graph which is why this check is
+        # here. Make sure it will support many graphs and remove this
+
+        if resp == None:
+            # You are here because the graph was created for first time or
+            # the graph was deleted then created. Anyways next loop should
+            # check if it got created ok
+            continue
+        elif resp != None and resp["id"] != pg_id:
+            delete_graph(runtime_url)
+        elif resp != None and resp["id"] == pg_id:
+            return True
+
+    return False