linting

Author: dandye

ingestion/v1alpha/events_get.py

@@ -41,20 +41,20 @@ def get_event(http_session: requests.AuthorizedSession, proj_id: str,
               proj_instance: str, proj_region: str, event_id: str) -> None:
   """Get event details from Chronicle using the Events Get API.
 
-    Args:
-        http_session: Authorized session for HTTP requests.
-        proj_id: GCP project id or number to which the target instance belongs.
-        proj_instance: Customer ID (uuid with dashes) for the Chronicle instance.
-        proj_region: region in which the target project is located.
-        event_id: The ID of the event to retrieve.
+  Args:
+    http_session: Authorized session for HTTP requests.
+    proj_id: GCP project id or number to which the target instance belongs.
+    proj_instance: Customer ID (uuid with dashes) for the Chronicle instance.
+    proj_region: region in which the target project is located.
+    event_id: The ID of the event to retrieve.
 
-    Raises:
-        requests.exceptions.HTTPError: HTTP request resulted in an error
-            (response.status_code >= 400).
+  Raises:
+    requests.exceptions.HTTPError: HTTP request resulted in an error
+        (response.status_code >= 400).
 
-    Requires the following IAM permission on the parent resource:
-    chronicle.events.get
-    """
+  Requires the following IAM permission on the parent resource:
+  chronicle.events.get
+  """
   # URL encode the event_id in Base64
   encoded_event_id = base64.urlsafe_b64encode(event_id.encode()).decode()
   base_url_with_region = regions.url_always_prepend_region(

sdk/commands/common.py

@@ -1,7 +1,7 @@
 """Common utilities for CLI commands."""
 
-import os
 from functools import wraps
+import os
 
 import click
 from dotenv import load_dotenv

search/v1alpha/udm_events_find.py

@@ -59,7 +59,9 @@ def find_udm_events(http_session: requests.AuthorizedSession,
                     tokens: Optional[List[str]] = None,
                     event_ids: Optional[List[str]] = None,
                     return_unenriched_data: bool = False,
-                    return_all_events_for_log: bool = False) -> None:
+                    return_all_events_for_log: bool = False
+) -> None:
+  # pylint: disable=line-too-long
   """Find UDM events in Chronicle using the Legacy Find UDM Events API.
 
   Args:
@@ -80,12 +82,12 @@ def find_udm_events(http_session: requests.AuthorizedSession,
   Requires the following IAM permission on the parent resource:
   chronicle.events.batchGet
   """
+  # pylint: enable=line-too-long
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
-  # pylint: disable=line-too-long
+  # pylint: disable=line-too-long-next
   instance = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"
   url = f"{base_url_with_region}/v1alpha/{instance}/legacy:legacyFindUdmEvents"
-  # pylint: enable=line-too-long
 
   # Build query parameters
   params = []