Skip to content

Commit 3e2360a

Browse files
Create private method for updating sdkVersion
Created a private method _append_sdk_version to update sdkVersion in headers
1 parent 4ab46dc commit 3e2360a

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

msgraphcore/graph_session.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,7 @@ class GraphSession(Session):
2020
"""
2121
def __init__(self, auth_provider: AuthProviderBase, middleware: list = []):
2222
super().__init__()
23-
if 'sdkVersion' not in self.headers:
24-
self.headers.update({'sdkVersion': 'graph-python-' + SDK_VERSION})
25-
else:
26-
self.headers.update({'sdkVersion': 'graph-python-' + SDK_VERSION + ', '
27-
+ self.headers.get('sdkVersion')})
23+
self._append_sdk_version()
2824
self._base_url = BASE_URL
2925

3026
auth_handler = AuthorizationHandler(auth_provider)
@@ -110,3 +106,12 @@ def _register(self, middleware: [BaseMiddleware]) -> None:
110106
middleware_pipeline.add_middleware(ware)
111107

112108
self.mount('https://', middleware_pipeline)
109+
110+
def _append_sdk_version(self) -> None:
111+
"""Updates sdkVersion in headers with comma-separated new values
112+
"""
113+
if 'sdkVersion' in self.headers:
114+
self.headers.update({'sdkVersion': 'graph-python-' + SDK_VERSION + ', '
115+
+ self.headers.get('sdkVersion')})
116+
else:
117+
self.headers.update({'sdkVersion': 'graph-python-' + SDK_VERSION})

0 commit comments

Comments
 (0)