Merge pull request #287 from guyzmo/features/dependency_injection

Gauvain Pocentek · web-flow · commit 46b7f488c3dc · 2017-07-09T08:46:41.000+02:00
Added dependency injection support for Session
diff --git a/gitlab/__init__.py b/gitlab/__init__.py
@@ -71,7 +71,7 @@ class Gitlab(object):
 
     def __init__(self, url, private_token=None, email=None, password=None,
                  ssl_verify=True, http_username=None, http_password=None,
-                 timeout=None, api_version='3'):
+                 timeout=None, api_version='3', session=None):
 
         self._api_version = str(api_version)
         self._url = '%s/api/v%s' % (url, api_version)
@@ -90,7 +90,7 @@ def __init__(self, url, private_token=None, email=None, password=None,
         self.http_password = http_password
 
         #: Create a session object for requests
-        self.session = requests.Session()
+        self.session = session or requests.Session()
 
         objects = importlib.import_module('gitlab.v%s.objects' %
                                           self._api_version)
