No public description

feeds/fetch_feed_service_account.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+"""Executable and reusable sample for retrieving a feed."""
+
+import argparse
+import json
+from typing import Any, Mapping
+
+from common import chronicle_auth
+from common import regions
+from google.auth.transport import requests
+
+CHRONICLE_API_BASE_URL = "https://backstory.googleapis.com"
+
+
+def fetch_feed_service_account(
+    http_session: requests.AuthorizedSession,
+) -> Mapping[str, Any]:
+  """Retrieves chronicle service account used to fetch data form GCS via feeds.
+
+  Args:
+    http_session: Authorized session for HTTP requests.
+
+  Returns:
+    Chronicle's service_account used to pull GCS data via feed for the customer.
+
+  Raises:
+    requests.exceptions.HTTPError: HTTP request resulted in an error
+      (response.status_code >= 400).
+  """
+  url = f"{CHRONICLE_API_BASE_URL}/v1/fetchFeedServiceAccount"
+
+  response = http_session.request("GET", url)
+
+  if response.status_code >= 400:
+    print(response.text)
+  response.raise_for_status()
+  return response.json()
+
+
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser()
+  chronicle_auth.add_argument_credentials_file(parser)
+  regions.add_argument_region(parser)
+
+  args = parser.parse_args()
+  CHRONICLE_API_BASE_URL = regions.url(CHRONICLE_API_BASE_URL, args.region)
+  session = chronicle_auth.initialize_http_session(args.credentials_file)
+  print(json.dumps(fetch_feed_service_account(session), indent=2))