From 1dfdea94584855c8ffbe3f8dd660eda005fa9594 Mon Sep 17 00:00:00 2001 From: Watchful1 Date: Mon, 18 Sep 2023 18:11:17 -0700 Subject: [PATCH] And flush --- personal/combine/build_day.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/personal/combine/build_day.py b/personal/combine/build_day.py index e572fed..9e65a80 100644 --- a/personal/combine/build_day.py +++ b/personal/combine/build_day.py @@ -44,11 +44,17 @@ def re_auth_pushshift(old_token): response = requests.post(url) result = response.json() log.warning(f"Reauth response: {str(result)}") - new_token = result['access_token'] - log.warning(f"New pushshift token: {new_token}") - save_pushshift_token(new_token) discord_logging.flush_discord() - return new_token + if 'access_token' in result: + new_token = result['access_token'] + log.warning(f"New pushshift token: {new_token}") + save_pushshift_token(new_token) + discord_logging.flush_discord() + return new_token + elif 'detail' in result: + log.warning(f"Reauth failed: {result['detail']}") + discord_logging.flush_discord() + sys.exit(1) def query_pushshift(ids, bearer, object_type): @@ -76,6 +82,7 @@ def query_pushshift(ids, bearer, object_type): time.sleep(2) if response.status_code != 200: log.warning(f"4 requests failed with status code {response.status_code}") + sys.exit(1) return response.json()['data'], bearer