athena-intelligence 0.1.227__py3-none-any.whl → 0.1.357__py3-none-any.whl

athena/tools/sheets/raw_client.py

@@ -10,10 +10,13 @@ from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
 from ...core.serialization import convert_and_respect_annotation_metadata
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.cell_format import CellFormat
 from ...types.create_new_sheet_tab_response import CreateNewSheetTabResponse
 from ...types.get_table_response import GetTableResponse
+from ...types.sheet import Sheet
 from ...types.sheet_operation_response import SheetOperationResponse
-from .types.insert_table_row_request_row_data_item_value import InsertTableRowRequestRowDataItemValue
+from ...types.table_row_data import TableRowData
+from .types.update_sheet_range_request_values_item_item import UpdateSheetRangeRequestValuesItemItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -504,10 +507,9 @@ class RawSheetsClient:
         asset_id: str,
         end_column: int,
         end_row: int,
+        formatting: CellFormat,
         start_column: int,
         start_row: int,
-        type: str,
-        value_json: str,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
@@ -525,18 +527,15 @@ class RawSheetsClient:
         end_row : int
             1-based ending row index
 
+        formatting : CellFormat
+            Cell format
+
         start_column : int
             1-based starting column index
 
         start_row : int
             1-based starting row index
 
-        type : str
-            Formatting type (e.g. backgroundColor, textFormat, borders, etc.)
-
-        value_json : str
-            JSON string for value param (e.g. '"#FF0000"' or '{"fontSize":12}')
-
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
 
