linting

Author: dandye

detect/v1alpha/create_rule.py

@@ -81,6 +81,7 @@ def create_rule(
   """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
+  # pylint: disable-next=line-too-long
   parent = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"
   url = f"{base_url_with_region}/v1alpha/{parent}/rules"
 

iocs/v1alpha/batch_get_iocs.py

@@ -61,6 +61,7 @@ def batch_get_iocs(
   """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
+  # pylint: disable-next=line-too-long
   instance = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"
   url = f"{base_url_with_region}/v1alpha/{instance}/iocs:batchGet"
 

iocs/v1alpha/get_ioc_state.py

@@ -33,32 +33,32 @@ def get_ioc_state(
 ) -> Mapping[str, Any]:
   """Get the state of an IoC by its value from Chronicle.
 
-    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 instance.
-        proj_region: region in which the target project is located.
-        ioc_value: Value of the IoC to get state for.
-        ioc_type: Type of IoC being requested. One of:
-            IOC_TYPE_UNSPECIFIED
-            DOMAIN
-            IP
-            FILE_HASH
-            URL
-            USER_EMAIL
-            MUTEX
-            FILE_HASH_MD5
-            FILE_HASH_SHA1
-            FILE_HASH_SHA256
-            IOC_TYPE_RESOURCE
+  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 instance.
+    proj_region: region in which the target project is located.
+    ioc_value: Value of the IoC to get state for.
+    ioc_type: Type of IoC being requested. One of:
+        IOC_TYPE_UNSPECIFIED
+        DOMAIN
+        IP
+        FILE_HASH
+        URL
+        USER_EMAIL
+        MUTEX
+        FILE_HASH_MD5
+        FILE_HASH_SHA1
+        FILE_HASH_SHA256
+        IOC_TYPE_RESOURCE
 
-    Returns:
-        Dict containing the IoC state.
+  Returns:
+    Dict containing the IoC state.
 
-    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).
+  """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
   instance = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"

lists/v1alpha/create_list.py

@@ -91,6 +91,7 @@ def create_list(
   """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
+  # pylint: disable-next=line-too-long
   parent = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"
   url = f"{base_url_with_region}/v1alpha/{parent}/referenceLists"
 

sdk/commands/common.py

@@ -1,6 +1,6 @@
 """Common utilities for CLI commands."""
 
-from functools import wraps
+import functools
 import os
 
 import click
@@ -64,7 +64,7 @@ def add_common_options(func):
       required=False,
       help="Path to .env file containing configuration variables.",
   )
-  @wraps(func)
+  @functools.wraps(func)
   def wrapper(*args, **kwargs):
     # Load environment variables from .env file
     env_file = kwargs.pop("env_file", None)

sdk/commands/search.py

@@ -15,8 +15,6 @@
 # limitations under the License.
 """Chronicle Search API commands."""
 
-import json
-
 import click
 from common import chronicle_auth
 
@@ -140,7 +138,8 @@ def find_raw_logs_cmd(ctx, query, batch_tokens, log_ids, regex_search,
     "--tokens",
     multiple=True,
     help=
-    "Optional list of tokens, with each token referring to a group of UDM/Entity events.",
+    "Optional list of tokens, with each token referring to a group of "
+    "UDM/Entity events.",
 )
 @click.option(
     "--event-ids",

search/v1alpha/search_query_get.py

@@ -41,21 +41,21 @@ def get_search_query(http_session: requests.AuthorizedSession, proj_id: str,
                      query_id: str) -> None:
   """Get a search query by ID from Chronicle.
 
-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 instance.
-  proj_region: region in which the target project is located.
-  user_id: ID of the user who owns the search query.
-  query_id: ID of the search query 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 instance.
+    proj_region: region in which the target project is located.
+    user_id: ID of the user who owns the search query.
+    query_id: ID of the search query 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 instance resource:
-chronicle.searchQueries.get
-"""
+  Requires the following IAM permission on the instance resource:
+  chronicle.searchQueries.get
+  """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
   # pylint: disable=line-too-long