mypy-boto3-athena 1.26.0.post1__py3-none-any.whl → 1.28.36__py3-none-any.whl

mypy_boto3_athena/__main__.py

@@ -9,8 +9,8 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3.Athena 1.26.0\nVersion:         1.26.0.post1\nBuilder version:"
-        " 7.11.10\nDocs:           "
+        "Type annotations for boto3.Athena 1.28.36\nVersion:         1.28.36\nBuilder version:"
+        " 7.18.0\nDocs:           "
         " https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena//\nBoto3 docs:     "
         " https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena\nOther"
         " services:  https://pypi.org/project/boto3-stubs/\nChangelog:      "
@@ -22,7 +22,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.26.0.post1")
+    print("1.28.36")
 
 
 def main() -> None:

mypy_boto3_athena/client.py

@@ -18,7 +18,13 @@ from typing import Any, Dict, Mapping, Sequence, Type, overload
 
 from botocore.client import BaseClient, ClientMeta
 
-from .literals import DataCatalogTypeType, WorkGroupStateType
+from .literals import (
+    CalculationExecutionStateType,
+    DataCatalogTypeType,
+    ExecutorStateType,
+    SessionStateType,
+    WorkGroupStateType,
+)
 from .paginator import (
     GetQueryResultsPaginator,
     ListDatabasesPaginator,
@@ -32,34 +38,62 @@ from .type_defs import (
     BatchGetNamedQueryOutputTypeDef,
     BatchGetPreparedStatementOutputTypeDef,
     BatchGetQueryExecutionOutputTypeDef,
+    CalculationConfigurationTypeDef,
+    CapacityAssignmentTypeDef,
     CreateNamedQueryOutputTypeDef,
+    CreateNotebookOutputTypeDef,
+    CreatePresignedNotebookUrlResponseTypeDef,
+    EngineConfigurationTypeDef,
+    ExportNotebookOutputTypeDef,
+    FilterDefinitionTypeDef,
+    GetCalculationExecutionCodeResponseTypeDef,
+    GetCalculationExecutionResponseTypeDef,
+    GetCalculationExecutionStatusResponseTypeDef,
+    GetCapacityAssignmentConfigurationOutputTypeDef,
+    GetCapacityReservationOutputTypeDef,
     GetDatabaseOutputTypeDef,
     GetDataCatalogOutputTypeDef,
     GetNamedQueryOutputTypeDef,
+    GetNotebookMetadataOutputTypeDef,
     GetPreparedStatementOutputTypeDef,
     GetQueryExecutionOutputTypeDef,
     GetQueryResultsOutputTypeDef,
     GetQueryRuntimeStatisticsOutputTypeDef,
+    GetSessionResponseTypeDef,
+    GetSessionStatusResponseTypeDef,
     GetTableMetadataOutputTypeDef,
     GetWorkGroupOutputTypeDef,
+    ImportNotebookOutputTypeDef,
+    ListApplicationDPUSizesOutputTypeDef,
+    ListCalculationExecutionsResponseTypeDef,
+    ListCapacityReservationsOutputTypeDef,
     ListDatabasesOutputTypeDef,
     ListDataCatalogsOutputTypeDef,
     ListEngineVersionsOutputTypeDef,
+    ListExecutorsResponseTypeDef,
     ListNamedQueriesOutputTypeDef,
+    ListNotebookMetadataOutputTypeDef,
+    ListNotebookSessionsResponseTypeDef,
     ListPreparedStatementsOutputTypeDef,
     ListQueryExecutionsOutputTypeDef,
+    ListSessionsResponseTypeDef,
     ListTableMetadataOutputTypeDef,
     ListTagsForResourceOutputTypeDef,
     ListWorkGroupsOutputTypeDef,
     QueryExecutionContextTypeDef,
     ResultConfigurationTypeDef,
+    ResultReuseConfigurationTypeDef,
+    StartCalculationExecutionResponseTypeDef,
     StartQueryExecutionOutputTypeDef,
+    StartSessionResponseTypeDef,
+    StopCalculationExecutionResponseTypeDef,
     TagTypeDef,
+    TerminateSessionResponseTypeDef,
     WorkGroupConfigurationTypeDef,
     WorkGroupConfigurationUpdatesTypeDef,
 )
 
-if sys.version_info >= (3, 9):
+if sys.version_info >= (3, 12):
     from typing import Literal
 else:
     from typing_extensions import Literal
@@ -82,6 +116,7 @@ class Exceptions:
     InvalidRequestException: Type[BotocoreClientError]
     MetadataException: Type[BotocoreClientError]
     ResourceNotFoundException: Type[BotocoreClientError]
+    SessionAlreadyExistsException: Type[BotocoreClientError]
     TooManyRequestsException: Type[BotocoreClientError]
 
 
@@ -143,6 +178,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#can_paginate)
         """
 
+    def cancel_capacity_reservation(self, *, Name: str) -> Dict[str, Any]:
+        """
+        Cancels the capacity reservation with the specified name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.cancel_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#cancel_capacity_reservation)
+        """
+
     def close(self) -> None:
         """
         Closes underlying endpoint connections.
@@ -151,6 +194,17 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#close)
         """
 
+    def create_capacity_reservation(
+        self, *, TargetDpus: int, Name: str, Tags: Sequence[TagTypeDef] = ...
+    ) -> Dict[str, Any]:
+        """
+        Creates a capacity reservation with the specified name and number of requested
+        data processing units.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.create_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_capacity_reservation)
+        """
+
     def create_data_catalog(
         self,
         *,
@@ -184,6 +238,16 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_named_query)
         """
 
