Merge pull request #31 from Azure/hooksmsrest

Add requests hooks

Author: lmazuel

msrest/configuration.py

@@ -74,6 +74,11 @@ def __init__(self, base_url, filepath=None):
             requests.__version__,
             msrest_version)
 
+        # Requests hooks. Must respect requests hook callback signature
+        # Note that we will inject the following parameters:
+        # - kwargs['msrest']['session'] with the current session
+        self.hooks = []
+
         self._config = configparser.ConfigParser()
         self._config.optionxform = str
 

msrest/service_client.py

@@ -131,6 +131,15 @@ def log_hook(r, *args, **kwargs):
             log_response(None, r.request, r, result=r)
         session.hooks['response'].append(log_hook)
 
+        def make_user_hook_cb(user_hook, session):
+            def user_hook_cb(r, *args, **kwargs):
+                kwargs.setdefault("msrest", {})['session'] = session
+                return user_hook(r, *args, **kwargs)
+            return user_hook_cb
+
+        for user_hook in self.config.hooks:
+            session.hooks['response'].append(make_user_hook_cb(user_hook, session))
+
         max_retries = config.get(
             'retries', self.config.retry_policy())
         for protocol in self._protocols: