mypydantic.models.athena

   1# Model Generated: Thu Mar  2 21:56:16 2023
   2
   3from __future__ import annotations
   4
   5from datetime import datetime
   6from typing import Any, Dict, List, Literal, Mapping, Optional, Sequence
   7
   8from pydantic import Field
   9
  10from mypydantic.models.base_model import BaseModel
  11
  12
  13class AclConfigurationModel(BaseModel):
  14    s3_acl_option: Literal["BUCKET_OWNER_FULL_CONTROL"] = Field(alias="S3AclOption")
  15
  16
  17class ApplicationDPUSizesModel(BaseModel):
  18    application_runtime_id: Optional[str] = Field(
  19        default=None, alias="ApplicationRuntimeId"
  20    )
  21    supported_dp_usizes: Optional[List[int]] = Field(
  22        default=None, alias="SupportedDPUSizes"
  23    )
  24
  25
  26class AthenaErrorModel(BaseModel):
  27    error_category: Optional[int] = Field(default=None, alias="ErrorCategory")
  28    error_type: Optional[int] = Field(default=None, alias="ErrorType")
  29    retryable: Optional[bool] = Field(default=None, alias="Retryable")
  30    error_message: Optional[str] = Field(default=None, alias="ErrorMessage")
  31
  32
  33class BatchGetNamedQueryInputRequestModel(BaseModel):
  34    named_query_ids: Sequence[str] = Field(alias="NamedQueryIds")
  35
  36
  37class NamedQueryModel(BaseModel):
  38    name: str = Field(alias="Name")
  39    database: str = Field(alias="Database")
  40    query_string: str = Field(alias="QueryString")
  41    description: Optional[str] = Field(default=None, alias="Description")
  42    named_query_id: Optional[str] = Field(default=None, alias="NamedQueryId")
  43    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
  44
  45
  46class ResponseMetadataModel(BaseModel):
  47    request_id: str = Field(alias="RequestId")
  48    host_id: str = Field(alias="HostId")
  49    http_status_code: int = Field(alias="HTTPStatusCode")
  50    http_headers: Dict[str, str] = Field(alias="HTTPHeaders")
  51    retry_attempts: int = Field(alias="RetryAttempts")
  52
  53
  54class UnprocessedNamedQueryIdModel(BaseModel):
  55    named_query_id: Optional[str] = Field(default=None, alias="NamedQueryId")
  56    error_code: Optional[str] = Field(default=None, alias="ErrorCode")
  57    error_message: Optional[str] = Field(default=None, alias="ErrorMessage")
  58
  59
  60class BatchGetPreparedStatementInputRequestModel(BaseModel):
  61    prepared_statement_names: Sequence[str] = Field(alias="PreparedStatementNames")
  62    work_group: str = Field(alias="WorkGroup")
  63
  64
  65class PreparedStatementModel(BaseModel):
  66    statement_name: Optional[str] = Field(default=None, alias="StatementName")
  67    query_statement: Optional[str] = Field(default=None, alias="QueryStatement")
  68    work_group_name: Optional[str] = Field(default=None, alias="WorkGroupName")
  69    description: Optional[str] = Field(default=None, alias="Description")
  70    last_modified_time: Optional[datetime] = Field(
  71        default=None, alias="LastModifiedTime"
  72    )
  73
  74
  75class UnprocessedPreparedStatementNameModel(BaseModel):
  76    statement_name: Optional[str] = Field(default=None, alias="StatementName")
  77    error_code: Optional[str] = Field(default=None, alias="ErrorCode")
  78    error_message: Optional[str] = Field(default=None, alias="ErrorMessage")
  79
  80
  81class BatchGetQueryExecutionInputRequestModel(BaseModel):
  82    query_execution_ids: Sequence[str] = Field(alias="QueryExecutionIds")
  83
  84
  85class UnprocessedQueryExecutionIdModel(BaseModel):
  86    query_execution_id: Optional[str] = Field(default=None, alias="QueryExecutionId")
  87    error_code: Optional[str] = Field(default=None, alias="ErrorCode")
  88    error_message: Optional[str] = Field(default=None, alias="ErrorMessage")
  89
  90
  91class CalculationConfigurationModel(BaseModel):
  92    code_block: Optional[str] = Field(default=None, alias="CodeBlock")
  93
  94
  95class CalculationResultModel(BaseModel):
  96    std_out_s3_uri: Optional[str] = Field(default=None, alias="StdOutS3Uri")
  97    std_error_s3_uri: Optional[str] = Field(default=None, alias="StdErrorS3Uri")
  98    result_s3_uri: Optional[str] = Field(default=None, alias="ResultS3Uri")
  99    result_type: Optional[str] = Field(default=None, alias="ResultType")
 100
 101
 102class CalculationStatisticsModel(BaseModel):
 103    dpu_execution_in_millis: Optional[int] = Field(
 104        default=None, alias="DpuExecutionInMillis"
 105    )
 106    progress: Optional[str] = Field(default=None, alias="Progress")
 107
 108
 109class CalculationStatusModel(BaseModel):
 110    submission_date_time: Optional[datetime] = Field(
 111        default=None, alias="SubmissionDateTime"
 112    )
 113    completion_date_time: Optional[datetime] = Field(
 114        default=None, alias="CompletionDateTime"
 115    )
 116    state: Optional[
 117        Literal[
 118            "CANCELED",
 119            "CANCELING",
 120            "COMPLETED",
 121            "CREATED",
 122            "CREATING",
 123            "FAILED",
 124            "QUEUED",
 125            "RUNNING",
 126        ]
 127    ] = Field(default=None, alias="State")
 128    state_change_reason: Optional[str] = Field(default=None, alias="StateChangeReason")
 129
 130
 131class ColumnInfoModel(BaseModel):
 132    name: str = Field(alias="Name")
 133    type: str = Field(alias="Type")
 134    catalog_name: Optional[str] = Field(default=None, alias="CatalogName")
 135    schema_name: Optional[str] = Field(default=None, alias="SchemaName")
 136    table_name: Optional[str] = Field(default=None, alias="TableName")
 137    label: Optional[str] = Field(default=None, alias="Label")
 138    precision: Optional[int] = Field(default=None, alias="Precision")
 139    scale: Optional[int] = Field(default=None, alias="Scale")
 140    nullable: Optional[Literal["NOT_NULL", "NULLABLE", "UNKNOWN"]] = Field(
 141        default=None, alias="Nullable"
 142    )
 143    case_sensitive: Optional[bool] = Field(default=None, alias="CaseSensitive")
 144
 145
 146class ColumnModel(BaseModel):
 147    name: str = Field(alias="Name")
 148    type: Optional[str] = Field(default=None, alias="Type")
 149    comment: Optional[str] = Field(default=None, alias="Comment")
 150
 151
 152class TagModel(BaseModel):
 153    key: Optional[str] = Field(default=None, alias="Key")
 154    value: Optional[str] = Field(default=None, alias="Value")
 155
 156
 157class CreateNamedQueryInputRequestModel(BaseModel):
 158    name: str = Field(alias="Name")
 159    database: str = Field(alias="Database")
 160    query_string: str = Field(alias="QueryString")
 161    description: Optional[str] = Field(default=None, alias="Description")
 162    client_request_token: Optional[str] = Field(
 163        default=None, alias="ClientRequestToken"
 164    )
 165    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
 166
 167
 168class CreateNotebookInputRequestModel(BaseModel):
 169    work_group: str = Field(alias="WorkGroup")
 170    name: str = Field(alias="Name")
 171    client_request_token: Optional[str] = Field(
 172        default=None, alias="ClientRequestToken"
 173    )
 174
 175
 176class CreatePreparedStatementInputRequestModel(BaseModel):
 177    statement_name: str = Field(alias="StatementName")
 178    work_group: str = Field(alias="WorkGroup")
 179    query_statement: str = Field(alias="QueryStatement")
 180    description: Optional[str] = Field(default=None, alias="Description")
 181
 182
 183class CreatePresignedNotebookUrlRequestModel(BaseModel):
 184    session_id: str = Field(alias="SessionId")
 185
 186
 187class CustomerContentEncryptionConfigurationModel(BaseModel):
 188    kms_key: str = Field(alias="KmsKey")
 189
 190
 191class DataCatalogSummaryModel(BaseModel):
 192    catalog_name: Optional[str] = Field(default=None, alias="CatalogName")
 193    type: Optional[Literal["GLUE", "HIVE", "LAMBDA"]] = Field(
 194        default=None, alias="Type"
 195    )
 196
 197
 198class DataCatalogModel(BaseModel):
 199    name: str = Field(alias="Name")
 200    type: Literal["GLUE", "HIVE", "LAMBDA"] = Field(alias="Type")
 201    description: Optional[str] = Field(default=None, alias="Description")
 202    parameters: Optional[Dict[str, str]] = Field(default=None, alias="Parameters")
 203
 204
 205class DatabaseModel(BaseModel):
 206    name: str = Field(alias="Name")
 207    description: Optional[str] = Field(default=None, alias="Description")
 208    parameters: Optional[Dict[str, str]] = Field(default=None, alias="Parameters")
 209
 210
 211class DatumModel(BaseModel):
 212    var_char_value: Optional[str] = Field(default=None, alias="VarCharValue")
 213
 214
 215class DeleteDataCatalogInputRequestModel(BaseModel):
 216    name: str = Field(alias="Name")
 217
 218
 219class DeleteNamedQueryInputRequestModel(BaseModel):
 220    named_query_id: str = Field(alias="NamedQueryId")
 221
 222
 223class DeleteNotebookInputRequestModel(BaseModel):
 224    notebook_id: str = Field(alias="NotebookId")
 225
 226
 227class DeletePreparedStatementInputRequestModel(BaseModel):
 228    statement_name: str = Field(alias="StatementName")
 229    work_group: str = Field(alias="WorkGroup")
 230
 231
 232class DeleteWorkGroupInputRequestModel(BaseModel):
 233    work_group: str = Field(alias="WorkGroup")
 234    recursive_delete_option: Optional[bool] = Field(
 235        default=None, alias="RecursiveDeleteOption"
 236    )
 237
 238
 239class EncryptionConfigurationModel(BaseModel):
 240    encryption_option: Literal["CSE_KMS", "SSE_KMS", "SSE_S3"] = Field(
 241        alias="EncryptionOption"
 242    )
 243    kms_key: Optional[str] = Field(default=None, alias="KmsKey")
 244
 245
 246class EngineConfigurationModel(BaseModel):
 247    max_concurrent_dpus: int = Field(alias="MaxConcurrentDpus")
 248    coordinator_dpu_size: Optional[int] = Field(
 249        default=None, alias="CoordinatorDpuSize"
 250    )
 251    default_executor_dpu_size: Optional[int] = Field(
 252        default=None, alias="DefaultExecutorDpuSize"
 253    )
 254    additional_configs: Optional[Dict[str, str]] = Field(
 255        default=None, alias="AdditionalConfigs"
 256    )
 257
 258
 259class EngineVersionModel(BaseModel):
 260    selected_engine_version: Optional[str] = Field(
 261        default=None, alias="SelectedEngineVersion"
 262    )
 263    effective_engine_version: Optional[str] = Field(
 264        default=None, alias="EffectiveEngineVersion"
 265    )
 266
 267
 268class ExecutorsSummaryModel(BaseModel):
 269    executor_id: str = Field(alias="ExecutorId")
 270    executor_type: Optional[Literal["COORDINATOR", "GATEWAY", "WORKER"]] = Field(
 271        default=None, alias="ExecutorType"
 272    )
 273    start_date_time: Optional[int] = Field(default=None, alias="StartDateTime")
 274    termination_date_time: Optional[int] = Field(
 275        default=None, alias="TerminationDateTime"
 276    )
 277    executor_state: Optional[
 278        Literal[
 279            "CREATED", "CREATING", "FAILED", "REGISTERED", "TERMINATED", "TERMINATING"
 280        ]
 281    ] = Field(default=None, alias="ExecutorState")
 282    executor_size: Optional[int] = Field(default=None, alias="ExecutorSize")
 283
 284
 285class ExportNotebookInputRequestModel(BaseModel):
 286    notebook_id: str = Field(alias="NotebookId")
 287
 288
 289class NotebookMetadataModel(BaseModel):
 290    notebook_id: Optional[str] = Field(default=None, alias="NotebookId")
 291    name: Optional[str] = Field(default=None, alias="Name")
 292    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
 293    creation_time: Optional[datetime] = Field(default=None, alias="CreationTime")
 294    type: Optional[Literal["IPYNB"]] = Field(default=None, alias="Type")
 295    last_modified_time: Optional[datetime] = Field(
 296        default=None, alias="LastModifiedTime"
 297    )
 298
 299
 300class FilterDefinitionModel(BaseModel):
 301    name: Optional[str] = Field(default=None, alias="Name")
 302
 303
 304class GetCalculationExecutionCodeRequestModel(BaseModel):
 305    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 306
 307
 308class GetCalculationExecutionRequestModel(BaseModel):
 309    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 310
 311
 312class GetCalculationExecutionStatusRequestModel(BaseModel):
 313    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 314
 315
 316class GetDataCatalogInputRequestModel(BaseModel):
 317    name: str = Field(alias="Name")
 318
 319
 320class GetDatabaseInputRequestModel(BaseModel):
 321    catalog_name: str = Field(alias="CatalogName")
 322    database_name: str = Field(alias="DatabaseName")
 323
 324
 325class GetNamedQueryInputRequestModel(BaseModel):
 326    named_query_id: str = Field(alias="NamedQueryId")
 327
 328
 329class GetNotebookMetadataInputRequestModel(BaseModel):
 330    notebook_id: str = Field(alias="NotebookId")
 331
 332
 333class GetPreparedStatementInputRequestModel(BaseModel):
 334    statement_name: str = Field(alias="StatementName")
 335    work_group: str = Field(alias="WorkGroup")
 336
 337
 338class GetQueryExecutionInputRequestModel(BaseModel):
 339    query_execution_id: str = Field(alias="QueryExecutionId")
 340
 341
 342class PaginatorConfigModel(BaseModel):
 343    max_items: Optional[int] = Field(default=None, alias="MaxItems")
 344    page_size: Optional[int] = Field(default=None, alias="PageSize")
 345    starting_token: Optional[str] = Field(default=None, alias="StartingToken")
 346
 347
 348class GetQueryResultsInputRequestModel(BaseModel):
 349    query_execution_id: str = Field(alias="QueryExecutionId")
 350    next_token: Optional[str] = Field(default=None, alias="NextToken")
 351    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 352
 353
 354class GetQueryRuntimeStatisticsInputRequestModel(BaseModel):
 355    query_execution_id: str = Field(alias="QueryExecutionId")
 356
 357
 358class GetSessionRequestModel(BaseModel):
 359    session_id: str = Field(alias="SessionId")
 360
 361
 362class SessionStatisticsModel(BaseModel):
 363    dpu_execution_in_millis: Optional[int] = Field(
 364        default=None, alias="DpuExecutionInMillis"
 365    )
 366
 367
 368class SessionStatusModel(BaseModel):
 369    start_date_time: Optional[datetime] = Field(default=None, alias="StartDateTime")
 370    last_modified_date_time: Optional[datetime] = Field(
 371        default=None, alias="LastModifiedDateTime"
 372    )
 373    end_date_time: Optional[datetime] = Field(default=None, alias="EndDateTime")
 374    idle_since_date_time: Optional[datetime] = Field(
 375        default=None, alias="IdleSinceDateTime"
 376    )
 377    state: Optional[
 378        Literal[
 379            "BUSY",
 380            "CREATED",
 381            "CREATING",
 382            "DEGRADED",
 383            "FAILED",
 384            "IDLE",
 385            "TERMINATED",
 386            "TERMINATING",
 387        ]
 388    ] = Field(default=None, alias="State")
 389    state_change_reason: Optional[str] = Field(default=None, alias="StateChangeReason")
 390
 391
 392class GetSessionStatusRequestModel(BaseModel):
 393    session_id: str = Field(alias="SessionId")
 394
 395
 396class GetTableMetadataInputRequestModel(BaseModel):
 397    catalog_name: str = Field(alias="CatalogName")
 398    database_name: str = Field(alias="DatabaseName")
 399    table_name: str = Field(alias="TableName")
 400
 401
 402class GetWorkGroupInputRequestModel(BaseModel):
 403    work_group: str = Field(alias="WorkGroup")
 404
 405
 406class ImportNotebookInputRequestModel(BaseModel):
 407    work_group: str = Field(alias="WorkGroup")
 408    name: str = Field(alias="Name")
 409    payload: str = Field(alias="Payload")
 410    type: Literal["IPYNB"] = Field(alias="Type")
 411    client_request_token: Optional[str] = Field(
 412        default=None, alias="ClientRequestToken"
 413    )
 414
 415
 416class ListApplicationDPUSizesInputRequestModel(BaseModel):
 417    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 418    next_token: Optional[str] = Field(default=None, alias="NextToken")
 419
 420
 421class ListCalculationExecutionsRequestModel(BaseModel):
 422    session_id: str = Field(alias="SessionId")
 423    state_filter: Optional[
 424        Literal[
 425            "CANCELED",
 426            "CANCELING",
 427            "COMPLETED",
 428            "CREATED",
 429            "CREATING",
 430            "FAILED",
 431            "QUEUED",
 432            "RUNNING",
 433        ]
 434    ] = Field(default=None, alias="StateFilter")
 435    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 436    next_token: Optional[str] = Field(default=None, alias="NextToken")
 437
 438
 439class ListDataCatalogsInputRequestModel(BaseModel):
 440    next_token: Optional[str] = Field(default=None, alias="NextToken")
 441    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 442
 443
 444class ListDatabasesInputRequestModel(BaseModel):
 445    catalog_name: str = Field(alias="CatalogName")
 446    next_token: Optional[str] = Field(default=None, alias="NextToken")
 447    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 448
 449
 450class ListEngineVersionsInputRequestModel(BaseModel):
 451    next_token: Optional[str] = Field(default=None, alias="NextToken")
 452    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 453
 454
 455class ListExecutorsRequestModel(BaseModel):
 456    session_id: str = Field(alias="SessionId")
 457    executor_state_filter: Optional[
 458        Literal[
 459            "CREATED", "CREATING", "FAILED", "REGISTERED", "TERMINATED", "TERMINATING"
 460        ]
 461    ] = Field(default=None, alias="ExecutorStateFilter")
 462    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 463    next_token: Optional[str] = Field(default=None, alias="NextToken")
 464
 465
 466class ListNamedQueriesInputRequestModel(BaseModel):
 467    next_token: Optional[str] = Field(default=None, alias="NextToken")
 468    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 469    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
 470
 471
 472class ListNotebookSessionsRequestModel(BaseModel):
 473    notebook_id: str = Field(alias="NotebookId")
 474    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 475    next_token: Optional[str] = Field(default=None, alias="NextToken")
 476
 477
 478class NotebookSessionSummaryModel(BaseModel):
 479    session_id: Optional[str] = Field(default=None, alias="SessionId")
 480    creation_time: Optional[datetime] = Field(default=None, alias="CreationTime")
 481
 482
 483class ListPreparedStatementsInputRequestModel(BaseModel):
 484    work_group: str = Field(alias="WorkGroup")
 485    next_token: Optional[str] = Field(default=None, alias="NextToken")
 486    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 487
 488
 489class PreparedStatementSummaryModel(BaseModel):
 490    statement_name: Optional[str] = Field(default=None, alias="StatementName")
 491    last_modified_time: Optional[datetime] = Field(
 492        default=None, alias="LastModifiedTime"
 493    )
 494
 495
 496class ListQueryExecutionsInputRequestModel(BaseModel):
 497    next_token: Optional[str] = Field(default=None, alias="NextToken")
 498    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 499    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
 500
 501
 502class ListSessionsRequestModel(BaseModel):
 503    work_group: str = Field(alias="WorkGroup")
 504    state_filter: Optional[
 505        Literal[
 506            "BUSY",
 507            "CREATED",
 508            "CREATING",
 509            "DEGRADED",
 510            "FAILED",
 511            "IDLE",
 512            "TERMINATED",
 513            "TERMINATING",
 514        ]
 515    ] = Field(default=None, alias="StateFilter")
 516    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 517    next_token: Optional[str] = Field(default=None, alias="NextToken")
 518
 519
 520class ListTableMetadataInputRequestModel(BaseModel):
 521    catalog_name: str = Field(alias="CatalogName")
 522    database_name: str = Field(alias="DatabaseName")
 523    expression: Optional[str] = Field(default=None, alias="Expression")
 524    next_token: Optional[str] = Field(default=None, alias="NextToken")
 525    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 526
 527
 528class ListTagsForResourceInputRequestModel(BaseModel):
 529    resource_arn: str = Field(alias="ResourceARN")
 530    next_token: Optional[str] = Field(default=None, alias="NextToken")
 531    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 532
 533
 534class ListWorkGroupsInputRequestModel(BaseModel):
 535    next_token: Optional[str] = Field(default=None, alias="NextToken")
 536    max_results: Optional[int] = Field(default=None, alias="MaxResults")
 537
 538
 539class QueryExecutionContextModel(BaseModel):
 540    database: Optional[str] = Field(default=None, alias="Database")
 541    catalog: Optional[str] = Field(default=None, alias="Catalog")
 542
 543
 544class ResultReuseInformationModel(BaseModel):
 545    reused_previous_result: bool = Field(alias="ReusedPreviousResult")
 546
 547
 548class QueryRuntimeStatisticsRowsModel(BaseModel):
 549    input_rows: Optional[int] = Field(default=None, alias="InputRows")
 550    input_bytes: Optional[int] = Field(default=None, alias="InputBytes")
 551    output_bytes: Optional[int] = Field(default=None, alias="OutputBytes")
 552    output_rows: Optional[int] = Field(default=None, alias="OutputRows")
 553
 554
 555class QueryRuntimeStatisticsTimelineModel(BaseModel):
 556    query_queue_time_in_millis: Optional[int] = Field(
 557        default=None, alias="QueryQueueTimeInMillis"
 558    )
 559    query_planning_time_in_millis: Optional[int] = Field(
 560        default=None, alias="QueryPlanningTimeInMillis"
 561    )
 562    engine_execution_time_in_millis: Optional[int] = Field(
 563        default=None, alias="EngineExecutionTimeInMillis"
 564    )
 565    service_processing_time_in_millis: Optional[int] = Field(
 566        default=None, alias="ServiceProcessingTimeInMillis"
 567    )
 568    total_execution_time_in_millis: Optional[int] = Field(
 569        default=None, alias="TotalExecutionTimeInMillis"
 570    )
 571
 572
 573class QueryStagePlanNodeModel(BaseModel):
 574    name: Optional[str] = Field(default=None, alias="Name")
 575    identifier: Optional[str] = Field(default=None, alias="Identifier")
 576    children: Optional[List[Dict[str, Any]]] = Field(default=None, alias="Children")
 577    remote_sources: Optional[List[str]] = Field(default=None, alias="RemoteSources")
 578
 579
 580class QueryStageModel(BaseModel):
 581    stage_id: Optional[int] = Field(default=None, alias="StageId")
 582    state: Optional[str] = Field(default=None, alias="State")
 583    output_bytes: Optional[int] = Field(default=None, alias="OutputBytes")
 584    output_rows: Optional[int] = Field(default=None, alias="OutputRows")
 585    input_bytes: Optional[int] = Field(default=None, alias="InputBytes")
 586    input_rows: Optional[int] = Field(default=None, alias="InputRows")
 587    execution_time: Optional[int] = Field(default=None, alias="ExecutionTime")
 588    query_stage_plan: Optional[QueryStagePlanNodeModel] = Field(
 589        default=None, alias="QueryStagePlan"
 590    )
 591    sub_stages: Optional[List[Dict[str, Any]]] = Field(default=None, alias="SubStages")
 592
 593
 594class ResultReuseByAgeConfigurationModel(BaseModel):
 595    enabled: bool = Field(alias="Enabled")
 596    max_age_in_minutes: Optional[int] = Field(default=None, alias="MaxAgeInMinutes")
 597
 598
 599class StopCalculationExecutionRequestModel(BaseModel):
 600    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 601
 602
 603class StopQueryExecutionInputRequestModel(BaseModel):
 604    query_execution_id: str = Field(alias="QueryExecutionId")
 605
 606
 607class TerminateSessionRequestModel(BaseModel):
 608    session_id: str = Field(alias="SessionId")
 609
 610
 611class UntagResourceInputRequestModel(BaseModel):
 612    resource_arn: str = Field(alias="ResourceARN")
 613    tag_keys: Sequence[str] = Field(alias="TagKeys")
 614
 615
 616class UpdateDataCatalogInputRequestModel(BaseModel):
 617    name: str = Field(alias="Name")
 618    type: Literal["GLUE", "HIVE", "LAMBDA"] = Field(alias="Type")
 619    description: Optional[str] = Field(default=None, alias="Description")
 620    parameters: Optional[Mapping[str, str]] = Field(default=None, alias="Parameters")
 621
 622
 623class UpdateNamedQueryInputRequestModel(BaseModel):
 624    named_query_id: str = Field(alias="NamedQueryId")
 625    name: str = Field(alias="Name")
 626    query_string: str = Field(alias="QueryString")
 627    description: Optional[str] = Field(default=None, alias="Description")
 628
 629
 630class UpdateNotebookInputRequestModel(BaseModel):
 631    notebook_id: str = Field(alias="NotebookId")
 632    payload: str = Field(alias="Payload")
 633    type: Literal["IPYNB"] = Field(alias="Type")
 634    session_id: Optional[str] = Field(default=None, alias="SessionId")
 635    client_request_token: Optional[str] = Field(
 636        default=None, alias="ClientRequestToken"
 637    )
 638
 639
 640class UpdateNotebookMetadataInputRequestModel(BaseModel):
 641    notebook_id: str = Field(alias="NotebookId")
 642    name: str = Field(alias="Name")
 643    client_request_token: Optional[str] = Field(
 644        default=None, alias="ClientRequestToken"
 645    )
 646
 647
 648class UpdatePreparedStatementInputRequestModel(BaseModel):
 649    statement_name: str = Field(alias="StatementName")
 650    work_group: str = Field(alias="WorkGroup")
 651    query_statement: str = Field(alias="QueryStatement")
 652    description: Optional[str] = Field(default=None, alias="Description")
 653
 654
 655class QueryExecutionStatusModel(BaseModel):
 656    state: Optional[
 657        Literal["CANCELLED", "FAILED", "QUEUED", "RUNNING", "SUCCEEDED"]
 658    ] = Field(default=None, alias="State")
 659    state_change_reason: Optional[str] = Field(default=None, alias="StateChangeReason")
 660    submission_date_time: Optional[datetime] = Field(
 661        default=None, alias="SubmissionDateTime"
 662    )
 663    completion_date_time: Optional[datetime] = Field(
 664        default=None, alias="CompletionDateTime"
 665    )
 666    athena_error: Optional[AthenaErrorModel] = Field(default=None, alias="AthenaError")
 667
 668
 669class CreateNamedQueryOutputModel(BaseModel):
 670    named_query_id: str = Field(alias="NamedQueryId")
 671    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 672
 673
 674class CreateNotebookOutputModel(BaseModel):
 675    notebook_id: str = Field(alias="NotebookId")
 676    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 677
 678
 679class CreatePresignedNotebookUrlResponseModel(BaseModel):
 680    notebook_url: str = Field(alias="NotebookUrl")
 681    auth_token: str = Field(alias="AuthToken")
 682    auth_token_expiration_time: int = Field(alias="AuthTokenExpirationTime")
 683    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 684
 685
 686class GetCalculationExecutionCodeResponseModel(BaseModel):
 687    code_block: str = Field(alias="CodeBlock")
 688    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 689
 690
 691class GetNamedQueryOutputModel(BaseModel):
 692    named_query: NamedQueryModel = Field(alias="NamedQuery")
 693    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 694
 695
 696class ImportNotebookOutputModel(BaseModel):
 697    notebook_id: str = Field(alias="NotebookId")
 698    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 699
 700
 701class ListApplicationDPUSizesOutputModel(BaseModel):
 702    application_dp_usizes: List[ApplicationDPUSizesModel] = Field(
 703        alias="ApplicationDPUSizes"
 704    )
 705    next_token: str = Field(alias="NextToken")
 706    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 707
 708
 709class ListNamedQueriesOutputModel(BaseModel):
 710    named_query_ids: List[str] = Field(alias="NamedQueryIds")
 711    next_token: str = Field(alias="NextToken")
 712    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 713
 714
 715class ListQueryExecutionsOutputModel(BaseModel):
 716    query_execution_ids: List[str] = Field(alias="QueryExecutionIds")
 717    next_token: str = Field(alias="NextToken")
 718    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 719
 720
 721class StartCalculationExecutionResponseModel(BaseModel):
 722    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 723    state: Literal[
 724        "CANCELED",
 725        "CANCELING",
 726        "COMPLETED",
 727        "CREATED",
 728        "CREATING",
 729        "FAILED",
 730        "QUEUED",
 731        "RUNNING",
 732    ] = Field(alias="State")
 733    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 734
 735
 736class StartQueryExecutionOutputModel(BaseModel):
 737    query_execution_id: str = Field(alias="QueryExecutionId")
 738    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 739
 740
 741class StartSessionResponseModel(BaseModel):
 742    session_id: str = Field(alias="SessionId")
 743    state: Literal[
 744        "BUSY",
 745        "CREATED",
 746        "CREATING",
 747        "DEGRADED",
 748        "FAILED",
 749        "IDLE",
 750        "TERMINATED",
 751        "TERMINATING",
 752    ] = Field(alias="State")
 753    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 754
 755
 756class StopCalculationExecutionResponseModel(BaseModel):
 757    state: Literal[
 758        "CANCELED",
 759        "CANCELING",
 760        "COMPLETED",
 761        "CREATED",
 762        "CREATING",
 763        "FAILED",
 764        "QUEUED",
 765        "RUNNING",
 766    ] = Field(alias="State")
 767    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 768
 769
 770class TerminateSessionResponseModel(BaseModel):
 771    state: Literal[
 772        "BUSY",
 773        "CREATED",
 774        "CREATING",
 775        "DEGRADED",
 776        "FAILED",
 777        "IDLE",
 778        "TERMINATED",
 779        "TERMINATING",
 780    ] = Field(alias="State")
 781    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 782
 783
 784class BatchGetNamedQueryOutputModel(BaseModel):
 785    named_queries: List[NamedQueryModel] = Field(alias="NamedQueries")
 786    unprocessed_named_query_ids: List[UnprocessedNamedQueryIdModel] = Field(
 787        alias="UnprocessedNamedQueryIds"
 788    )
 789    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 790
 791
 792class GetPreparedStatementOutputModel(BaseModel):
 793    prepared_statement: PreparedStatementModel = Field(alias="PreparedStatement")
 794    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 795
 796
 797class BatchGetPreparedStatementOutputModel(BaseModel):
 798    prepared_statements: List[PreparedStatementModel] = Field(
 799        alias="PreparedStatements"
 800    )
 801    unprocessed_prepared_statement_names: List[
 802        UnprocessedPreparedStatementNameModel
 803    ] = Field(alias="UnprocessedPreparedStatementNames")
 804    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 805
 806
 807class StartCalculationExecutionRequestModel(BaseModel):
 808    session_id: str = Field(alias="SessionId")
 809    description: Optional[str] = Field(default=None, alias="Description")
 810    calculation_configuration: Optional[CalculationConfigurationModel] = Field(
 811        default=None, alias="CalculationConfiguration"
 812    )
 813    code_block: Optional[str] = Field(default=None, alias="CodeBlock")
 814    client_request_token: Optional[str] = Field(
 815        default=None, alias="ClientRequestToken"
 816    )
 817
 818
 819class CalculationSummaryModel(BaseModel):
 820    calculation_execution_id: Optional[str] = Field(
 821        default=None, alias="CalculationExecutionId"
 822    )
 823    description: Optional[str] = Field(default=None, alias="Description")
 824    status: Optional[CalculationStatusModel] = Field(default=None, alias="Status")
 825
 826
 827class GetCalculationExecutionResponseModel(BaseModel):
 828    calculation_execution_id: str = Field(alias="CalculationExecutionId")
 829    session_id: str = Field(alias="SessionId")
 830    description: str = Field(alias="Description")
 831    working_directory: str = Field(alias="WorkingDirectory")
 832    status: CalculationStatusModel = Field(alias="Status")
 833    statistics: CalculationStatisticsModel = Field(alias="Statistics")
 834    result: CalculationResultModel = Field(alias="Result")
 835    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 836
 837
 838class GetCalculationExecutionStatusResponseModel(BaseModel):
 839    status: CalculationStatusModel = Field(alias="Status")
 840    statistics: CalculationStatisticsModel = Field(alias="Statistics")
 841    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 842
 843
 844class ResultSetMetadataModel(BaseModel):
 845    column_info: Optional[List[ColumnInfoModel]] = Field(
 846        default=None, alias="ColumnInfo"
 847    )
 848
 849
 850class TableMetadataModel(BaseModel):
 851    name: str = Field(alias="Name")
 852    create_time: Optional[datetime] = Field(default=None, alias="CreateTime")
 853    last_access_time: Optional[datetime] = Field(default=None, alias="LastAccessTime")
 854    table_type: Optional[str] = Field(default=None, alias="TableType")
 855    columns: Optional[List[ColumnModel]] = Field(default=None, alias="Columns")
 856    partition_keys: Optional[List[ColumnModel]] = Field(
 857        default=None, alias="PartitionKeys"
 858    )
 859    parameters: Optional[Dict[str, str]] = Field(default=None, alias="Parameters")
 860
 861
 862class CreateDataCatalogInputRequestModel(BaseModel):
 863    name: str = Field(alias="Name")
 864    type: Literal["GLUE", "HIVE", "LAMBDA"] = Field(alias="Type")
 865    description: Optional[str] = Field(default=None, alias="Description")
 866    parameters: Optional[Mapping[str, str]] = Field(default=None, alias="Parameters")
 867    tags: Optional[Sequence[TagModel]] = Field(default=None, alias="Tags")
 868
 869
 870class ListTagsForResourceOutputModel(BaseModel):
 871    tags: List[TagModel] = Field(alias="Tags")
 872    next_token: str = Field(alias="NextToken")
 873    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 874
 875
 876class TagResourceInputRequestModel(BaseModel):
 877    resource_arn: str = Field(alias="ResourceARN")
 878    tags: Sequence[TagModel] = Field(alias="Tags")
 879
 880
 881class ListDataCatalogsOutputModel(BaseModel):
 882    data_catalogs_summary: List[DataCatalogSummaryModel] = Field(
 883        alias="DataCatalogsSummary"
 884    )
 885    next_token: str = Field(alias="NextToken")
 886    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 887
 888
 889class GetDataCatalogOutputModel(BaseModel):
 890    data_catalog: DataCatalogModel = Field(alias="DataCatalog")
 891    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 892
 893
 894class GetDatabaseOutputModel(BaseModel):
 895    database: DatabaseModel = Field(alias="Database")
 896    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 897
 898
 899class ListDatabasesOutputModel(BaseModel):
 900    database_list: List[DatabaseModel] = Field(alias="DatabaseList")
 901    next_token: str = Field(alias="NextToken")
 902    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 903
 904
 905class RowModel(BaseModel):
 906    data: Optional[List[DatumModel]] = Field(default=None, alias="Data")
 907
 908
 909class ResultConfigurationModel(BaseModel):
 910    output_location: Optional[str] = Field(default=None, alias="OutputLocation")
 911    encryption_configuration: Optional[EncryptionConfigurationModel] = Field(
 912        default=None, alias="EncryptionConfiguration"
 913    )
 914    expected_bucket_owner: Optional[str] = Field(
 915        default=None, alias="ExpectedBucketOwner"
 916    )
 917    acl_configuration: Optional[AclConfigurationModel] = Field(
 918        default=None, alias="AclConfiguration"
 919    )
 920
 921
 922class ResultConfigurationUpdatesModel(BaseModel):
 923    output_location: Optional[str] = Field(default=None, alias="OutputLocation")
 924    remove_output_location: Optional[bool] = Field(
 925        default=None, alias="RemoveOutputLocation"
 926    )
 927    encryption_configuration: Optional[EncryptionConfigurationModel] = Field(
 928        default=None, alias="EncryptionConfiguration"
 929    )
 930    remove_encryption_configuration: Optional[bool] = Field(
 931        default=None, alias="RemoveEncryptionConfiguration"
 932    )
 933    expected_bucket_owner: Optional[str] = Field(
 934        default=None, alias="ExpectedBucketOwner"
 935    )
 936    remove_expected_bucket_owner: Optional[bool] = Field(
 937        default=None, alias="RemoveExpectedBucketOwner"
 938    )
 939    acl_configuration: Optional[AclConfigurationModel] = Field(
 940        default=None, alias="AclConfiguration"
 941    )
 942    remove_acl_configuration: Optional[bool] = Field(
 943        default=None, alias="RemoveAclConfiguration"
 944    )
 945
 946
 947class SessionConfigurationModel(BaseModel):
 948    execution_role: Optional[str] = Field(default=None, alias="ExecutionRole")
 949    working_directory: Optional[str] = Field(default=None, alias="WorkingDirectory")
 950    idle_timeout_seconds: Optional[int] = Field(
 951        default=None, alias="IdleTimeoutSeconds"
 952    )
 953    encryption_configuration: Optional[EncryptionConfigurationModel] = Field(
 954        default=None, alias="EncryptionConfiguration"
 955    )
 956
 957
 958class StartSessionRequestModel(BaseModel):
 959    work_group: str = Field(alias="WorkGroup")
 960    engine_configuration: EngineConfigurationModel = Field(alias="EngineConfiguration")
 961    description: Optional[str] = Field(default=None, alias="Description")
 962    notebook_version: Optional[str] = Field(default=None, alias="NotebookVersion")
 963    session_idle_timeout_in_minutes: Optional[int] = Field(
 964        default=None, alias="SessionIdleTimeoutInMinutes"
 965    )
 966    client_request_token: Optional[str] = Field(
 967        default=None, alias="ClientRequestToken"
 968    )
 969
 970
 971class ListEngineVersionsOutputModel(BaseModel):
 972    engine_versions: List[EngineVersionModel] = Field(alias="EngineVersions")
 973    next_token: str = Field(alias="NextToken")
 974    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 975
 976
 977class WorkGroupSummaryModel(BaseModel):
 978    name: Optional[str] = Field(default=None, alias="Name")
 979    state: Optional[Literal["DISABLED", "ENABLED"]] = Field(default=None, alias="State")
 980    description: Optional[str] = Field(default=None, alias="Description")
 981    creation_time: Optional[datetime] = Field(default=None, alias="CreationTime")
 982    engine_version: Optional[EngineVersionModel] = Field(
 983        default=None, alias="EngineVersion"
 984    )
 985
 986
 987class ListExecutorsResponseModel(BaseModel):
 988    session_id: str = Field(alias="SessionId")
 989    next_token: str = Field(alias="NextToken")
 990    executors_summary: List[ExecutorsSummaryModel] = Field(alias="ExecutorsSummary")
 991    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 992
 993
 994class ExportNotebookOutputModel(BaseModel):
 995    notebook_metadata: NotebookMetadataModel = Field(alias="NotebookMetadata")
 996    payload: str = Field(alias="Payload")
 997    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
 998
 999
