athena-intelligence 0.1.210__py3-none-any.whl → 0.1.303__py3-none-any.whl

athena/tools/sheets/raw_client.py

@@ -8,10 +8,15 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.http_response import AsyncHttpResponse, HttpResponse
 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.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, ...)
@@ -502,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]:
@@ -523,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)
 
@@ -553,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",
@@ -597,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]:
@@ -615,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)
@@ -634,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",
@@ -750,6 +765,7 @@ class RawSheetsClient:
         self,
         *,
         asset_id: str,
+        new_sheet_id: typing.Optional[int] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
@@ -761,6 +777,9 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
+        new_sheet_id : typing.Optional[int]
+            New sheet ID for the duplicated sheet (auto-generated if not provided)
+
         sheet_id : typing.Optional[int]
             Sheet ID to duplicate
 
@@ -777,6 +796,7 @@ class RawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "new_sheet_id": new_sheet_id,
                 "sheet_id": sheet_id,
             },
             headers={
@@ -812,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.
@@ -829,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.
@@ -853,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",
@@ -1145,8 +1146,8 @@ class RawSheetsClient:
         self,
         *,
         asset_id: str,
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[GetTableResponse]:
         """
@@ -1157,12 +1158,12 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
+        table_name : str
+            Table name to retrieve
+
         table_id : typing.Optional[str]
             Table ID to retrieve
 
-        table_name : typing.Optional[str]
-            Table name to retrieve
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1215,9 +1216,9 @@ class RawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[TableRowData],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
         """
@@ -1228,15 +1229,15 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[TableRowData]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1867,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]:
@@ -1888,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)
 
@@ -1918,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",
@@ -1962,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]:
@@ -1980,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)
@@ -1999,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",
@@ -2115,6 +2126,7 @@ class AsyncRawSheetsClient:
         self,
         *,
         asset_id: str,
+        new_sheet_id: typing.Optional[int] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
@@ -2126,6 +2138,9 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
+        new_sheet_id : typing.Optional[int]
+            New sheet ID for the duplicated sheet (auto-generated if not provided)
+
         sheet_id : typing.Optional[int]
             Sheet ID to duplicate
 
@@ -2142,6 +2157,7 @@ class AsyncRawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "new_sheet_id": new_sheet_id,
                 "sheet_id": sheet_id,
             },
             headers={
@@ -2177,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.
@@ -2194,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.
@@ -2218,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",
@@ -2510,8 +2507,8 @@ class AsyncRawSheetsClient:
         self,
         *,
         asset_id: str,
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[GetTableResponse]:
         """
@@ -2522,12 +2519,12 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
+        table_name : str
+            Table name to retrieve
+
         table_id : typing.Optional[str]
             Table ID to retrieve
 
-        table_name : typing.Optional[str]
-            Table name to retrieve
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -2580,9 +2577,9 @@ class AsyncRawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[TableRowData],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
         """
@@ -2593,15 +2590,15 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[TableRowData]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 

athena/tools/sheets/types/__init__.py

@@ -0,0 +1,36 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+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/update_sheet_range_request_values_item_item.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+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"]