adjust

Author: AllyW

src/application-insights/azext_applicationinsights/_client_factory.py

@@ -3,28 +3,37 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
+class _Track1Credential:
+
+    def __init__(self, credential, resource):
+        """Track 1 credential that can be fed into Track 1 SDK clients. Exposes signed_session protocol.
+        :param credential: Track 2 credential that exposes get_token protocol
+        :param resource: AAD resource
+        """
+        self._credential = credential
+        self._resource = resource
+
+    def signed_session(self, session=None):
+        import requests
+        from azure.cli.core.auth.util import resource_to_scopes
+        session = session or requests.Session()
+        token = self._credential.get_token(*resource_to_scopes(self._resource))
+        header = "{} {}".format('Bearer', token.token)
+        session.headers['Authorization'] = header
+        return session
+
 
 def applicationinsights_data_plane_client(cli_ctx, _, subscription=None):
     """Initialize Log Analytics data client for use with CLI."""
     from .vendored_sdks.applicationinsights import ApplicationInsightsDataClient
     from azure.cli.core._profile import Profile
-    from .util import Track1Credential
     profile = Profile(cli_ctx=cli_ctx)
     # Note: temporarily adapt track2 auth to track1 by the guidance:
     # https://github.com/Azure/azure-cli/pull/29631#issuecomment-2716799520
     # need to be removed after migrated by codegen
-    # cred, _, _ = profile.get_login_credentials(
-    #     resource=cli_ctx.cloud.endpoints.app_insights_resource_id,
-    #     subscription_id=subscription
-    # )
-    # return ApplicationInsightsDataClient(
-    #     cred,
-    #     base_url=f'{cli_ctx.cloud.endpoints.app_insights_resource_id}/v1'
-    # )
-
     cred, _, _ = profile.get_login_credentials(subscription_id=subscription)
     return ApplicationInsightsDataClient(
-        Track1Credential(cred, cli_ctx.cloud.endpoints.app_insights_resource_id),
+        _Track1Credential(cred, cli_ctx.cloud.endpoints.app_insights_resource_id),
         base_url=f'{cli_ctx.cloud.endpoints.app_insights_resource_id}/v1'
     )