mypy-boto3-athena 1.28.0__py3-none-any.whl → 1.34.130__py3-none-any.whl

mypy_boto3_athena/__init__.py

@@ -29,6 +29,7 @@ Usage::
     list_tags_for_resource_paginator: ListTagsForResourcePaginator = client.get_paginator("list_tags_for_resource")
     ```
 """
+
 from .client import AthenaClient
 from .paginator import (
     GetQueryResultsPaginator,
@@ -42,7 +43,6 @@ from .paginator import (
 
 Client = AthenaClient
 
-
 __all__ = (
     "AthenaClient",
     "Client",

mypy_boto3_athena/__init__.pyi

@@ -29,6 +29,7 @@ Usage::
     list_tags_for_resource_paginator: ListTagsForResourcePaginator = client.get_paginator("list_tags_for_resource")
     ```
 """
+
 from .client import AthenaClient
 from .paginator import (
     GetQueryResultsPaginator,

mypy_boto3_athena/__main__.py

@@ -1,6 +1,7 @@
 """
 Main CLI entrypoint.
 """
+
 import sys
 
 
@@ -9,12 +10,13 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3.Athena 1.28.0\nVersion:         1.28.0\nBuilder version:"
-        " 7.14.5\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:      "
-        " https://github.com/youtype/mypy_boto3_builder/releases"
+        "Type annotations for boto3.Athena 1.34.130\n"
+        "Version:         1.34.130\n"
+        "Builder version: 7.24.0\n"
+        "Docs:            https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena//\n"
+        "Boto3 docs:      https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena\n"
+        "Other services:  https://pypi.org/project/boto3-stubs/\n"
+        "Changelog:       https://github.com/youtype/mypy_boto3_builder/releases"
     )
 
 
@@ -22,7 +24,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.28.0")
+    print("1.34.130")
 
 
 def main() -> None:

mypy_boto3_athena/client.py

@@ -13,6 +13,7 @@ Usage::
     client: AthenaClient = session.client("athena")
     ```
 """
+
 import sys
 from typing import Any, Dict, Mapping, Sequence, Type, overload
 
@@ -39,11 +40,11 @@ from .type_defs import (
     BatchGetPreparedStatementOutputTypeDef,
     BatchGetQueryExecutionOutputTypeDef,
     CalculationConfigurationTypeDef,
-    CapacityAssignmentTypeDef,
+    CapacityAssignmentUnionTypeDef,
     CreateNamedQueryOutputTypeDef,
     CreateNotebookOutputTypeDef,
     CreatePresignedNotebookUrlResponseTypeDef,
-    EngineConfigurationTypeDef,
+    EngineConfigurationUnionTypeDef,
     ExportNotebookOutputTypeDef,
     FilterDefinitionTypeDef,
     GetCalculationExecutionCodeResponseTypeDef,
@@ -93,16 +94,15 @@ from .type_defs import (
     WorkGroupConfigurationUpdatesTypeDef,
 )
 
-if sys.version_info >= (3, 9):
+if sys.version_info >= (3, 12):
     from typing import Literal
 else:
     from typing_extensions import Literal
 
-
 __all__ = ("AthenaClient",)
 
 
-class BotocoreClientError(BaseException):
+class BotocoreClientError(Exception):
     MSG_TEMPLATE: str
 
     def __init__(self, error_response: Mapping[str, Any], operation_name: str) -> None:
@@ -141,8 +141,9 @@ class AthenaClient(BaseClient):
         self, *, NamedQueryIds: Sequence[str]
     ) -> BatchGetNamedQueryOutputTypeDef:
         """
-        Returns the details of a single named query or a list of up to 50 queries, which
-        you provide as an array of query ID strings.
+        Returns the details of a single named query or a list of up to 50 queries,
+        which you provide as an array of query ID
+        strings.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_named_query)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_named_query)
@@ -153,7 +154,8 @@ class AthenaClient(BaseClient):
     ) -> BatchGetPreparedStatementOutputTypeDef:
         """
         Returns the details of a single prepared statement or a list of up to 256
-        prepared statements for the array of prepared statement names that you provide.
+        prepared statements for the array of prepared statement names that you
+        provide.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_prepared_statement)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_prepared_statement)
@@ -164,7 +166,8 @@ class AthenaClient(BaseClient):
     ) -> BatchGetQueryExecutionOutputTypeDef:
         """
         Returns the details of a single query execution or a list of up to 50 query
-        executions, which you provide as an array of query execution ID strings.
+        executions, which you provide as an array of query execution ID
+        strings.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_query_execution)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_query_execution)
@@ -199,7 +202,8 @@ class AthenaClient(BaseClient):
     ) -> Dict[str, Any]:
         """
         Creates a capacity reservation with the specified name and number of requested
-        data processing units.
+        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)
@@ -212,7 +216,7 @@ class AthenaClient(BaseClient):
         Type: DataCatalogTypeType,
         Description: str = ...,
         Parameters: Mapping[str, str] = ...,
