From f4dcc41a08e0652ed829f3929a4d87b98e0ebfd7 Mon Sep 17 00:00:00 2001 From: Leron Gray Date: Wed, 17 Nov 2021 10:42:56 -0800 Subject: [PATCH] fix china, usgov endpoints --- stormcollector/stormcollector/auth.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/stormcollector/stormcollector/auth.py b/stormcollector/stormcollector/auth.py index 0600d02..6527b86 100644 --- a/stormcollector/stormcollector/auth.py +++ b/stormcollector/stormcollector/auth.py @@ -24,14 +24,14 @@ "MGMT": "https://management.core.cloudapi.de", }, "CHINA": { - "AD": identity.KnownAuthorities.AZURE_PUBLIC_CLOUD, + "AD": identity.KnownAuthorities.AZURE_CHINA, "AAD": "https://graph.chinacloudapi.cn", "ARM": "https://management.chinacloudapi.cn", "GRAPH": "https://microsoftgraph.chinacloudapi.cn", "MGMT": "https://management.core.chinacloudapi.cn", }, "USGOV": { - "AD": identity.KnownAuthorities.AZURE_PUBLIC_CLOUD, + "AD": identity.KnownAuthorities.AZURE_GOVERNMENT, "AAD": "https://graph.windows.net", "ARM": "https://management.usgovcloudapi.net", "GRAPH": "https://graph.microsoft.us/", @@ -42,7 +42,10 @@ class Context: def __init__( - self, args: Namespace, cloud: dict, authenticatedCreds: Tuple[Any], + self, + args: Namespace, + cloud: dict, + authenticatedCreds: Tuple[Any], ): self.args = args self.cloud = cloud