1000class GetNotebookMetadataOutputModel(BaseModel):
1001    notebook_metadata: NotebookMetadataModel = Field(alias="NotebookMetadata")
1002    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1003
1004
1005class ListNotebookMetadataOutputModel(BaseModel):
1006    next_token: str = Field(alias="NextToken")
1007    notebook_metadata_list: List[NotebookMetadataModel] = Field(
1008        alias="NotebookMetadataList"
1009    )
1010    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1011
1012
1013class ListNotebookMetadataInputRequestModel(BaseModel):
1014    work_group: str = Field(alias="WorkGroup")
1015    filters: Optional[FilterDefinitionModel] = Field(default=None, alias="Filters")
1016    next_token: Optional[str] = Field(default=None, alias="NextToken")
1017    max_results: Optional[int] = Field(default=None, alias="MaxResults")
1018
1019
1020class GetQueryResultsInputGetQueryResultsPaginateModel(BaseModel):
1021    query_execution_id: str = Field(alias="QueryExecutionId")
1022    pagination_config: Optional[PaginatorConfigModel] = Field(
1023        default=None, alias="PaginationConfig"
1024    )
1025
1026
1027class ListDataCatalogsInputListDataCatalogsPaginateModel(BaseModel):
1028    pagination_config: Optional[PaginatorConfigModel] = Field(
1029        default=None, alias="PaginationConfig"
1030    )
1031
1032
1033class ListDatabasesInputListDatabasesPaginateModel(BaseModel):
1034    catalog_name: str = Field(alias="CatalogName")
1035    pagination_config: Optional[PaginatorConfigModel] = Field(
1036        default=None, alias="PaginationConfig"
1037    )
1038
1039
1040class ListNamedQueriesInputListNamedQueriesPaginateModel(BaseModel):
1041    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
1042    pagination_config: Optional[PaginatorConfigModel] = Field(
1043        default=None, alias="PaginationConfig"
1044    )
1045
1046
1047class ListQueryExecutionsInputListQueryExecutionsPaginateModel(BaseModel):
1048    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
1049    pagination_config: Optional[PaginatorConfigModel] = Field(
1050        default=None, alias="PaginationConfig"
1051    )
1052
1053
1054class ListTableMetadataInputListTableMetadataPaginateModel(BaseModel):
1055    catalog_name: str = Field(alias="CatalogName")
1056    database_name: str = Field(alias="DatabaseName")
1057    expression: Optional[str] = Field(default=None, alias="Expression")
1058    pagination_config: Optional[PaginatorConfigModel] = Field(
1059        default=None, alias="PaginationConfig"
1060    )
1061
1062
1063class ListTagsForResourceInputListTagsForResourcePaginateModel(BaseModel):
1064    resource_arn: str = Field(alias="ResourceARN")
1065    pagination_config: Optional[PaginatorConfigModel] = Field(
1066        default=None, alias="PaginationConfig"
1067    )
1068
1069
1070class GetSessionStatusResponseModel(BaseModel):
1071    session_id: str = Field(alias="SessionId")
1072    status: SessionStatusModel = Field(alias="Status")
1073    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1074
1075
1076class SessionSummaryModel(BaseModel):
1077    session_id: Optional[str] = Field(default=None, alias="SessionId")
1078    description: Optional[str] = Field(default=None, alias="Description")
1079    engine_version: Optional[EngineVersionModel] = Field(
1080        default=None, alias="EngineVersion"
1081    )
1082    notebook_version: Optional[str] = Field(default=None, alias="NotebookVersion")
1083    status: Optional[SessionStatusModel] = Field(default=None, alias="Status")
1084
1085
1086class ListNotebookSessionsResponseModel(BaseModel):
1087    notebook_sessions_list: List[NotebookSessionSummaryModel] = Field(
1088        alias="NotebookSessionsList"
1089    )
1090    next_token: str = Field(alias="NextToken")
1091    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1092
1093
1094class ListPreparedStatementsOutputModel(BaseModel):
1095    prepared_statements: List[PreparedStatementSummaryModel] = Field(
1096        alias="PreparedStatements"
1097    )
1098    next_token: str = Field(alias="NextToken")
1099    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1100
1101
1102class QueryExecutionStatisticsModel(BaseModel):
1103    engine_execution_time_in_millis: Optional[int] = Field(
1104        default=None, alias="EngineExecutionTimeInMillis"
1105    )
1106    data_scanned_in_bytes: Optional[int] = Field(
1107        default=None, alias="DataScannedInBytes"
1108    )
1109    data_manifest_location: Optional[str] = Field(
1110        default=None, alias="DataManifestLocation"
1111    )
1112    total_execution_time_in_millis: Optional[int] = Field(
1113        default=None, alias="TotalExecutionTimeInMillis"
1114    )
1115    query_queue_time_in_millis: Optional[int] = Field(
1116        default=None, alias="QueryQueueTimeInMillis"
1117    )
1118    query_planning_time_in_millis: Optional[int] = Field(
1119        default=None, alias="QueryPlanningTimeInMillis"
1120    )
1121    service_processing_time_in_millis: Optional[int] = Field(
1122        default=None, alias="ServiceProcessingTimeInMillis"
1123    )
1124    result_reuse_information: Optional[ResultReuseInformationModel] = Field(
1125        default=None, alias="ResultReuseInformation"
1126    )
1127
1128
1129class QueryRuntimeStatisticsModel(BaseModel):
1130    timeline: Optional[QueryRuntimeStatisticsTimelineModel] = Field(
1131        default=None, alias="Timeline"
1132    )
1133    rows: Optional[QueryRuntimeStatisticsRowsModel] = Field(default=None, alias="Rows")
1134    output_stage: Optional[QueryStageModel] = Field(default=None, alias="OutputStage")
1135
1136
1137class ResultReuseConfigurationModel(BaseModel):
1138    result_reuse_by_age_configuration: Optional[
1139        ResultReuseByAgeConfigurationModel
1140    ] = Field(default=None, alias="ResultReuseByAgeConfiguration")
1141
1142
1143class ListCalculationExecutionsResponseModel(BaseModel):
1144    next_token: str = Field(alias="NextToken")
1145    calculations: List[CalculationSummaryModel] = Field(alias="Calculations")
1146    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1147
1148
1149class GetTableMetadataOutputModel(BaseModel):
1150    table_metadata: TableMetadataModel = Field(alias="TableMetadata")
1151    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1152
1153
1154class ListTableMetadataOutputModel(BaseModel):
1155    table_metadata_list: List[TableMetadataModel] = Field(alias="TableMetadataList")
1156    next_token: str = Field(alias="NextToken")
1157    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1158
1159
1160class ResultSetModel(BaseModel):
1161    rows: Optional[List[RowModel]] = Field(default=None, alias="Rows")
1162    result_set_metadata: Optional[ResultSetMetadataModel] = Field(
1163        default=None, alias="ResultSetMetadata"
1164    )
1165
1166
1167class WorkGroupConfigurationModel(BaseModel):
1168    result_configuration: Optional[ResultConfigurationModel] = Field(
1169        default=None, alias="ResultConfiguration"
1170    )
1171    enforce_work_group_configuration: Optional[bool] = Field(
1172        default=None, alias="EnforceWorkGroupConfiguration"
1173    )
1174    publish_cloud_watch_metrics_enabled: Optional[bool] = Field(
1175        default=None, alias="PublishCloudWatchMetricsEnabled"
1176    )
1177    bytes_scanned_cutoff_per_query: Optional[int] = Field(
1178        default=None, alias="BytesScannedCutoffPerQuery"
1179    )
1180    requester_pays_enabled: Optional[bool] = Field(
1181        default=None, alias="RequesterPaysEnabled"
1182    )
1183    engine_version: Optional[EngineVersionModel] = Field(
1184        default=None, alias="EngineVersion"
1185    )
1186    additional_configuration: Optional[str] = Field(
1187        default=None, alias="AdditionalConfiguration"
1188    )
1189    execution_role: Optional[str] = Field(default=None, alias="ExecutionRole")
1190    customer_content_encryption_configuration: Optional[
1191        CustomerContentEncryptionConfigurationModel
1192    ] = Field(default=None, alias="CustomerContentEncryptionConfiguration")
1193
1194
1195class WorkGroupConfigurationUpdatesModel(BaseModel):
1196    enforce_work_group_configuration: Optional[bool] = Field(
1197        default=None, alias="EnforceWorkGroupConfiguration"
1198    )
1199    result_configuration_updates: Optional[ResultConfigurationUpdatesModel] = Field(
1200        default=None, alias="ResultConfigurationUpdates"
1201    )
1202    publish_cloud_watch_metrics_enabled: Optional[bool] = Field(
1203        default=None, alias="PublishCloudWatchMetricsEnabled"
1204    )
1205    bytes_scanned_cutoff_per_query: Optional[int] = Field(
1206        default=None, alias="BytesScannedCutoffPerQuery"
1207    )
1208    remove_bytes_scanned_cutoff_per_query: Optional[bool] = Field(
1209        default=None, alias="RemoveBytesScannedCutoffPerQuery"
1210    )
1211    requester_pays_enabled: Optional[bool] = Field(
1212        default=None, alias="RequesterPaysEnabled"
1213    )
1214    engine_version: Optional[EngineVersionModel] = Field(
1215        default=None, alias="EngineVersion"
1216    )
1217    remove_customer_content_encryption_configuration: Optional[bool] = Field(
1218        default=None, alias="RemoveCustomerContentEncryptionConfiguration"
1219    )
1220    additional_configuration: Optional[str] = Field(
1221        default=None, alias="AdditionalConfiguration"
1222    )
1223    execution_role: Optional[str] = Field(default=None, alias="ExecutionRole")
1224    customer_content_encryption_configuration: Optional[
1225        CustomerContentEncryptionConfigurationModel
1226    ] = Field(default=None, alias="CustomerContentEncryptionConfiguration")
1227
1228
1229class GetSessionResponseModel(BaseModel):
1230    session_id: str = Field(alias="SessionId")
1231    description: str = Field(alias="Description")
1232    work_group: str = Field(alias="WorkGroup")
1233    engine_version: str = Field(alias="EngineVersion")
1234    engine_configuration: EngineConfigurationModel = Field(alias="EngineConfiguration")
1235    notebook_version: str = Field(alias="NotebookVersion")
1236    session_configuration: SessionConfigurationModel = Field(
1237        alias="SessionConfiguration"
1238    )
1239    status: SessionStatusModel = Field(alias="Status")
1240    statistics: SessionStatisticsModel = Field(alias="Statistics")
1241    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1242
1243
1244class ListWorkGroupsOutputModel(BaseModel):
1245    work_groups: List[WorkGroupSummaryModel] = Field(alias="WorkGroups")
1246    next_token: str = Field(alias="NextToken")
1247    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1248
1249
1250class ListSessionsResponseModel(BaseModel):
1251    next_token: str = Field(alias="NextToken")
1252    sessions: List[SessionSummaryModel] = Field(alias="Sessions")
1253    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1254
1255
1256class GetQueryRuntimeStatisticsOutputModel(BaseModel):
1257    query_runtime_statistics: QueryRuntimeStatisticsModel = Field(
1258        alias="QueryRuntimeStatistics"
1259    )
1260    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1261
1262
1263class QueryExecutionModel(BaseModel):
1264    query_execution_id: Optional[str] = Field(default=None, alias="QueryExecutionId")
1265    query: Optional[str] = Field(default=None, alias="Query")
1266    statement_type: Optional[Literal["DDL", "DML", "UTILITY"]] = Field(
1267        default=None, alias="StatementType"
1268    )
1269    result_configuration: Optional[ResultConfigurationModel] = Field(
1270        default=None, alias="ResultConfiguration"
1271    )
1272    result_reuse_configuration: Optional[ResultReuseConfigurationModel] = Field(
1273        default=None, alias="ResultReuseConfiguration"
1274    )
1275    query_execution_context: Optional[QueryExecutionContextModel] = Field(
1276        default=None, alias="QueryExecutionContext"
1277    )
1278    status: Optional[QueryExecutionStatusModel] = Field(default=None, alias="Status")
1279    statistics: Optional[QueryExecutionStatisticsModel] = Field(
1280        default=None, alias="Statistics"
1281    )
1282    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
1283    engine_version: Optional[EngineVersionModel] = Field(
1284        default=None, alias="EngineVersion"
1285    )
1286    execution_parameters: Optional[List[str]] = Field(
1287        default=None, alias="ExecutionParameters"
1288    )
1289
1290
1291class StartQueryExecutionInputRequestModel(BaseModel):
1292    query_string: str = Field(alias="QueryString")
1293    client_request_token: Optional[str] = Field(
1294        default=None, alias="ClientRequestToken"
1295    )
1296    query_execution_context: Optional[QueryExecutionContextModel] = Field(
1297        default=None, alias="QueryExecutionContext"
1298    )
1299    result_configuration: Optional[ResultConfigurationModel] = Field(
1300        default=None, alias="ResultConfiguration"
1301    )
1302    work_group: Optional[str] = Field(default=None, alias="WorkGroup")
1303    execution_parameters: Optional[Sequence[str]] = Field(
1304        default=None, alias="ExecutionParameters"
1305    )
1306    result_reuse_configuration: Optional[ResultReuseConfigurationModel] = Field(
1307        default=None, alias="ResultReuseConfiguration"
1308    )
1309
1310
1311class GetQueryResultsOutputModel(BaseModel):
1312    update_count: int = Field(alias="UpdateCount")
1313    result_set: ResultSetModel = Field(alias="ResultSet")
1314    next_token: str = Field(alias="NextToken")
1315    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1316
1317
1318class CreateWorkGroupInputRequestModel(BaseModel):
1319    name: str = Field(alias="Name")
1320    configuration: Optional[WorkGroupConfigurationModel] = Field(
1321        default=None, alias="Configuration"
1322    )
1323    description: Optional[str] = Field(default=None, alias="Description")
1324    tags: Optional[Sequence[TagModel]] = Field(default=None, alias="Tags")
1325
1326
1327class WorkGroupModel(BaseModel):
1328    name: str = Field(alias="Name")
1329    state: Optional[Literal["DISABLED", "ENABLED"]] = Field(default=None, alias="State")
1330    configuration: Optional[WorkGroupConfigurationModel] = Field(
1331        default=None, alias="Configuration"
1332    )
1333    description: Optional[str] = Field(default=None, alias="Description")
1334    creation_time: Optional[datetime] = Field(default=None, alias="CreationTime")
1335
1336
1337class UpdateWorkGroupInputRequestModel(BaseModel):
1338    work_group: str = Field(alias="WorkGroup")
1339    description: Optional[str] = Field(default=None, alias="Description")
1340    configuration_updates: Optional[WorkGroupConfigurationUpdatesModel] = Field(
1341        default=None, alias="ConfigurationUpdates"
1342    )
1343    state: Optional[Literal["DISABLED", "ENABLED"]] = Field(default=None, alias="State")
1344
1345
1346class BatchGetQueryExecutionOutputModel(BaseModel):
1347    query_executions: List[QueryExecutionModel] = Field(alias="QueryExecutions")
1348    unprocessed_query_execution_ids: List[UnprocessedQueryExecutionIdModel] = Field(
1349        alias="UnprocessedQueryExecutionIds"
1350    )
1351    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1352
1353
1354class GetQueryExecutionOutputModel(BaseModel):
1355    query_execution: QueryExecutionModel = Field(alias="QueryExecution")
1356    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
1357
1358
1359class GetWorkGroupOutputModel(BaseModel):
1360    work_group: WorkGroupModel = Field(alias="WorkGroup")
1361    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")