def wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except ApiError as e:
ctype = e.response.headers.get("Content-Type", "text/plain")
headers = [("Content-Type", ctype)]
statuscode = e.response.status_code or 500
raise exceptions.BaseError(statuscode, headers, e.response.content)
def get(self, taskguid, **kwargs):
task = self.redis.get('tasks.{}'.format(taskguid))
greenlet = self.rest.tasks.get(taskguid)
if task is None and greenlet is None:
raise exceptions.NotFound("Task is not found")
elif task is None and greenlet is not None:
raise exceptions.BaseError(202, [], '')
elif task is not None:
return json.loads(task)
def _zero_access_request(self, ctx, url, params=None):
jwt = self._get_jwt(ctx)
resp = requests.get(
url,
headers={'Authorization': 'Bearer {jwt}'.format(jwt=jwt)},
params=params)
if resp.status_code != 200:
headers = []
for name, value in resp.headers.items():
if name.lower() not in ('connection', 'content-length'):
headers.append((name, value))
raise exceptions.BaseError(resp.status_code, headers,
j.tools.text.toStr(resp.text))
return resp.json()