4
4
from requests import Session
5
5
6
6
from msgraphcore .constants import BASE_URL , SDK_VERSION
7
- from msgraphcore .middleware .middleware import MiddlewarePipeline , BaseMiddleware
8
7
from msgraphcore .middleware .abc_token_credential import TokenCredential
9
8
from msgraphcore .middleware .authorization import AuthorizationHandler
9
+ from msgraphcore .middleware .middleware import BaseMiddleware , MiddlewarePipeline
10
10
from msgraphcore .middleware .options .middleware_control import middleware_control
11
11
12
12
@@ -15,12 +15,13 @@ class GraphSession(Session):
15
15
16
16
Extends Session by adding support for middleware options and middleware pipeline
17
17
"""
18
- def __init__ (self ,
19
- credential : TokenCredential ,
20
- scopes : [str ] = ['.default' ],
21
- middleware : list = [],
22
- api_version : str = 'v1.0'
23
- ):
18
+ def __init__ (
19
+ self ,
20
+ credential : TokenCredential ,
21
+ scopes : [str ] = ['.default' ],
22
+ middleware : list = [],
23
+ api_version : str = 'v1.0'
24
+ ):
24
25
super ().__init__ ()
25
26
self ._append_sdk_version ()
26
27
self ._base_url = BASE_URL + '/' + api_version
@@ -94,7 +95,7 @@ def _graph_url(self, url: str) -> str:
94
95
:param url: user provided path
95
96
:return: graph_url
96
97
"""
97
- return self ._base_url + url if (url [0 ] == '/' ) else url
98
+ return self ._base_url + url if (url [0 ] == '/' ) else url
98
99
99
100
def _register (self , middleware : [BaseMiddleware ]) -> None :
100
101
"""Adds middleware to middleware_pipeline
@@ -113,7 +114,11 @@ def _append_sdk_version(self) -> None:
113
114
"""Updates sdkVersion in headers with comma-separated new values
114
115
"""
115
116
if 'sdkVersion' in self .headers :
116
- self .headers .update ({'sdkVersion' : 'graph-python-' + SDK_VERSION + ', '
117
- + self .headers .get ('sdkVersion' )})
117
+ self .headers .update (
118
+ {
119
+ 'sdkVersion' :
120
+ 'graph-python-' + SDK_VERSION + ', ' + self .headers .get ('sdkVersion' )
121
+ }
122
+ )
118
123
else :
119
124
self .headers .update ({'sdkVersion' : 'graph-python-' + SDK_VERSION })
0 commit comments