isort imports

dandye · dandye · commit e64d3b013c5c · 2025-03-07T18:09:02.000-05:00
diff --git a/detect/v1alpha/batch_update_curated_rule_set_deployments.py b/detect/v1alpha/batch_update_curated_rule_set_deployments.py
@@ -42,12 +42,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/bulk_update_alerts.py b/detect/v1alpha/bulk_update_alerts.py
@@ -43,12 +43,11 @@
 import argparse
 import json
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 from . import update_alert
 
diff --git a/detect/v1alpha/create_retrohunt.py b/detect/v1alpha/create_retrohunt.py
@@ -37,13 +37,12 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import datetime_converter
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/create_rule.py b/detect/v1alpha/create_rule.py
@@ -41,12 +41,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/enable_rule.py b/detect/v1alpha/enable_rule.py
@@ -34,12 +34,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/get_alert.py b/detect/v1alpha/get_alert.py
@@ -33,12 +33,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/get_detection.py b/detect/v1alpha/get_detection.py
@@ -33,12 +33,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/get_retrohunt.py b/detect/v1alpha/get_retrohunt.py
@@ -43,12 +43,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/get_rule.py b/detect/v1alpha/get_rule.py
@@ -27,12 +27,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/list_detections.py b/detect/v1alpha/list_detections.py
@@ -44,6 +44,7 @@
 import argparse
 import json
 from typing import Any, Mapping
+
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
diff --git a/detect/v1alpha/list_errors.py b/detect/v1alpha/list_errors.py
@@ -36,6 +36,7 @@
 import argparse
 import json
 from typing import Any, Mapping
+
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
diff --git a/detect/v1alpha/list_rules.py b/detect/v1alpha/list_rules.py
@@ -33,12 +33,11 @@
 import json
 from typing import Any, Mapping
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/detect/v1alpha/update_alert.py b/detect/v1alpha/update_alert.py
@@ -55,7 +55,6 @@
 from common import project_id
 from common import project_instance
 from common import regions
-
 from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
diff --git a/detect/v1alpha/update_rule.py b/detect/v1alpha/update_rule.py
@@ -35,6 +35,7 @@
 import argparse
 import json
 from typing import Any, Mapping
+
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
diff --git a/ingestion/v1alpha/create_udm_events.py b/ingestion/v1alpha/create_udm_events.py
@@ -32,12 +32,11 @@
 import argparse
 import json
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/ingestion/v1alpha/event_import.py b/ingestion/v1alpha/event_import.py
@@ -25,12 +25,11 @@
 import argparse
 import json
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/ingestion/v1alpha/events_batch_get.py b/ingestion/v1alpha/events_batch_get.py
@@ -26,12 +26,11 @@
 import base64
 import json
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/ingestion/v1alpha/events_get.py b/ingestion/v1alpha/events_get.py
@@ -25,12 +25,11 @@
 import argparse
 import base64
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/ingestion/v1alpha/get_udm_event.py b/ingestion/v1alpha/get_udm_event.py
@@ -25,12 +25,11 @@
 import argparse
 import json
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 SCOPES = [
     "https://www.googleapis.com/auth/cloud-platform",
diff --git a/lists/v1alpha/create_list.py b/lists/v1alpha/create_list.py
@@ -35,12 +35,11 @@
 import json
 from typing import Any, Dict, Optional, Sequence
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/lists/v1alpha/get_list.py b/lists/v1alpha/get_list.py
@@ -36,7 +36,6 @@
 from common import project_id
 from common import project_instance
 from common import regions
-
 from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
diff --git a/sdk/commands/common.py b/sdk/commands/common.py
@@ -8,12 +8,31 @@
 
 
 def get_env_value(key, default=None):
-  """Get value from environment variable with Chronicle prefix."""
+  """Gets value from environment variable with Chronicle prefix.
+
+  Args:
+    key: The environment variable key (without CHRONICLE_ prefix).
+    default: Default value if environment variable is not set.
+
+  Returns:
+    The value of the environment variable or the default value.
+  """
   return os.getenv(f"CHRONICLE_{key}", default)
 
 
 def add_common_options(func):
-  """Add common options to a command."""
+  """Adds common CLI options to a command.
+
+  Adds standard Chronicle CLI options for region, project instance, project ID,
+  credentials file, and environment file. Values can be provided via command line
+  arguments or environment variables.
+
+  Args:
+    func: The function to wrap with common options.
+
+  Returns:
+    A decorated function that includes common Chronicle CLI options.
+  """
 
   # Add CLI options first
   @click.option(
@@ -49,7 +68,7 @@ def wrapper(*args, **kwargs):
       load_dotenv(env_file)
     else:
       # Look for .env in the current working directory
-      cwd_env = os.path.join(os.getcwd(), '.env')
+      cwd_env = os.path.join(os.getcwd(), ".env")
       load_dotenv(cwd_env)
 
     # Now validate required options
diff --git a/search/v1alpha/asset_events_find.py b/search/v1alpha/asset_events_find.py
@@ -23,16 +23,16 @@
 # pylint: enable=line-too-long
 
 import argparse
+from datetime import datetime
+from datetime import timezone
 import json
-from datetime import datetime, timezone
 from typing import Optional
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/search/v1alpha/raw_logs_find.py b/search/v1alpha/raw_logs_find.py
@@ -26,12 +26,11 @@
 import json
 from typing import List, Optional
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
diff --git a/search/v1alpha/udm_events_find.py b/search/v1alpha/udm_events_find.py
@@ -40,12 +40,11 @@
 import json
 from typing import List, Optional
 
-from google.auth.transport import requests
-
 from common import chronicle_auth
 from common import project_id
 from common import project_instance
 from common import regions
+from google.auth.transport import requests
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 SCOPES = [
