refactor: make NREL API downloads resilient to 5xx errors

Breakthrough-Energy · Jan 8, 2022 · 466a803 · 466a803
1 parent 01b94e1
commit 466a803
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/prereise/gather/solardata/nsrdb/nrel_api.py b/prereise/gather/solardata/nsrdb/nrel_api.py
@@ -164,6 +164,10 @@ def get_psm3_at(
         )
         def download(url):
             resp = requests.get(url)
+            if resp.status_code // 100 == 5:  # all 5xx errors, server side
+                raise TransientError(
+                    f"Server side error, retry_count={download.retry_count}"
+                )
             if resp.status_code == 429:
                 raise TransientError(
                     f"Too many requests, retry_count={download.retry_count}"