From 258b804576e1babe59100537e687b1c1929c272d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maximilian=20Ke=C3=9Fler?= <git@maximilian-kessler.de>
Date: Sun, 14 May 2023 14:04:57 +0200
Subject: [PATCH] add alive_bar to requirements

---
 requirements.txt |  1 +
 site_api.py      | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 32d5f74..714ec44 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ termcolor
 more_itertools
 psycopg2
 alive_progress
+argparse
diff --git a/site_api.py b/site_api.py
index 58ac19b..499618f 100644
--- a/site_api.py
+++ b/site_api.py
@@ -1,31 +1,36 @@
 import json
 import requests
 import requests_cache
+from log_setup import logger
 
 
 # Cache all requests to site to reduce traffic and latency
 session = requests_cache.CachedSession('hanab.live')
 
 
-def get(url):
+def get(url, refresh=False):
     #    print("sending request for " + url)
     query = "https://hanab.live/" + url
-    response = session.get(query)
+    logger.debug("GET {}".format(query))
+    response = session.get(query, force_refresh=refresh)
     if not response:
-        raise RuntimeError("Failed to get request {} from hanab.live".format(query))
+        logger.error("Failed to get request {} from hanab.live".format(query))
+        return None
     if not response.status_code == 200:
         return None
     if "application/json" in response.headers['content-type']:
         return json.loads(response.text)
 
-def api(url):
+
+def api(url, refresh=False):
     link = "api/v1/" + url
     if "?" in url:
         link += "&"
     else:
         link += "?"
     link += "size=100"
-    return get(link)
+    return get(link, refresh)
+
 
 def replay(seed):
     r = api("seed/" + str(seed))