add user defined headers to webservice

xlcnd · Jul 7, 2014 · 341b4d1 · 341b4d1
1 parent c74342a
commit 341b4d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/isbnlib/dev/webservice.py b/isbnlib/dev/webservice.py
@@ -21,11 +21,14 @@ class WEBService(object):
 
     """Class to query web services."""
 
-    def __init__(self, url, user_agent=UA, values=None):
+    def __init__(self, url, user_agent=UA, values=None, appheaders=None):
         """Initialize main properties."""
         self._url = url
         # headers to accept gzipped content
         headers = {'Accept-Encoding': 'gzip', 'User-Agent': user_agent}
+        # add more user provided headers
+        if appheaders is not None:  # pragma: no cover
+            headers.update(appheaders)
         # if 'data' it does a PUT request (data must be urlencoded)
         data = urlencode(values) if values else None
         self._request = Request(url, data, headers=headers)