Move feature usage flags into an enum

Author: samwelkanda

msgraphcore/enums.py

@@ -9,6 +9,17 @@ class APIVersion(str, Enum):
     v1 = 'v1.0'
 
 
+class FeatureUsageFlag(int, Enum):
+    """Enumerated list of values used to flag usage of specific middleware"""
+
+    NONE = 0
+    REDIRECT_HANDLER_ENABLED = 1
+    RETRY_HANDLER_ENABLED = 2
+    AUTH_HANDLER_ENABLED = 4
+    DEFAULT_HTTPROVIDER_ENABLED = 8
+    LOGGING_HANDLER_ENABLED = 10
+
+
 class NationalClouds(str, Enum):
     """Enumerated list of supported sovereign clouds"""
 

msgraphcore/middleware/request_context.py

@@ -1,25 +1,19 @@
 import uuid
 
+from msgraphcore.enums import FeatureUsageFlag
 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
+        self._feature_usage = FeatureUsageFlag.NONE
 
     @property
     def feature_usage(self):
         return hex(self._feature_usage)
 
     @feature_usage.setter
-    def set_feature_usage(self, flag):
+    def set_feature_usage(self, flag: FeatureUsageFlag):
         self._feature_usage = self.feature_usage | flag