Skip to content

Commit b9e7cbf

Browse files
committed
Update feature usage flag in auth handler
1 parent eeeb166 commit b9e7cbf

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

msgraphcore/middleware/authorization.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from msgraphcore.constants import AUTH_MIDDLEWARE_OPTIONS
2+
from msgraphcore.enums import FeatureUsageFlag
23
from msgraphcore.middleware.abc_token_credential import TokenCredential
34
from msgraphcore.middleware.middleware import BaseMiddleware
45
from msgraphcore.middleware.options.middleware_control import middleware_control
@@ -13,6 +14,7 @@ def __init__(self, credential: TokenCredential, **kwargs):
1314

1415
def send(self, request, **kwargs):
1516
request.headers.update({'Authorization': 'Bearer {}'.format(self._get_access_token())})
17+
request.context.set_feature_usage = FeatureUsageFlag.AUTH_HANDLER_ENABLED
1618
response = super().send(request, **kwargs)
1719

1820
# Token might have expired just before transmission, retry the request one more time

0 commit comments

Comments
 (0)