-        Tags: Sequence[TagTypeDef] = ...
+        Tags: Sequence[TagTypeDef] = ...,
     ) -> Dict[str, Any]:
         """
         Creates (registers) a data catalog with the specified name and properties.
@@ -229,7 +233,7 @@ class AthenaClient(BaseClient):
         QueryString: str,
         Description: str = ...,
         ClientRequestToken: str = ...,
-        WorkGroup: str = ...
+        WorkGroup: str = ...,
     ) -> CreateNamedQueryOutputTypeDef:
         """
         Creates a named query in the specified workgroup.
@@ -274,7 +278,7 @@ class AthenaClient(BaseClient):
         Name: str,
         Configuration: WorkGroupConfigurationTypeDef = ...,
         Description: str = ...,
-        Tags: Sequence[TagTypeDef] = ...
+        Tags: Sequence[TagTypeDef] = ...,
     ) -> Dict[str, Any]:
         """
         Creates a workgroup with the specified name.
@@ -302,7 +306,8 @@ class AthenaClient(BaseClient):
     def delete_named_query(self, *, NamedQueryId: str) -> Dict[str, Any]:
         """
         Deletes the named query if you have access to the workgroup in which the query
-        was saved.
+        was
+        saved.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.delete_named_query)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_named_query)
@@ -406,7 +411,7 @@ class AthenaClient(BaseClient):
         [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:
+    def get_data_catalog(self, *, Name: str, WorkGroup: str = ...) -> GetDataCatalogOutputTypeDef:
         """
         Returns the specified data catalog.
 
@@ -414,7 +419,9 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_data_catalog)
         """
 
-    def get_database(self, *, CatalogName: str, DatabaseName: str) -> GetDatabaseOutputTypeDef:
+    def get_database(
+        self, *, CatalogName: str, DatabaseName: str, WorkGroup: str = ...
+    ) -> GetDatabaseOutputTypeDef:
         """
         Returns a database object for the specified database and data catalog.
 
@@ -452,7 +459,8 @@ class AthenaClient(BaseClient):
     def get_query_execution(self, *, QueryExecutionId: str) -> GetQueryExecutionOutputTypeDef:
         """
         Returns information about a single execution of a query if you have access to
-        the workgroup in which the query ran.
+        the workgroup in which the query
+        ran.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_execution)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_execution)
@@ -463,7 +471,8 @@ class AthenaClient(BaseClient):
     ) -> GetQueryResultsOutputTypeDef:
         """
         Streams the results of a single query execution specified by `QueryExecutionId`
-        from the Athena query results location in Amazon S3.
+        from the Athena query results location in Amazon
+        S3.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_results)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_results)
@@ -474,7 +483,8 @@ class AthenaClient(BaseClient):
     ) -> GetQueryRuntimeStatisticsOutputTypeDef:
         """
         Returns query execution runtime statistics related to a single execution of a
-        query if you have access to the workgroup in which the query ran.
+        query if you have access to the workgroup in which the query
+        ran.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_runtime_statistics)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_runtime_statistics)
@@ -483,7 +493,8 @@ class AthenaClient(BaseClient):
     def get_session(self, *, SessionId: str) -> GetSessionResponseTypeDef:
         """
         Gets the full details of a previously created session, including the session
-        status and configuration.
+        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)
@@ -498,7 +509,7 @@ class AthenaClient(BaseClient):
         """
 
     def get_table_metadata(
-        self, *, CatalogName: str, DatabaseName: str, TableName: str
+        self, *, CatalogName: str, DatabaseName: str, TableName: str, WorkGroup: str = ...
     ) -> GetTableMetadataOutputTypeDef:
         """
         Returns table metadata for the specified catalog, database, and table.
@@ -520,9 +531,10 @@ class AthenaClient(BaseClient):
         *,
         WorkGroup: str,
         Name: str,
-        Payload: str,
         Type: Literal["IPYNB"],
-        ClientRequestToken: str = ...
+        Payload: str = ...,
+        NotebookS3LocationUri: str = ...,
+        ClientRequestToken: str = ...,
     ) -> ImportNotebookOutputTypeDef:
         """
         Imports a single `ipynb` file to a Spark enabled workgroup.
@@ -536,7 +548,8 @@ class AthenaClient(BaseClient):
     ) -> ListApplicationDPUSizesOutputTypeDef:
         """
         Returns the supported DPU sizes for the supported application runtimes (for
-        example, `Athena notebook version 1`).
+        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)
@@ -548,7 +561,7 @@ class AthenaClient(BaseClient):
         SessionId: str,
         StateFilter: CalculationExecutionStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListCalculationExecutionsResponseTypeDef:
         """
         Lists the calculations that have been submitted to a session in descending
@@ -569,7 +582,7 @@ class AthenaClient(BaseClient):
         """
 
     def list_data_catalogs(
-        self, *, NextToken: str = ..., MaxResults: int = ...
+        self, *, NextToken: str = ..., MaxResults: int = ..., WorkGroup: str = ...
     ) -> ListDataCatalogsOutputTypeDef:
         """
         Lists the data catalogs in the current Amazon Web Services account.
