Added custom session option for proxies (#43)

* option to add custom session to NewsApiClient

* session only if requested

Author: rickykim93

newsapi/newsapi_client.py

@@ -7,8 +7,12 @@
 
 class NewsApiClient(object):
 
-    def __init__(self, api_key):
+    def __init__(self, api_key, session=None):
         self.auth = NewsApiAuth(api_key=api_key)
+        if session is None:
+            self.request_method = requests
+        else:
+            self.request_method = session
 
     def get_top_headlines(self, q=None, sources=None, language='en', country=None, category=None, page_size=None,
                           page=None):
@@ -113,7 +117,7 @@ def get_top_headlines(self, q=None, sources=None, language='en', country=None, c
                 raise TypeError('page param should be an int')
 
         # Send Request
-        r = requests.get(const.TOP_HEADLINES_URL, auth=self.auth, timeout=30, params=payload)
+        r = self.request_method.get(const.TOP_HEADLINES_URL, auth=self.auth, timeout=30, params=payload)
 
         # Check Status of Request
         if r.status_code != requests.codes.ok:
@@ -255,7 +259,7 @@ def get_everything(self, q=None, sources=None, domains=None, exclude_domains=Non
                 raise TypeError('page param should be an int')
 
         # Send Request
-        r = requests.get(const.EVERYTHING_URL, auth=self.auth, timeout=30, params=payload)
+        r = self.request_method.get(const.EVERYTHING_URL, auth=self.auth, timeout=30, params=payload)
 
         # Check Status of Request
         if r.status_code != requests.codes.ok:
@@ -318,7 +322,7 @@ def get_sources(self, category=None, language=None, country=None):
                 raise TypeError('category param should be of type str')
 
         # Send Request
-        r = requests.get(const.SOURCES_URL, auth=self.auth, timeout=30, params=payload)
+        r = self.request_method.get(const.SOURCES_URL, auth=self.auth, timeout=30, params=payload)
 
         # Check Status of Request
         if r.status_code != requests.codes.ok: