Skip to content

Commit 5611cd7

Browse files
committed
Simplified initialization of GraphServiceClient
1 parent 15c97c3 commit 5611cd7

File tree

2 files changed

+10
-22
lines changed

2 files changed

+10
-22
lines changed

app-auth/graphapponlytutorial/graph.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,12 @@
44
# <AppAuthConfigSnippet>
55
from configparser import SectionProxy
66
from azure.identity.aio import ClientSecretCredential
7-
from kiota_authentication_azure.azure_identity_authentication_provider import (
8-
AzureIdentityAuthenticationProvider
9-
)
10-
from msgraph import GraphRequestAdapter, GraphServiceClient
7+
from msgraph import GraphServiceClient
118
from msgraph.generated.users.users_request_builder import UsersRequestBuilder
129

1310
class Graph:
1411
settings: SectionProxy
1512
client_credential: ClientSecretCredential
16-
adapter: GraphRequestAdapter
1713
app_client: GraphServiceClient
1814

1915
def __init__(self, config: SectionProxy):
@@ -23,9 +19,7 @@ def __init__(self, config: SectionProxy):
2319
client_secret = self.settings['clientSecret']
2420

2521
self.client_credential = ClientSecretCredential(tenant_id, client_id, client_secret)
26-
auth_provider = AzureIdentityAuthenticationProvider(self.client_credential) # type: ignore
27-
self.adapter = GraphRequestAdapter(auth_provider)
28-
self.app_client = GraphServiceClient(self.adapter)
22+
self.app_client = GraphServiceClient(self.client_credential) # type: ignore
2923
# </AppAuthConfigSnippet>
3024

3125
# <GetAppOnlyTokenSnippet>

user-auth/graphtutorial/graph.py

Lines changed: 8 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,11 @@
44
# <UserAuthConfigSnippet>
55
from configparser import SectionProxy
66
from azure.identity import DeviceCodeCredential
7-
from kiota_authentication_azure.azure_identity_authentication_provider import (
8-
AzureIdentityAuthenticationProvider)
9-
from msgraph import GraphRequestAdapter, GraphServiceClient
10-
from msgraph.generated.me.me_request_builder import MeRequestBuilder
11-
from msgraph.generated.me.mail_folders.item.messages.messages_request_builder import (
7+
from msgraph import GraphServiceClient
8+
from msgraph.generated.users.item.user_item_request_builder import UserItemRequestBuilder
9+
from msgraph.generated.users.item.mail_folders.item.messages.messages_request_builder import (
1210
MessagesRequestBuilder)
13-
from msgraph.generated.me.send_mail.send_mail_post_request_body import SendMailPostRequestBody
11+
from msgraph.generated.users.item.send_mail.send_mail_post_request_body import SendMailPostRequestBody
1412
from msgraph.generated.models.message import Message
1513
from msgraph.generated.models.item_body import ItemBody
1614
from msgraph.generated.models.body_type import BodyType
@@ -20,7 +18,6 @@
2018
class Graph:
2119
settings: SectionProxy
2220
device_code_credential: DeviceCodeCredential
23-
adapter: GraphRequestAdapter
2421
user_client: GraphServiceClient
2522

2623
def __init__(self, config: SectionProxy):
@@ -30,11 +27,7 @@ def __init__(self, config: SectionProxy):
3027
graph_scopes = self.settings['graphUserScopes'].split(' ')
3128

3229
self.device_code_credential = DeviceCodeCredential(client_id, tenant_id = tenant_id)
33-
auth_provider = AzureIdentityAuthenticationProvider(
34-
self.device_code_credential,
35-
scopes=graph_scopes)
36-
self.adapter = GraphRequestAdapter(auth_provider)
37-
self.user_client = GraphServiceClient(self.adapter)
30+
self.user_client = GraphServiceClient(self.device_code_credential, graph_scopes)
3831
# </UserAuthConfigSnippet>
3932

4033
# <GetUserTokenSnippet>
@@ -47,10 +40,11 @@ async def get_user_token(self):
4740
# <GetUserSnippet>
4841
async def get_user(self):
4942
# Only request specific properties using $select
50-
query_params = MeRequestBuilder.MeRequestBuilderGetQueryParameters(
43+
query_params = UserItemRequestBuilder.UserItemRequestBuilderGetQueryParameters(
5144
select=['displayName', 'mail', 'userPrincipalName']
5245
)
53-
request_config = MeRequestBuilder.MeRequestBuilderGetRequestConfiguration(
46+
47+
request_config = UserItemRequestBuilder.UserItemRequestBuilderGetRequestConfiguration(
5448
query_parameters=query_params
5549
)
5650

0 commit comments

Comments
 (0)