+    def create_notebook(
+        self, *, WorkGroup: str, Name: str, ClientRequestToken: str = ...
+    ) -> CreateNotebookOutputTypeDef:
+        """
+        Creates an empty `ipynb` file in the specified Apache Spark enabled workgroup.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.create_notebook)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_notebook)
+        """
+
     def create_prepared_statement(
         self, *, StatementName: str, WorkGroup: str, QueryStatement: str, Description: str = ...
     ) -> Dict[str, Any]:
@@ -194,6 +258,16 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_prepared_statement)
         """
 
+    def create_presigned_notebook_url(
+        self, *, SessionId: str
+    ) -> CreatePresignedNotebookUrlResponseTypeDef:
+        """
+        Gets an authentication token and the URL at which the notebook can be accessed.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.create_presigned_notebook_url)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_presigned_notebook_url)
+        """
+
     def create_work_group(
         self,
         *,
@@ -209,6 +283,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_work_group)
         """
 
+    def delete_capacity_reservation(self, *, Name: str) -> Dict[str, Any]:
+        """
+        Deletes a cancelled capacity reservation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.delete_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_capacity_reservation)
+        """
+
     def delete_data_catalog(self, *, Name: str) -> Dict[str, Any]:
         """
         Deletes a data catalog.
@@ -226,6 +308,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_named_query)
         """
 
+    def delete_notebook(self, *, NotebookId: str) -> Dict[str, Any]:
+        """
+        Deletes the specified notebook.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.delete_notebook)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_notebook)
+        """
+
     def delete_prepared_statement(self, *, StatementName: str, WorkGroup: str) -> Dict[str, Any]:
         """
         Deletes the prepared statement with the specified name from the specified
@@ -245,6 +335,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_work_group)
         """
 
+    def export_notebook(self, *, NotebookId: str) -> ExportNotebookOutputTypeDef:
+        """
+        Exports the specified notebook and its metadata.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.export_notebook)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#export_notebook)
+        """
+
     def generate_presigned_url(
         self,
         ClientMethod: str,
@@ -259,6 +357,55 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#generate_presigned_url)
         """
 
+    def get_calculation_execution(
+        self, *, CalculationExecutionId: str
+    ) -> GetCalculationExecutionResponseTypeDef:
+        """
+        Describes a previously submitted calculation execution.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution)
+        """
+
+    def get_calculation_execution_code(
+        self, *, CalculationExecutionId: str
+    ) -> GetCalculationExecutionCodeResponseTypeDef:
+        """
+        Retrieves the unencrypted code that was executed for the calculation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution_code)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_code)
+        """
+
+    def get_calculation_execution_status(
+        self, *, CalculationExecutionId: str
+    ) -> GetCalculationExecutionStatusResponseTypeDef:
+        """
+        Gets the status of a current calculation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution_status)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_status)
+        """
+
+    def get_capacity_assignment_configuration(
+        self, *, CapacityReservationName: str
+    ) -> GetCapacityAssignmentConfigurationOutputTypeDef:
+        """
+        Gets the capacity assignment configuration for a capacity reservation, if one
+        exists.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_capacity_assignment_configuration)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_capacity_assignment_configuration)
+        """
+
+    def get_capacity_reservation(self, *, Name: str) -> GetCapacityReservationOutputTypeDef:
+        """
+        Returns information about the capacity reservation with the specified name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_capacity_reservation)
+        """
+
     def get_data_catalog(self, *, Name: str) -> GetDataCatalogOutputTypeDef:
         """
         Returns the specified data catalog.