@@ -555,11 +554,12 @@ class RawSheetsClient:
                 "asset_id": asset_id,
                 "end_column": end_column,
                 "end_row": end_row,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting, annotation=CellFormat, direction="write"
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "type": type,
-                "value_json": value_json,
             },
             headers={
                 "content-type": "application/json",
@@ -599,7 +599,8 @@ class RawSheetsClient:
         asset_id: str,
         start_column: int,
         start_row: int,
-        values: typing.Sequence[str],
+        values: typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+        formatting: typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
@@ -617,8 +618,11 @@ class RawSheetsClient:
         start_row : int
             1-based starting row index
 
-        values : typing.Sequence[str]
-            List of pipe-separated strings representing rows (e.g. ['A1|B1|C1', 'A2|B2|C2'])
+        values : typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]]
+            2D list of cells for each row
+
+        formatting : typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]]
+            Optional 2D list of cell formats matching the structure of values. Each row is a list of CellFormat objects for each cell in that row. Use None for cells without formatting. numberFormat is not required unless user explicity asked to change
 
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
@@ -636,10 +640,19 @@ class RawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]],
+                    direction="write",
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "values": values,
+                "values": convert_and_respect_annotation_metadata(
+                    object_=values,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+                    direction="write",
+                ),
             },
             headers={
                 "content-type": "application/json",
@@ -819,14 +832,7 @@ class RawSheetsClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     def create_tab(
-        self,
-        *,
-        asset_id: str,
-        sheet_id: int,
-        title: str,
-        active_sheet_id: typing.Optional[int] = OMIT,
-        tab_color: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, asset_id: str, sheet: Sheet, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[CreateNewSheetTabResponse]:
         """
         Create a new tab in an Athena spreadsheet.
@@ -836,17 +842,8 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        sheet_id : int
-            Sheet ID of the new tab
-
-        title : str
-            Title of the new tab
-
-        active_sheet_id : typing.Optional[int]
-            Currently active sheet ID
-
-        tab_color : typing.Optional[str]
-            Optional color of the new tab
+        sheet : Sheet
+            Sheet Specification
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -860,11 +857,8 @@ class RawSheetsClient:
             "api/v0/tools/sheets/tab/create",
             method="POST",
             json={
-                "active_sheet_id": active_sheet_id,
                 "asset_id": asset_id,
-                "sheet_id": sheet_id,
-                "tab_color": tab_color,
-                "title": title,
+                "sheet": convert_and_respect_annotation_metadata(object_=sheet, annotation=Sheet, direction="write"),
             },
             headers={
                 "content-type": "application/json",
@@ -1222,7 +1216,7 @@ class RawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        row_data: typing.Sequence[TableRowData],
         table_name: str,
         table_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -1235,7 +1229,7 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
+        row_data : typing.Sequence[TableRowData]
             Array of row objects where keys are column names and values are cell values
 
         table_name : str
@@ -1257,13 +1251,7 @@ class RawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
-                "row_data": convert_and_respect_annotation_metadata(
-                    object_=row_data,
-                    annotation=typing.Sequence[
-                        typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]
-                    ],
-                    direction="write",
-                ),
+                "row_data": row_data,
                 "table_id": table_id,
                 "table_name": table_name,
             },
@@ -1880,10 +1868,9 @@ class AsyncRawSheetsClient:
         asset_id: str,
         end_column: int,
         end_row: int,
+        formatting: CellFormat,
         start_column: int,
         start_row: int,
-        type: str,
-        value_json: str,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
@@ -1901,18 +1888,15 @@ class AsyncRawSheetsClient:
         end_row : int
             1-based ending row index
 
+        formatting : CellFormat
+            Cell format
+
         start_column : int
             1-based starting column index
 
         start_row : int
             1-based starting row index
 
-        type : str
-            Formatting type (e.g. backgroundColor, textFormat, borders, etc.)
-
-        value_json : str
-            JSON string for value param (e.g. '"#FF0000"' or '{"fontSize":12}')
-
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
 
@@ -1931,11 +1915,12 @@ class AsyncRawSheetsClient:
                 "asset_id": asset_id,
                 "end_column": end_column,
                 "end_row": end_row,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting, annotation=CellFormat, direction="write"
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "type": type,
-                "value_json": value_json,
             },
             headers={
                 "content-type": "application/json",
@@ -1975,7 +1960,8 @@ class AsyncRawSheetsClient:
         asset_id: str,
         start_column: int,
         start_row: int,
-        values: typing.Sequence[str],
+        values: typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+        formatting: typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
@@ -1993,8 +1979,11 @@ class AsyncRawSheetsClient:
         start_row : int
             1-based starting row index
 
-        values : typing.Sequence[str]
-            List of pipe-separated strings representing rows (e.g. ['A1|B1|C1', 'A2|B2|C2'])
+        values : typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]]
+            2D list of cells for each row
+
+        formatting : typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]]
+            Optional 2D list of cell formats matching the structure of values. Each row is a list of CellFormat objects for each cell in that row. Use None for cells without formatting. numberFormat is not required unless user explicity asked to change
 
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
@@ -2012,10 +2001,19 @@ class AsyncRawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]],
+                    direction="write",
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "values": values,
+                "values": convert_and_respect_annotation_metadata(
+                    object_=values,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+                    direction="write",
+                ),
             },
             headers={
                 "content-type": "application/json",
@@ -2195,14 +2193,7 @@ class AsyncRawSheetsClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def create_tab(
-        self,
-        *,
-        asset_id: str,
-        sheet_id: int,
-        title: str,
-        active_sheet_id: typing.Optional[int] = OMIT,
-        tab_color: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, asset_id: str, sheet: Sheet, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[CreateNewSheetTabResponse]:
         """
         Create a new tab in an Athena spreadsheet.
@@ -2212,17 +2203,8 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        sheet_id : int
-            Sheet ID of the new tab
-
-        title : str
-            Title of the new tab
-
-        active_sheet_id : typing.Optional[int]
-            Currently active sheet ID
-
-        tab_color : typing.Optional[str]
-            Optional color of the new tab
+        sheet : Sheet
+            Sheet Specification
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -2236,11 +2218,8 @@ class AsyncRawSheetsClient:
             "api/v0/tools/sheets/tab/create",
             method="POST",
             json={
-                "active_sheet_id": active_sheet_id,
                 "asset_id": asset_id,
-                "sheet_id": sheet_id,
-                "tab_color": tab_color,
-                "title": title,
+                "sheet": convert_and_respect_annotation_metadata(object_=sheet, annotation=Sheet, direction="write"),
             },
             headers={
                 "content-type": "application/json",
@@ -2598,7 +2577,7 @@ class AsyncRawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        row_data: typing.Sequence[TableRowData],
         table_name: str,
         table_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -2611,7 +2590,7 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
+        row_data : typing.Sequence[TableRowData]
             Array of row objects where keys are column names and values are cell values
 
         table_name : str
@@ -2633,13 +2612,7 @@ class AsyncRawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
-                "row_data": convert_and_respect_annotation_metadata(
-                    object_=row_data,
-                    annotation=typing.Sequence[
-                        typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]
-                    ],
-                    direction="write",
-                ),
+                "row_data": row_data,
                 "table_id": table_id,
                 "table_name": table_name,
             },

athena/tools/sheets/types/__init__.py

@@ -2,6 +2,35 @@
 
 # isort: skip_file
 
-from .insert_table_row_request_row_data_item_value import InsertTableRowRequestRowDataItemValue
+import typing
+from importlib import import_module
 
-__all__ = ["InsertTableRowRequestRowDataItemValue"]
+if typing.TYPE_CHECKING:
+    from .update_sheet_range_request_values_item_item import UpdateSheetRangeRequestValuesItemItem
+_dynamic_imports: typing.Dict[str, str] = {
+    "UpdateSheetRangeRequestValuesItemItem": ".update_sheet_range_request_values_item_item"
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["UpdateSheetRangeRequestValuesItemItem"]

athena/tools/sheets/types/{insert_table_row_request_row_data_item_value.py → update_sheet_range_request_values_item_item.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-InsertTableRowRequestRowDataItemValue = typing.Union[str, int, float, bool]
+UpdateSheetRangeRequestValuesItemItem = typing.Union[str, bool]

athena/tools/types/__init__.py

@@ -2,6 +2,33 @@
 
 # isort: skip_file
 
-from .tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
+_dynamic_imports: typing.Dict[str, str] = {"ToolsDataFrameRequestColumnsItem": ".tools_data_frame_request_columns_item"}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = ["ToolsDataFrameRequestColumnsItem"]