@@ -579,7 +592,7 @@ class AthenaClient(BaseClient):
         """
 
     def list_databases(
-        self, *, CatalogName: str, NextToken: str = ..., MaxResults: int = ...
+        self, *, CatalogName: str, NextToken: str = ..., MaxResults: int = ..., WorkGroup: str = ...
     ) -> ListDatabasesOutputTypeDef:
         """
         Lists the databases in the specified data catalog.
@@ -593,7 +606,8 @@ class AthenaClient(BaseClient):
     ) -> ListEngineVersionsOutputTypeDef:
         """
         Returns a list of engine versions that are available to choose from, including
-        the Auto option.
+        the Auto
+        option.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_engine_versions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_engine_versions)
@@ -605,7 +619,7 @@ class AthenaClient(BaseClient):
         SessionId: str,
         ExecutorStateFilter: ExecutorStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListExecutorsResponseTypeDef:
         """
         Lists, in descending order, the executors that joined a session.
@@ -631,7 +645,7 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         Filters: FilterDefinitionTypeDef = ...,
         NextToken: str = ...,
-        MaxResults: int = ...
+        MaxResults: int = ...,
     ) -> ListNotebookMetadataOutputTypeDef:
         """
         Displays the notebook files for the specified workgroup in paginated format.
@@ -645,7 +659,8 @@ class AthenaClient(BaseClient):
     ) -> 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`.
+        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)
@@ -666,7 +681,8 @@ class AthenaClient(BaseClient):
     ) -> ListQueryExecutionsOutputTypeDef:
         """
         Provides a list of available query execution IDs for the queries in the
-        specified workgroup.
+        specified
+        workgroup.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_query_executions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_query_executions)
@@ -678,11 +694,12 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         StateFilter: SessionStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListSessionsResponseTypeDef:
         """
         Lists the sessions in a workgroup that are in an active state like `CREATING`,
-        `CREATED`, `IDLE`, or `BUSY`.
+        `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)
@@ -695,7 +712,8 @@ class AthenaClient(BaseClient):
         DatabaseName: str,
         Expression: str = ...,
         NextToken: str = ...,
-        MaxResults: int = ...
+        MaxResults: int = ...,
+        WorkGroup: str = ...,
     ) -> ListTableMetadataOutputTypeDef:
         """
         Lists the metadata for the tables in the specified data catalog database.
@@ -728,7 +746,7 @@ class AthenaClient(BaseClient):
         self,
         *,
         CapacityReservationName: str,
-        CapacityAssignments: Sequence[CapacityAssignmentTypeDef]
+        CapacityAssignments: Sequence[CapacityAssignmentUnionTypeDef],
     ) -> Dict[str, Any]:
         """
         Puts a new capacity assignment configuration for a specified capacity
@@ -745,7 +763,7 @@ class AthenaClient(BaseClient):
         Description: str = ...,
         CalculationConfiguration: CalculationConfigurationTypeDef = ...,
         CodeBlock: str = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> StartCalculationExecutionResponseTypeDef:
         """
         Submits calculations for execution within a session.
@@ -763,7 +781,7 @@ class AthenaClient(BaseClient):
         ResultConfiguration: ResultConfigurationTypeDef = ...,
         WorkGroup: str = ...,
         ExecutionParameters: Sequence[str] = ...,
-        ResultReuseConfiguration: ResultReuseConfigurationTypeDef = ...
+        ResultReuseConfiguration: ResultReuseConfigurationTypeDef = ...,
     ) -> StartQueryExecutionOutputTypeDef:
         """
         Runs the SQL query statements contained in the `Query`.
@@ -776,11 +794,11 @@ class AthenaClient(BaseClient):
         self,
         *,
         WorkGroup: str,
-        EngineConfiguration: EngineConfigurationTypeDef,
+        EngineConfiguration: EngineConfigurationUnionTypeDef,
         Description: str = ...,
         NotebookVersion: str = ...,
         SessionIdleTimeoutInMinutes: int = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> StartSessionResponseTypeDef:
         """
         Creates a session for running calculations within a workgroup.
@@ -834,7 +852,8 @@ class AthenaClient(BaseClient):
     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.
+        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)
@@ -846,7 +865,7 @@ class AthenaClient(BaseClient):
         Name: str,
         Type: DataCatalogTypeType,
         Description: str = ...,
-        Parameters: Mapping[str, str] = ...
+        Parameters: Mapping[str, str] = ...,
     ) -> Dict[str, Any]:
         """
         Updates the data catalog that has the specified name.
@@ -872,7 +891,7 @@ class AthenaClient(BaseClient):
         Payload: str,
         Type: Literal["IPYNB"],
         SessionId: str = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> Dict[str, Any]:
         """
         Updates the contents of a Spark notebook.
@@ -907,7 +926,7 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         Description: str = ...,
         ConfigurationUpdates: WorkGroupConfigurationUpdatesTypeDef = ...,
-        State: WorkGroupStateType = ...
+        State: WorkGroupStateType = ...,
     ) -> Dict[str, Any]:
         """
         Updates the workgroup with the specified name.