@@ -283,6 +430,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_named_query)
         """
 
+    def get_notebook_metadata(self, *, NotebookId: str) -> GetNotebookMetadataOutputTypeDef:
+        """
+        Retrieves notebook metadata for the specified notebook ID.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_notebook_metadata)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_notebook_metadata)
+        """
+
     def get_prepared_statement(
         self, *, StatementName: str, WorkGroup: str
     ) -> GetPreparedStatementOutputTypeDef:
@@ -325,6 +480,23 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_runtime_statistics)
         """
 
+    def get_session(self, *, SessionId: str) -> GetSessionResponseTypeDef:
+        """
+        Gets the full details of a previously created session, including the session
+        status and configuration.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_session)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_session)
+        """
+
+    def get_session_status(self, *, SessionId: str) -> GetSessionStatusResponseTypeDef:
+        """
+        Gets the current status of a session.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_session_status)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_session_status)
+        """
+
     def get_table_metadata(
         self, *, CatalogName: str, DatabaseName: str, TableName: str
     ) -> GetTableMetadataOutputTypeDef:
@@ -343,6 +515,59 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_work_group)
         """
 
+    def import_notebook(
+        self,
+        *,
+        WorkGroup: str,
+        Name: str,
+        Payload: str,
+        Type: Literal["IPYNB"],
+        ClientRequestToken: str = ...
+    ) -> ImportNotebookOutputTypeDef:
+        """
+        Imports a single `ipynb` file to a Spark enabled workgroup.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.import_notebook)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#import_notebook)
+        """
+
+    def list_application_dpu_sizes(
+        self, *, MaxResults: int = ..., NextToken: str = ...
+    ) -> ListApplicationDPUSizesOutputTypeDef:
+        """
+        Returns the supported DPU sizes for the supported application runtimes (for
+        example, `Athena notebook version 1`).
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_application_dpu_sizes)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_application_dpu_sizes)
+        """
+
+    def list_calculation_executions(
+        self,
+        *,
+        SessionId: str,
+        StateFilter: CalculationExecutionStateType = ...,
+        MaxResults: int = ...,
+        NextToken: str = ...
+    ) -> ListCalculationExecutionsResponseTypeDef:
+        """
+        Lists the calculations that have been submitted to a session in descending
+        order.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_calculation_executions)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_calculation_executions)
+        """
+
+    def list_capacity_reservations(
+        self, *, NextToken: str = ..., MaxResults: int = ...
+    ) -> ListCapacityReservationsOutputTypeDef:
+        """
+        Lists the capacity reservations for the current account.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_capacity_reservations)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_capacity_reservations)
+        """
+
     def list_data_catalogs(
         self, *, NextToken: str = ..., MaxResults: int = ...
     ) -> ListDataCatalogsOutputTypeDef:
@@ -374,6 +599,21 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_engine_versions)
         """
 
+    def list_executors(
+        self,
+        *,
+        SessionId: str,
+        ExecutorStateFilter: ExecutorStateType = ...,
+        MaxResults: int = ...,
+        NextToken: str = ...
+    ) -> ListExecutorsResponseTypeDef:
+        """
+        Lists, in descending order, the executors that joined a session.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_executors)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_executors)
+        """
+
     def list_named_queries(
         self, *, NextToken: str = ..., MaxResults: int = ..., WorkGroup: str = ...
     ) -> ListNamedQueriesOutputTypeDef:
@@ -385,6 +625,32 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_named_queries)
         """
 
+    def list_notebook_metadata(
+        self,
+        *,
+        WorkGroup: str,
+        Filters: FilterDefinitionTypeDef = ...,
+        NextToken: str = ...,
+        MaxResults: int = ...
+    ) -> ListNotebookMetadataOutputTypeDef:
+        """
+        Displays the notebook files for the specified workgroup in paginated format.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_notebook_metadata)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_notebook_metadata)
+        """
+
+    def list_notebook_sessions(
+        self, *, NotebookId: str, MaxResults: int = ..., NextToken: str = ...
+    ) -> ListNotebookSessionsResponseTypeDef:
+        """
+        Lists, in descending order, the sessions that have been created in a notebook
+        that are in an active state like `CREATING`, `CREATED`, `IDLE` or `BUSY`.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_notebook_sessions)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_notebook_sessions)
+        """
+
     def list_prepared_statements(
         self, *, WorkGroup: str, NextToken: str = ..., MaxResults: int = ...
     ) -> ListPreparedStatementsOutputTypeDef:
@@ -406,6 +672,22 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_query_executions)
         """
 
