URLFile: exception type for failed requests (commaai#30330)

URLFileException for URLFile request errors
deanlee · Oct 25, 2023 · c27e977 · c27e977
1 parent 4094d58
commit c27e977
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tools/lib/url_file.py b/tools/lib/url_file.py
@@ -17,6 +17,10 @@ def hash_256(link):
   return hsh
 
 
+class URLFileException(Exception):
+  pass
+
+
 class URLFile:
   _tlocal = threading.local()
 
@@ -158,11 +162,11 @@ def test(debug_type, debug_msg):
 
     response_code = c.getinfo(pycurl.RESPONSE_CODE)
     if response_code == 416:  # Requested Range Not Satisfiable
-      raise Exception(f"Error, range out of bounds {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
+      raise URLFileException(f"Error, range out of bounds {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
     if download_range and response_code != 206:  # Partial Content
-      raise Exception(f"Error, requested range but got unexpected response {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
+      raise URLFileException(f"Error, requested range but got unexpected response {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
     if (not download_range) and response_code != 200:  # OK
-      raise Exception(f"Error {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
+      raise URLFileException(f"Error {response_code} {headers} ({self._url}): {repr(dats.getvalue())[:500]}")
 
     ret = dats.getvalue()
     self._pos += len(ret)