Skip to content

Commit 282b794

Browse files
committed
First draft of API for retrieving runs
1 parent 5c3352f commit 282b794

File tree

1 file changed

+39
-0
lines changed

1 file changed

+39
-0
lines changed

simvue/client.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,45 @@ def __init__(self):
3636
self._url, self._token = get_auth()
3737
self._headers = {"Authorization": f"Bearer {self._token}"}
3838

39+
def get_run(self, run, system=False, metadata=False):
40+
"""
41+
Get a single run
42+
"""
43+
params = {'name': run,
44+
'filter': None,
45+
'system': system,
46+
'metadata': metadata}
47+
48+
try:
49+
response = requests.get(f"{self._url}/api/runs", headers=self._headers, params=params)
50+
except requests.exceptions.RequestException:
51+
return None
52+
53+
if response.status_code == 200:
54+
return response.json()
55+
56+
return None
57+
58+
59+
def get_runs(self, filter, system=False, metadata=False):
60+
"""
61+
Get runs
62+
"""
63+
params = {'name': None,
64+
'filter': filter,
65+
'system': system,
66+
'metadata': metadata}
67+
68+
try:
69+
response = requests.get(f"{self._url}/api/runs", headers=self._headers, params=params)
70+
except requests.exceptions.RequestException:
71+
return None
72+
73+
if response.status_code == 200:
74+
return response.json()
75+
76+
return None
77+
3978
def list_artifacts(self, run, category=None):
4079
"""
4180
List artifacts associated with a run

0 commit comments

Comments
 (0)