+    def list_sessions(
+        self,
+        *,
+        WorkGroup: str,
+        StateFilter: SessionStateType = ...,
+        MaxResults: int = ...,
+        NextToken: str = ...
+    ) -> ListSessionsResponseTypeDef:
+        """
+        Lists the sessions in a workgroup that are in an active state like `CREATING`,
+        `CREATED`, `IDLE`, or `BUSY`.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_sessions)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_sessions)
+        """
+
     def list_table_metadata(
         self,
         *,
@@ -426,7 +708,7 @@ class AthenaClient(BaseClient):
         self, *, ResourceARN: str, NextToken: str = ..., MaxResults: int = ...
     ) -> ListTagsForResourceOutputTypeDef:
         """
-        Lists the tags associated with an Athena workgroup or data catalog resource.
+        Lists the tags associated with an Athena resource.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_tags_for_resource)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_tags_for_resource)
@@ -442,6 +724,36 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_work_groups)
         """
 
+    def put_capacity_assignment_configuration(
+        self,
+        *,
+        CapacityReservationName: str,
+        CapacityAssignments: Sequence[CapacityAssignmentTypeDef]
+    ) -> Dict[str, Any]:
+        """
+        Puts a new capacity assignment configuration for a specified capacity
+        reservation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.put_capacity_assignment_configuration)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#put_capacity_assignment_configuration)
+        """
+
+    def start_calculation_execution(
+        self,
+        *,
+        SessionId: str,
+        Description: str = ...,
+        CalculationConfiguration: CalculationConfigurationTypeDef = ...,
+        CodeBlock: str = ...,
+        ClientRequestToken: str = ...
+    ) -> StartCalculationExecutionResponseTypeDef:
+        """
+        Submits calculations for execution within a session.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.start_calculation_execution)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#start_calculation_execution)
+        """
+
     def start_query_execution(
         self,
         *,
@@ -450,7 +762,8 @@ class AthenaClient(BaseClient):
         QueryExecutionContext: QueryExecutionContextTypeDef = ...,
         ResultConfiguration: ResultConfigurationTypeDef = ...,
         WorkGroup: str = ...,
-        ExecutionParameters: Sequence[str] = ...
+        ExecutionParameters: Sequence[str] = ...,
+        ResultReuseConfiguration: ResultReuseConfigurationTypeDef = ...
     ) -> StartQueryExecutionOutputTypeDef:
         """
         Runs the SQL query statements contained in the `Query`.
@@ -459,6 +772,33 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#start_query_execution)
         """
 
+    def start_session(
+        self,
+        *,
+        WorkGroup: str,
+        EngineConfiguration: EngineConfigurationTypeDef,
+        Description: str = ...,
+        NotebookVersion: str = ...,
+        SessionIdleTimeoutInMinutes: int = ...,
+        ClientRequestToken: str = ...
+    ) -> StartSessionResponseTypeDef:
+        """
+        Creates a session for running calculations within a workgroup.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.start_session)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#start_session)
+        """
+
+    def stop_calculation_execution(
+        self, *, CalculationExecutionId: str
+    ) -> StopCalculationExecutionResponseTypeDef:
+        """
+        Requests the cancellation of a calculation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.stop_calculation_execution)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#stop_calculation_execution)
+        """
+
     def stop_query_execution(self, *, QueryExecutionId: str) -> Dict[str, Any]:
         """
         Stops a query execution.
@@ -475,14 +815,31 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#tag_resource)
         """
 
+    def terminate_session(self, *, SessionId: str) -> TerminateSessionResponseTypeDef:
+        """
+        Terminates an active session.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.terminate_session)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#terminate_session)
+        """
+
     def untag_resource(self, *, ResourceARN: str, TagKeys: Sequence[str]) -> Dict[str, Any]:
         """
-        Removes one or more tags from a data catalog or workgroup resource.
+        Removes one or more tags from an Athena resource.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.untag_resource)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#untag_resource)
         """
 
+    def update_capacity_reservation(self, *, TargetDpus: int, Name: str) -> Dict[str, Any]:
+        """
+        Updates the number of requested data processing units for the capacity
+        reservation with the specified name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.update_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#update_capacity_reservation)
+        """
+
     def update_data_catalog(
         self,
         *,
@@ -508,6 +865,32 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#update_named_query)
         """
 
+    def update_notebook(
+        self,
+        *,
+        NotebookId: str,
+        Payload: str,
+        Type: Literal["IPYNB"],
+        SessionId: str = ...,
+        ClientRequestToken: str = ...
+    ) -> Dict[str, Any]:
+        """
+        Updates the contents of a Spark notebook.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.update_notebook)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#update_notebook)
+        """
+
+    def update_notebook_metadata(
+        self, *, NotebookId: str, Name: str, ClientRequestToken: str = ...
+    ) -> Dict[str, Any]:
+        """
+        Updates the metadata for a notebook.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.update_notebook_metadata)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#update_notebook_metadata)
+        """
+
     def update_prepared_statement(
         self, *, StatementName: str, WorkGroup: str, QueryStatement: str, Description: str = ...
     ) -> Dict[str, Any]: