Skip to content

Commit

Permalink
Only print JSON response of failed API requests
Browse files Browse the repository at this point in the history
If the response is not valid JSON, do not print it to stdout and only
log it to debug output.
  • Loading branch information
Kevinjil committed Dec 14, 2024
1 parent 7d7ed2e commit 34afef1
Showing 1 changed file with 15 additions and 9 deletions.
24 changes: 15 additions & 9 deletions submit/submit
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,15 @@ def error(msg: str) -> NoReturn:
exit(-1)


def print_if_json(text: str):
'''Print the text if it is valid JSON, and ignore otherwise'''
try:
data = json.loads(text)
print(json.dumps(data, indent=2))
except json.decoder.JSONDecodeError as e:
pass


def read_contests() -> list:
'''Read all contests from the API.
Expand Down Expand Up @@ -202,15 +211,14 @@ def do_api_request(name: str):
except requests.exceptions.RequestException as e:
raise RuntimeError(e)

logging.debug(f"API call '{name}' returned:\n{response.text}")
if response.status_code >= 300:
print(response.text)
print_if_json(response.text)
if response.status_code == 401:
raise RuntimeError('Authentication failed, please check your DOMjudge credentials in ~/.netrc.')
else:
raise RuntimeError(f'API request {name} failed (code {response.status_code}).')

logging.debug(f"API call '{name}' returned:\n{response.text}")

return json.loads(response.text)


Expand Down Expand Up @@ -322,11 +330,10 @@ def do_api_print():

response = requests.post(url, data=data, headers=headers)

logging.debug(f"API call 'printing' returned:\n{response.text}")

# The connection worked, but we may have received an HTTP error
logging.debug(f"API call '{name}' returned:\n{response.text}")
if response.status_code >= 300:
print(response.text)
print_if_json(response.text)
if response.status_code == 401:
raise RuntimeError('Authentication failed, please check your DOMjudge credentials in ~/.netrc.')
else:
Expand Down Expand Up @@ -367,11 +374,10 @@ def do_api_submit():

response = requests.post(url, data=data, files=files, headers=headers)

logging.debug(f"API call 'submissions' returned:\n{response.text}")

# The connection worked, but we may have received an HTTP error
logging.debug(f"API call '{name}' returned:\n{response.text}")
if response.status_code >= 300:
print(response.text)
print_if_json(response.text)
if response.status_code == 401:
raise RuntimeError('Authentication failed, please check your DOMjudge credentials in ~/.netrc.')
else:
Expand Down

0 comments on commit 34afef1

Please sign in to comment.