ADD URL to exception

Author: mfeurer

openml/_api_calls.py

@@ -95,7 +95,7 @@ def _read_url_files(url, data=None, file_dictionary=None, file_elements=None):
     # 'gzip,deflate'
     response = requests.post(url, data=data, files=file_elements)
     if response.status_code != 200:
-        raise _parse_server_exception(response)
+        raise _parse_server_exception(response, url=url)
     if 'Content-Encoding' not in response.headers or \
             response.headers['Content-Encoding'] != 'gzip':
         warnings.warn('Received uncompressed content from OpenML for %s.' % url)
@@ -117,14 +117,14 @@ def _read_url(url, data=None):
         response = requests.post(url, data=data)
 
     if response.status_code != 200:
-        raise _parse_server_exception(response)
+        raise _parse_server_exception(response, url=url)
     if 'Content-Encoding' not in response.headers or \
             response.headers['Content-Encoding'] != 'gzip':
         warnings.warn('Received uncompressed content from OpenML for %s.' % url)
     return response.text
 
 
-def _parse_server_exception(response):
+def _parse_server_exception(response, url=None):
     # OpenML has a sopisticated error system
     # where information about failures is provided. try to parse this
     try:
@@ -143,4 +143,9 @@ def _parse_server_exception(response):
         # 512 for runs, 370 for datasets (should be 372), 500 for flows
         # 482 for tasks
         return OpenMLServerNoResult(code, message, additional)
-    return OpenMLServerException(code, message, additional)
+    return OpenMLServerException(
+        code=code,
+        message=message,
+        additional=additional,
+        url=url
+    )

openml/exceptions.py

@@ -16,9 +16,13 @@ class OpenMLServerException(OpenMLServerError):
     """exception for when the result of the server was
        not 200 (e.g., listing call w/o results). """
 
-    def __init__(self, code, message, additional=None):
+    # Code needs to be optional to allow the exceptino to be picklable:
+    # https://stackoverflow.com/questions/16244923/how-to-make-a-custom-exception-class-with-multiple-init-args-pickleable
+    def __init__(self, message, code=None, additional=None, url=None):
+        self.message = message
         self.code = code
         self.additional = additional
+        self.url = url
         super(OpenMLServerException, self).__init__(message)