1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
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
|
