18 lines
457 B
Python
18 lines
457 B
Python
import requests
|
|
|
|
from shipyard_airflow.errors import AirflowError
|
|
|
|
|
|
class AirflowClient(object):
|
|
def __init__(self, url):
|
|
self.url = url
|
|
|
|
def get(self):
|
|
response = requests.get(self.url).json()
|
|
|
|
# This gives us more freedom to handle the responses from airflow
|
|
if response["output"]["stderr"]:
|
|
raise AirflowError(response["output"]["stderr"])
|
|
else:
|
|
return response["output"]["stdout"]
|