Add request context

Author: samwelkanda

msgraphcore/middleware/middleware.py

@@ -1,8 +1,11 @@
 import ssl
+import uuid
 
 from requests.adapters import HTTPAdapter
 from urllib3 import PoolManager
 
+from msgraphcore.middleware.request_context import RequestContext
+
 
 class MiddlewarePipeline(HTTPAdapter):
     """MiddlewarePipeline, entry point of middleware
@@ -22,6 +25,9 @@ def add_middleware(self, middleware):
             self._middleware = middleware
 
     def send(self, request, **kwargs):
+
+        request.context = RequestContext(request)
+
         if self._middleware_present():
             return self._middleware.send(request, **kwargs)
         # No middleware in pipeline, call superclass' send

msgraphcore/middleware/options/middleware_control.py

@@ -1,6 +1,5 @@
 from msgraphcore.constants import AUTH_MIDDLEWARE_OPTIONS
-
-from .auth_middleware_options import AuthMiddlewareOptions
+from msgraphcore.middleware.options.auth_middleware_options import AuthMiddlewareOptions
 
 
 class MiddlewareControl:
@@ -18,10 +17,12 @@ def get_middleware_options(self, func):
 
         def wrapper(*args, **kwargs):
             # Get middleware options from **kwargs
-            scopes = kwargs.pop('scopes', None)
-            if scopes:
-                # Set middleware options, for use by middleware in the middleware pipeline
-                self.set(AUTH_MIDDLEWARE_OPTIONS, AuthMiddlewareOptions(scopes))
+            middleware_options = kwargs.pop('middleware_options', None)
+            if middleware_options:
+                scopes = middleware_options.get('scopes', None)
+                if scopes:
+                    # Set middleware options, for use by middleware in the middleware pipeline
+                    self.set(AUTH_MIDDLEWARE_OPTIONS, AuthMiddlewareOptions(scopes))
             return func(*args, **kwargs)
 
         return wrapper

msgraphcore/middleware/request_context.py

@@ -0,0 +1,25 @@
+import uuid
+
+from msgraphcore.middleware.options.middleware_control import middleware_control
+
+
+class RequestContext:
+    NONE_FLAG = 0
+    REDIRECT_HANDLER_ENABLED_FLAG = 1
+    RETRY_HANDLER_ENABLED_FLAG = 2
+    AUTH_HANDLER_ENABLED_FLAG = 4
+    DEFAULT_HTTPROVIDER_ENABLED_FLAG = 8
+    LOGGING_HANDLER_ENABLED_FLAG = 10
+
+    def __init__(self, request):
+        self.middleware_control = middleware_control
+        self.client_request_id = request.headers.get('client-request-id', str(uuid.uuid4()))
+        self._feature_usage = self.NONE_FLAG
+
+    @property
+    def feature_usage(self):
+        return hex(self._feature_usage)
+
+    @feature_usage.setter
+    def set_feature_usage(self, flag):
+        self._feature_usage = self.feature_usage | flag