chore: lint fix and formatting.

Author: mihirvala08

src/secops/chronicle/client.py

@@ -4186,7 +4186,8 @@ def list_dashboards(
 
         Returns:
             If as_list is True: List of dashboards.
-            If as_list is False: Dictionary containing list of dashboards and pagination info.
+            If as_list is False: Dictionary containing list of dashboards
+                and pagination info.
 
         Raises:
             APIError: If the API request fails
@@ -4360,7 +4361,7 @@ def duplicate_dashboard(
             dashboard_id: ID of the dashboard to duplicate
             display_name: New name for the duplicated dashboard
             access_type: Access type for the duplicated dashboard
-                        (DashboardAccessType.PRIVATE or DashboardAccessType.PUBLIC)
+                    (DashboardAccessType.PRIVATE or DashboardAccessType.PUBLIC)
             description: Description for the duplicated dashboard
             api_version: Preferred API version to use. Defaults to V1ALPHA
 
@@ -4407,7 +4408,7 @@ def remove_chart(
             self,
             dashboard_id=dashboard_id,
             chart_id=chart_id,
-            api_version=api_version
+            api_version=api_version,
         )
 
     def get_chart(

src/secops/chronicle/dashboard.py

@@ -153,7 +153,8 @@ def import_dashboard(
     """
     if not any(key in dashboard for key in _VALID_DASHBOARD_KEYS):
         raise SecOpsError(
-            f'Dashboard must contain at least one of: {", ".join(_VALID_DASHBOARD_KEYS)}'
+            "Dashboard must contain at least one "
+            f'of: {", ".join(_VALID_DASHBOARD_KEYS)}'
         )
 
     payload = {"source": {"dashboards": [dashboard]}}
@@ -224,7 +225,8 @@ def list_dashboards(
 
     Returns:
         If as_list is True: List of dashboards.
-        If as_list is False: Dictionary containing list of dashboards and pagination info.
+        If as_list is False: Dictionary containing list of dashboards
+            and pagination info.
 
     Raises:
         APIError: If the API request fails
@@ -275,6 +277,7 @@ def get_dashboard(
         error_message=f"Failed to get dashboard with ID {dashboard_id}",
     )
 
+
 def update_dashboard(
     client: "ChronicleClient",
     dashboard_id: str,
@@ -300,7 +303,8 @@ def update_dashboard(
 
     Raises:
         ValueError: If no fields are provided to update
-        APIError: If filters or charts JSON is invalid, or if the API request fails
+        APIError: If filters or charts JSON is invalid,
+            or if the API request fails
     """
     dashboard_id = format_resource_id(dashboard_id)
 
@@ -524,7 +528,9 @@ def add_chart(
         endpoint_path=f"nativeDashboards/{dashboard_id}:addChart",
         api_version=api_version,
         json=payload,
-        error_message=f"Failed to add chart to dashboard with ID {dashboard_id}",
+        error_message=(
+            f"Failed to add chart to dashboard with ID {dashboard_id}"
+        ),
     )
 
 
@@ -585,7 +591,10 @@ def remove_chart(
         endpoint_path=f"nativeDashboards/{dashboard_id}:removeChart",
         api_version=api_version,
         json={"dashboardChart": chart_id},
-        error_message=f"Failed to remove chart with ID {chart_id} from dashboard with ID {dashboard_id}",
+        error_message=(
+            f"Failed to remove chart with ID {chart_id} "
+            f"from dashboard with ID {dashboard_id}"
+        ),
     )
 
 
@@ -674,5 +683,7 @@ def edit_chart(
         endpoint_path=f"nativeDashboards/{dashboard_id}:editChart",
         api_version=api_version,
         json=payload,
-        error_message=f"Failed to edit chart in dashboard with ID {dashboard_id}",
+        error_message=(
+            f"Failed to edit chart in dashboard with ID {dashboard_id}"
+        ),
     )

src/secops/chronicle/utils/format_utils.py

@@ -33,7 +33,8 @@ def format_resource_id(resource_id: str) -> str:
         resource_id: The full resource string or just the ID.
 
     Returns:
-        The extracted ID from the resource string, or the original string if it doesn't match the expected format.
+        The extracted ID from the resource string,
+            or the original string if it doesn't match the expected format.
     """
     if resource_id.startswith("projects/"):
         return resource_id.split("/")[-1]
@@ -46,11 +47,13 @@ def parse_json_list(
     """Parse a JSON string into a list, or return the list as-is.
 
     Args:
-        value: A list of dictionaries or a JSON string representing a list of dictionaries.
+        value: A list of dictionaries or
+            a JSON string representing a list of dictionaries.
         field_name: The name of the field being parsed, used for error messages.
 
     Returns:
-        A list of dictionaries parsed from the JSON string, or the original list if it was already a list.
+        A list of dictionaries parsed from the JSON string,
+            or the original list if it was already a list.
 
     Raises:
         APIError: If the input is a string but cannot be parsed as valid JSON.

tests/chronicle/utils/test_format_utils.py

@@ -17,7 +17,10 @@
 
 import pytest
 
-from secops.chronicle.utils.format_utils import format_resource_id, parse_json_list
+from secops.chronicle.utils.format_utils import (
+    format_resource_id,
+    parse_json_list,
+)
 from secops.exceptions import APIError
 
 
@@ -39,7 +42,10 @@ def test_format_resource_id_handles_minimal_projects_prefix() -> None:
 
 def test_format_resource_id_does_not_alter_non_projects_paths() -> None:
     # Paths that don't start with "projects/" should be returned as-is
-    assert format_resource_id("instances/my-instance/dashboards/abc") == "instances/my-instance/dashboards/abc"
+    assert (
+        format_resource_id("instances/my-instance/dashboards/abc")
+        == "instances/my-instance/dashboards/abc"
+    )
 
 
 def test_format_resource_id_empty_string_returns_empty_string() -> None:
@@ -91,4 +97,4 @@ def test_parse_json_list_handles_empty_json_array() -> None:
 
 def test_parse_json_list_handles_empty_list_input() -> None:
     result = parse_json_list([], "filters")
-    assert result == []
+    assert result == []