mypydantic.models.cleanrooms

  1# Model Generated: Thu Mar  2 21:56:16 2023
  2
  3from __future__ import annotations
  4
  5from datetime import datetime
  6from typing import Dict, List, Literal, Optional, Sequence
  7
  8from pydantic import Field
  9
 10from mypydantic.models.base_model import BaseModel
 11
 12
 13class AggregateColumnModel(BaseModel):
 14    column_names: Sequence[str] = Field(alias="columnNames")
 15    function: Literal["AVG", "COUNT", "COUNT_DISTINCT", "SUM", "SUM_DISTINCT"] = Field(
 16        alias="function"
 17    )
 18
 19
 20class AggregationConstraintModel(BaseModel):
 21    column_name: str = Field(alias="columnName")
 22    minimum: int = Field(alias="minimum")
 23    type: Literal["COUNT_DISTINCT"] = Field(alias="type")
 24
 25
 26class AnalysisRuleListModel(BaseModel):
 27    join_columns: Sequence[str] = Field(alias="joinColumns")
 28    list_columns: Sequence[str] = Field(alias="listColumns")
 29
 30
 31class BatchGetSchemaErrorModel(BaseModel):
 32    name: str = Field(alias="name")
 33    code: str = Field(alias="code")
 34    message: str = Field(alias="message")
 35
 36
 37class BatchGetSchemaInputRequestModel(BaseModel):
 38    collaboration_identifier: str = Field(alias="collaborationIdentifier")
 39    names: Sequence[str] = Field(alias="names")
 40
 41
 42class ResponseMetadataModel(BaseModel):
 43    request_id: str = Field(alias="RequestId")
 44    host_id: str = Field(alias="HostId")
 45    http_status_code: int = Field(alias="HTTPStatusCode")
 46    http_headers: Dict[str, str] = Field(alias="HTTPHeaders")
 47    retry_attempts: int = Field(alias="RetryAttempts")
 48
 49
 50class CollaborationSummaryModel(BaseModel):
 51    id: str = Field(alias="id")
 52    arn: str = Field(alias="arn")
 53    name: str = Field(alias="name")
 54    creator_account_id: str = Field(alias="creatorAccountId")
 55    creator_display_name: str = Field(alias="creatorDisplayName")
 56    create_time: datetime = Field(alias="createTime")
 57    update_time: datetime = Field(alias="updateTime")
 58    member_status: Literal["ACTIVE", "INVITED", "LEFT", "REMOVED"] = Field(
 59        alias="memberStatus"
 60    )
 61    membership_id: Optional[str] = Field(default=None, alias="membershipId")
 62    membership_arn: Optional[str] = Field(default=None, alias="membershipArn")
 63
 64
 65class DataEncryptionMetadataModel(BaseModel):
 66    allow_cleartext: bool = Field(alias="allowCleartext")
 67    allow_duplicates: bool = Field(alias="allowDuplicates")
 68    allow_joins_on_columns_with_different_names: bool = Field(
 69        alias="allowJoinsOnColumnsWithDifferentNames"
 70    )
 71    preserve_nulls: bool = Field(alias="preserveNulls")
 72
 73
 74class ColumnModel(BaseModel):
 75    name: str = Field(alias="name")
 76    type: str = Field(alias="type")
 77
 78
 79class ConfiguredTableAssociationSummaryModel(BaseModel):
 80    configured_table_id: str = Field(alias="configuredTableId")
 81    membership_id: str = Field(alias="membershipId")
 82    membership_arn: str = Field(alias="membershipArn")
 83    name: str = Field(alias="name")
 84    create_time: datetime = Field(alias="createTime")
 85    update_time: datetime = Field(alias="updateTime")
 86    id: str = Field(alias="id")
 87    arn: str = Field(alias="arn")
 88
 89
 90class ConfiguredTableAssociationModel(BaseModel):
 91    arn: str = Field(alias="arn")
 92    id: str = Field(alias="id")
 93    configured_table_id: str = Field(alias="configuredTableId")
 94    configured_table_arn: str = Field(alias="configuredTableArn")
 95    membership_id: str = Field(alias="membershipId")
 96    membership_arn: str = Field(alias="membershipArn")
 97    role_arn: str = Field(alias="roleArn")
 98    name: str = Field(alias="name")
 99    create_time: datetime = Field(alias="createTime")
100    update_time: datetime = Field(alias="updateTime")
101    description: Optional[str] = Field(default=None, alias="description")
102
103
104class ConfiguredTableSummaryModel(BaseModel):
105    id: str = Field(alias="id")
106    arn: str = Field(alias="arn")
107    name: str = Field(alias="name")
108    create_time: datetime = Field(alias="createTime")
109    update_time: datetime = Field(alias="updateTime")
110    analysis_rule_types: List[Literal["AGGREGATION", "LIST"]] = Field(
111        alias="analysisRuleTypes"
112    )
113    analysis_method: Literal["DIRECT_QUERY"] = Field(alias="analysisMethod")
114
115
116class MemberSpecificationModel(BaseModel):
117    account_id: str = Field(alias="accountId")
118    member_abilities: Sequence[Literal["CAN_QUERY", "CAN_RECEIVE_RESULTS"]] = Field(
119        alias="memberAbilities"
120    )
121    display_name: str = Field(alias="displayName")
122
123
124class CreateConfiguredTableAssociationInputRequestModel(BaseModel):
125    name: str = Field(alias="name")
126    membership_identifier: str = Field(alias="membershipIdentifier")
127    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
128    role_arn: str = Field(alias="roleArn")
129    description: Optional[str] = Field(default=None, alias="description")
130
131
132class CreateMembershipInputRequestModel(BaseModel):
133    collaboration_identifier: str = Field(alias="collaborationIdentifier")
134    query_log_status: Literal["DISABLED", "ENABLED"] = Field(alias="queryLogStatus")
135
136
137class MembershipModel(BaseModel):
138    id: str = Field(alias="id")
139    arn: str = Field(alias="arn")
140    collaboration_arn: str = Field(alias="collaborationArn")
141    collaboration_id: str = Field(alias="collaborationId")
142    collaboration_creator_account_id: str = Field(alias="collaborationCreatorAccountId")
143    collaboration_creator_display_name: str = Field(
144        alias="collaborationCreatorDisplayName"
145    )
146    collaboration_name: str = Field(alias="collaborationName")
147    create_time: datetime = Field(alias="createTime")
148    update_time: datetime = Field(alias="updateTime")
149    status: Literal["ACTIVE", "COLLABORATION_DELETED", "REMOVED"] = Field(
150        alias="status"
151    )
152    member_abilities: List[Literal["CAN_QUERY", "CAN_RECEIVE_RESULTS"]] = Field(
153        alias="memberAbilities"
154    )
155    query_log_status: Literal["DISABLED", "ENABLED"] = Field(alias="queryLogStatus")
156
157
158class DeleteCollaborationInputRequestModel(BaseModel):
159    collaboration_identifier: str = Field(alias="collaborationIdentifier")
160
161
162class DeleteConfiguredTableAnalysisRuleInputRequestModel(BaseModel):
163    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
164    analysis_rule_type: Literal["AGGREGATION", "LIST"] = Field(alias="analysisRuleType")
165
166
167class DeleteConfiguredTableAssociationInputRequestModel(BaseModel):
168    configured_table_association_identifier: str = Field(
169        alias="configuredTableAssociationIdentifier"
170    )
171    membership_identifier: str = Field(alias="membershipIdentifier")
172
173
174class DeleteConfiguredTableInputRequestModel(BaseModel):
175    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
176
177
178class DeleteMemberInputRequestModel(BaseModel):
179    collaboration_identifier: str = Field(alias="collaborationIdentifier")
180    account_id: str = Field(alias="accountId")
181
182
183class DeleteMembershipInputRequestModel(BaseModel):
184    membership_identifier: str = Field(alias="membershipIdentifier")
185
186
187class GetCollaborationInputRequestModel(BaseModel):
188    collaboration_identifier: str = Field(alias="collaborationIdentifier")
189
190
191class GetConfiguredTableAnalysisRuleInputRequestModel(BaseModel):
192    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
193    analysis_rule_type: Literal["AGGREGATION", "LIST"] = Field(alias="analysisRuleType")
194
195
196class GetConfiguredTableAssociationInputRequestModel(BaseModel):
197    configured_table_association_identifier: str = Field(
198        alias="configuredTableAssociationIdentifier"
199    )
200    membership_identifier: str = Field(alias="membershipIdentifier")
201
202
203class GetConfiguredTableInputRequestModel(BaseModel):
204    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
205
206
207class GetMembershipInputRequestModel(BaseModel):
208    membership_identifier: str = Field(alias="membershipIdentifier")
209
210
211class GetProtectedQueryInputRequestModel(BaseModel):
212    membership_identifier: str = Field(alias="membershipIdentifier")
213    protected_query_identifier: str = Field(alias="protectedQueryIdentifier")
214
215
216class GetSchemaAnalysisRuleInputRequestModel(BaseModel):
217    collaboration_identifier: str = Field(alias="collaborationIdentifier")
218    name: str = Field(alias="name")
219    type: Literal["AGGREGATION", "LIST"] = Field(alias="type")
220
221
222class GetSchemaInputRequestModel(BaseModel):
223    collaboration_identifier: str = Field(alias="collaborationIdentifier")
224    name: str = Field(alias="name")
225
226
227class GlueTableReferenceModel(BaseModel):
228    table_name: str = Field(alias="tableName")
229    database_name: str = Field(alias="databaseName")
230
231
232class PaginatorConfigModel(BaseModel):
233    max_items: Optional[int] = Field(default=None, alias="MaxItems")
234    page_size: Optional[int] = Field(default=None, alias="PageSize")
235    starting_token: Optional[str] = Field(default=None, alias="StartingToken")
236
237
238class ListCollaborationsInputRequestModel(BaseModel):
239    next_token: Optional[str] = Field(default=None, alias="nextToken")
240    max_results: Optional[int] = Field(default=None, alias="maxResults")
241    member_status: Optional[Literal["ACTIVE", "INVITED"]] = Field(
242        default=None, alias="memberStatus"
243    )
244
245
246class ListConfiguredTableAssociationsInputRequestModel(BaseModel):
247    membership_identifier: str = Field(alias="membershipIdentifier")
248    next_token: Optional[str] = Field(default=None, alias="nextToken")
249    max_results: Optional[int] = Field(default=None, alias="maxResults")
250
251
252class ListConfiguredTablesInputRequestModel(BaseModel):
253    next_token: Optional[str] = Field(default=None, alias="nextToken")
254    max_results: Optional[int] = Field(default=None, alias="maxResults")
255
256
257class ListMembersInputRequestModel(BaseModel):
258    collaboration_identifier: str = Field(alias="collaborationIdentifier")
259    next_token: Optional[str] = Field(default=None, alias="nextToken")
260    max_results: Optional[int] = Field(default=None, alias="maxResults")
261
262
263class MemberSummaryModel(BaseModel):
264    account_id: str = Field(alias="accountId")
265    status: Literal["ACTIVE", "INVITED", "LEFT", "REMOVED"] = Field(alias="status")
266    display_name: str = Field(alias="displayName")
267    abilities: List[Literal["CAN_QUERY", "CAN_RECEIVE_RESULTS"]] = Field(
268        alias="abilities"
269    )
270    create_time: datetime = Field(alias="createTime")
271    update_time: datetime = Field(alias="updateTime")
272    membership_id: Optional[str] = Field(default=None, alias="membershipId")
273    membership_arn: Optional[str] = Field(default=None, alias="membershipArn")
274
275
276class ListMembershipsInputRequestModel(BaseModel):
277    next_token: Optional[str] = Field(default=None, alias="nextToken")
278    max_results: Optional[int] = Field(default=None, alias="maxResults")
279    status: Optional[Literal["ACTIVE", "COLLABORATION_DELETED", "REMOVED"]] = Field(
280        default=None, alias="status"
281    )
282
283
284class MembershipSummaryModel(BaseModel):
285    id: str = Field(alias="id")
286    arn: str = Field(alias="arn")
287    collaboration_arn: str = Field(alias="collaborationArn")
288    collaboration_id: str = Field(alias="collaborationId")
289    collaboration_creator_account_id: str = Field(alias="collaborationCreatorAccountId")
290    collaboration_creator_display_name: str = Field(
291        alias="collaborationCreatorDisplayName"
292    )
293    collaboration_name: str = Field(alias="collaborationName")
294    create_time: datetime = Field(alias="createTime")
295    update_time: datetime = Field(alias="updateTime")
296    status: Literal["ACTIVE", "COLLABORATION_DELETED", "REMOVED"] = Field(
297        alias="status"
298    )
299    member_abilities: List[Literal["CAN_QUERY", "CAN_RECEIVE_RESULTS"]] = Field(
300        alias="memberAbilities"
301    )
302
303
304class ListProtectedQueriesInputRequestModel(BaseModel):
305    membership_identifier: str = Field(alias="membershipIdentifier")
306    status: Optional[
307        Literal[
308            "CANCELLED",
309            "CANCELLING",
310            "FAILED",
311            "STARTED",
312            "SUBMITTED",
313            "SUCCESS",
314            "TIMED_OUT",
315        ]
316    ] = Field(default=None, alias="status")
317    next_token: Optional[str] = Field(default=None, alias="nextToken")
318    max_results: Optional[int] = Field(default=None, alias="maxResults")
319
320
321class ProtectedQuerySummaryModel(BaseModel):
322    id: str = Field(alias="id")
323    membership_id: str = Field(alias="membershipId")
324    membership_arn: str = Field(alias="membershipArn")
325    create_time: datetime = Field(alias="createTime")
326    status: Literal[
327        "CANCELLED",
328        "CANCELLING",
329        "FAILED",
330        "STARTED",
331        "SUBMITTED",
332        "SUCCESS",
333        "TIMED_OUT",
334    ] = Field(alias="status")
335
336
337class ListSchemasInputRequestModel(BaseModel):
338    collaboration_identifier: str = Field(alias="collaborationIdentifier")
339    schema_type: Optional[Literal["TABLE"]] = Field(default=None, alias="schemaType")
340    next_token: Optional[str] = Field(default=None, alias="nextToken")
341    max_results: Optional[int] = Field(default=None, alias="maxResults")
342
343
344class SchemaSummaryModel(BaseModel):
345    name: str = Field(alias="name")
346    type: Literal["TABLE"] = Field(alias="type")
347    creator_account_id: str = Field(alias="creatorAccountId")
348    create_time: datetime = Field(alias="createTime")
349    update_time: datetime = Field(alias="updateTime")
350    collaboration_id: str = Field(alias="collaborationId")
351    collaboration_arn: str = Field(alias="collaborationArn")
352    analysis_rule_types: List[Literal["AGGREGATION", "LIST"]] = Field(
353        alias="analysisRuleTypes"
354    )
355    analysis_method: Optional[Literal["DIRECT_QUERY"]] = Field(
356        default=None, alias="analysisMethod"
357    )
358
359
360class ProtectedQueryErrorModel(BaseModel):
361    message: str = Field(alias="message")
362    code: str = Field(alias="code")
363
364
365class ProtectedQueryS3OutputConfigurationModel(BaseModel):
366    result_format: Literal["CSV", "PARQUET"] = Field(alias="resultFormat")
367    bucket: str = Field(alias="bucket")
368    key_prefix: Optional[str] = Field(default=None, alias="keyPrefix")
369
370
371class ProtectedQueryS3OutputModel(BaseModel):
372    location: str = Field(alias="location")
373
374
375class ProtectedQuerySQLParametersModel(BaseModel):
376    query_string: str = Field(alias="queryString")
377
378
379class ProtectedQueryStatisticsModel(BaseModel):
380    total_duration_in_millis: Optional[int] = Field(
381        default=None, alias="totalDurationInMillis"
382    )
383
384
385class UpdateCollaborationInputRequestModel(BaseModel):
386    collaboration_identifier: str = Field(alias="collaborationIdentifier")
387    name: Optional[str] = Field(default=None, alias="name")
388    description: Optional[str] = Field(default=None, alias="description")
389
390
391class UpdateConfiguredTableAssociationInputRequestModel(BaseModel):
392    configured_table_association_identifier: str = Field(
393        alias="configuredTableAssociationIdentifier"
394    )
395    membership_identifier: str = Field(alias="membershipIdentifier")
396    description: Optional[str] = Field(default=None, alias="description")
397    role_arn: Optional[str] = Field(default=None, alias="roleArn")
398
399
400class UpdateConfiguredTableInputRequestModel(BaseModel):
401    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
402    name: Optional[str] = Field(default=None, alias="name")
403    description: Optional[str] = Field(default=None, alias="description")
404
405
406class UpdateMembershipInputRequestModel(BaseModel):
407    membership_identifier: str = Field(alias="membershipIdentifier")
408    query_log_status: Optional[Literal["DISABLED", "ENABLED"]] = Field(
409        default=None, alias="queryLogStatus"
410    )
411
412
413class UpdateProtectedQueryInputRequestModel(BaseModel):
414    membership_identifier: str = Field(alias="membershipIdentifier")
415    protected_query_identifier: str = Field(alias="protectedQueryIdentifier")
416    target_status: Literal["CANCELLED"] = Field(alias="targetStatus")
417
418
419class AnalysisRuleAggregationModel(BaseModel):
420    aggregate_columns: Sequence[AggregateColumnModel] = Field(alias="aggregateColumns")
421    join_columns: Sequence[str] = Field(alias="joinColumns")
422    dimension_columns: Sequence[str] = Field(alias="dimensionColumns")
423    scalar_functions: Sequence[
424        Literal[
425            "ABS",
426            "CAST",
427            "CEILING",
428            "COALESCE",
429            "FLOOR",
430            "LN",
431            "LOG",
432            "LOWER",
433            "ROUND",
434            "RTRIM",
435            "SQRT",
436            "TRUNC",
437            "UPPER",
438        ]
439    ] = Field(alias="scalarFunctions")
440    output_constraints: Sequence[AggregationConstraintModel] = Field(
441        alias="outputConstraints"
442    )
443    join_required: Optional[Literal["QUERY_RUNNER"]] = Field(
444        default=None, alias="joinRequired"
445    )
446
447
448class ListCollaborationsOutputModel(BaseModel):
449    next_token: str = Field(alias="nextToken")
450    collaboration_list: List[CollaborationSummaryModel] = Field(
451        alias="collaborationList"
452    )
453    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
454
455
456class CollaborationModel(BaseModel):
457    id: str = Field(alias="id")
458    arn: str = Field(alias="arn")
459    name: str = Field(alias="name")
460    creator_account_id: str = Field(alias="creatorAccountId")
461    creator_display_name: str = Field(alias="creatorDisplayName")
462    create_time: datetime = Field(alias="createTime")
463    update_time: datetime = Field(alias="updateTime")
464    member_status: Literal["ACTIVE", "INVITED", "LEFT", "REMOVED"] = Field(
465        alias="memberStatus"
466    )
467    query_log_status: Literal["DISABLED", "ENABLED"] = Field(alias="queryLogStatus")
468    description: Optional[str] = Field(default=None, alias="description")
469    membership_id: Optional[str] = Field(default=None, alias="membershipId")
470    membership_arn: Optional[str] = Field(default=None, alias="membershipArn")
471    data_encryption_metadata: Optional[DataEncryptionMetadataModel] = Field(
472        default=None, alias="dataEncryptionMetadata"
473    )
474
475
476class SchemaModel(BaseModel):
477    columns: List[ColumnModel] = Field(alias="columns")
478    partition_keys: List[ColumnModel] = Field(alias="partitionKeys")
479    analysis_rule_types: List[Literal["AGGREGATION", "LIST"]] = Field(
480        alias="analysisRuleTypes"
481    )
482    creator_account_id: str = Field(alias="creatorAccountId")
483    name: str = Field(alias="name")
484    collaboration_id: str = Field(alias="collaborationId")
485    collaboration_arn: str = Field(alias="collaborationArn")
486    description: str = Field(alias="description")
487    create_time: datetime = Field(alias="createTime")
488    update_time: datetime = Field(alias="updateTime")
489    type: Literal["TABLE"] = Field(alias="type")
490    analysis_method: Optional[Literal["DIRECT_QUERY"]] = Field(
491        default=None, alias="analysisMethod"
492    )
493
494
495class ListConfiguredTableAssociationsOutputModel(BaseModel):
496    configured_table_association_summaries: List[
497        ConfiguredTableAssociationSummaryModel
498    ] = Field(alias="configuredTableAssociationSummaries")
499    next_token: str = Field(alias="nextToken")
500    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
501
502
503class CreateConfiguredTableAssociationOutputModel(BaseModel):
504    configured_table_association: ConfiguredTableAssociationModel = Field(
505        alias="configuredTableAssociation"
506    )
507    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
508
509
510class GetConfiguredTableAssociationOutputModel(BaseModel):
511    configured_table_association: ConfiguredTableAssociationModel = Field(
512        alias="configuredTableAssociation"
513    )
514    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
515
516
517class UpdateConfiguredTableAssociationOutputModel(BaseModel):
518    configured_table_association: ConfiguredTableAssociationModel = Field(
519        alias="configuredTableAssociation"
520    )
521    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
522
523
524class ListConfiguredTablesOutputModel(BaseModel):
525    configured_table_summaries: List[ConfiguredTableSummaryModel] = Field(
526        alias="configuredTableSummaries"
527    )
528    next_token: str = Field(alias="nextToken")
529    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
530
531
532class CreateCollaborationInputRequestModel(BaseModel):
533    members: Sequence[MemberSpecificationModel] = Field(alias="members")
534    name: str = Field(alias="name")
535    description: str = Field(alias="description")
536    creator_member_abilities: Sequence[
537        Literal["CAN_QUERY", "CAN_RECEIVE_RESULTS"]
538    ] = Field(alias="creatorMemberAbilities")
539    creator_display_name: str = Field(alias="creatorDisplayName")
540    query_log_status: Literal["DISABLED", "ENABLED"] = Field(alias="queryLogStatus")
541    data_encryption_metadata: Optional[DataEncryptionMetadataModel] = Field(
542        default=None, alias="dataEncryptionMetadata"
543    )
544
545
546class CreateMembershipOutputModel(BaseModel):
547    membership: MembershipModel = Field(alias="membership")
548    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
549
550
551class GetMembershipOutputModel(BaseModel):
552    membership: MembershipModel = Field(alias="membership")
553    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
554
555
556class UpdateMembershipOutputModel(BaseModel):
557    membership: MembershipModel = Field(alias="membership")
558    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
559
560
561class TableReferenceModel(BaseModel):
562    glue: Optional[GlueTableReferenceModel] = Field(default=None, alias="glue")
563
564
565class ListCollaborationsInputListCollaborationsPaginateModel(BaseModel):
566    member_status: Optional[Literal["ACTIVE", "INVITED"]] = Field(
567        default=None, alias="memberStatus"
568    )
569    pagination_config: Optional[PaginatorConfigModel] = Field(
570        default=None, alias="PaginationConfig"
571    )
572
573
574class ListConfiguredTableAssociationsInputListConfiguredTableAssociationsPaginateModel(
575    BaseModel
576):
577    membership_identifier: str = Field(alias="membershipIdentifier")
578    pagination_config: Optional[PaginatorConfigModel] = Field(
579        default=None, alias="PaginationConfig"
580    )
581
582
583class ListConfiguredTablesInputListConfiguredTablesPaginateModel(BaseModel):
584    pagination_config: Optional[PaginatorConfigModel] = Field(
585        default=None, alias="PaginationConfig"
586    )
587
588
589class ListMembersInputListMembersPaginateModel(BaseModel):
590    collaboration_identifier: str = Field(alias="collaborationIdentifier")
591    pagination_config: Optional[PaginatorConfigModel] = Field(
592        default=None, alias="PaginationConfig"
593    )
594
595
596class ListMembershipsInputListMembershipsPaginateModel(BaseModel):
597    status: Optional[Literal["ACTIVE", "COLLABORATION_DELETED", "REMOVED"]] = Field(
598        default=None, alias="status"
599    )
600    pagination_config: Optional[PaginatorConfigModel] = Field(
601        default=None, alias="PaginationConfig"
602    )
603
604
605class ListProtectedQueriesInputListProtectedQueriesPaginateModel(BaseModel):
606    membership_identifier: str = Field(alias="membershipIdentifier")
607    status: Optional[
608        Literal[
609            "CANCELLED",
610            "CANCELLING",
611            "FAILED",
612            "STARTED",
613            "SUBMITTED",
614            "SUCCESS",
615            "TIMED_OUT",
616        ]
617    ] = Field(default=None, alias="status")
618    pagination_config: Optional[PaginatorConfigModel] = Field(
619        default=None, alias="PaginationConfig"
620    )
621
622
623class ListSchemasInputListSchemasPaginateModel(BaseModel):
624    collaboration_identifier: str = Field(alias="collaborationIdentifier")
625    schema_type: Optional[Literal["TABLE"]] = Field(default=None, alias="schemaType")
626    pagination_config: Optional[PaginatorConfigModel] = Field(
627        default=None, alias="PaginationConfig"
628    )
629
630
631class ListMembersOutputModel(BaseModel):
632    next_token: str = Field(alias="nextToken")
633    member_summaries: List[MemberSummaryModel] = Field(alias="memberSummaries")
634    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
635
636
637class ListMembershipsOutputModel(BaseModel):
638    next_token: str = Field(alias="nextToken")
639    membership_summaries: List[MembershipSummaryModel] = Field(
640        alias="membershipSummaries"
641    )
642    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
643
644
645class ListProtectedQueriesOutputModel(BaseModel):
646    next_token: str = Field(alias="nextToken")
647    protected_queries: List[ProtectedQuerySummaryModel] = Field(
648        alias="protectedQueries"
649    )
650    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
651
652
653class ListSchemasOutputModel(BaseModel):
654    schema_summaries: List[SchemaSummaryModel] = Field(alias="schemaSummaries")
655    next_token: str = Field(alias="nextToken")
656    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
657
658
659class ProtectedQueryOutputConfigurationModel(BaseModel):
660    s3: Optional[ProtectedQueryS3OutputConfigurationModel] = Field(
661        default=None, alias="s3"
662    )
663
664
665class ProtectedQueryOutputModel(BaseModel):
666    s3: Optional[ProtectedQueryS3OutputModel] = Field(default=None, alias="s3")
667
668
669class AnalysisRulePolicyV1Model(BaseModel):
670    list: Optional[AnalysisRuleListModel] = Field(default=None, alias="list")
671    aggregation: Optional[AnalysisRuleAggregationModel] = Field(
672        default=None, alias="aggregation"
673    )
674
675
676class ConfiguredTableAnalysisRulePolicyV1Model(BaseModel):
677    list: Optional[AnalysisRuleListModel] = Field(default=None, alias="list")
678    aggregation: Optional[AnalysisRuleAggregationModel] = Field(
679        default=None, alias="aggregation"
680    )
681
682
683class CreateCollaborationOutputModel(BaseModel):
684    collaboration: CollaborationModel = Field(alias="collaboration")
685    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
686
687
688class GetCollaborationOutputModel(BaseModel):
689    collaboration: CollaborationModel = Field(alias="collaboration")
690    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
691
692
693class UpdateCollaborationOutputModel(BaseModel):
694    collaboration: CollaborationModel = Field(alias="collaboration")
695    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
696
697
698class BatchGetSchemaOutputModel(BaseModel):
699    schemas: List[SchemaModel] = Field(alias="schemas")
700    errors: List[BatchGetSchemaErrorModel] = Field(alias="errors")
701    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
702
703
704class GetSchemaOutputModel(BaseModel):
705    schema_: SchemaModel = Field(alias="schema")
706    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
707
708
709class ConfiguredTableModel(BaseModel):
710    id: str = Field(alias="id")
711    arn: str = Field(alias="arn")
712    name: str = Field(alias="name")
713    table_reference: TableReferenceModel = Field(alias="tableReference")
714    create_time: datetime = Field(alias="createTime")
715    update_time: datetime = Field(alias="updateTime")
716    analysis_rule_types: List[Literal["AGGREGATION", "LIST"]] = Field(
717        alias="analysisRuleTypes"
718    )
719    analysis_method: Literal["DIRECT_QUERY"] = Field(alias="analysisMethod")
720    allowed_columns: List[str] = Field(alias="allowedColumns")
721    description: Optional[str] = Field(default=None, alias="description")
722
723
724class CreateConfiguredTableInputRequestModel(BaseModel):
725    name: str = Field(alias="name")
726    table_reference: TableReferenceModel = Field(alias="tableReference")
727    allowed_columns: Sequence[str] = Field(alias="allowedColumns")
728    analysis_method: Literal["DIRECT_QUERY"] = Field(alias="analysisMethod")
729    description: Optional[str] = Field(default=None, alias="description")
730
731
732class ProtectedQueryResultConfigurationModel(BaseModel):
733    output_configuration: ProtectedQueryOutputConfigurationModel = Field(
734        alias="outputConfiguration"
735    )
736
737
738class ProtectedQueryResultModel(BaseModel):
739    output: ProtectedQueryOutputModel = Field(alias="output")
740
741
742class AnalysisRulePolicyModel(BaseModel):
743    v1: Optional[AnalysisRulePolicyV1Model] = Field(default=None, alias="v1")
744
745
746class ConfiguredTableAnalysisRulePolicyModel(BaseModel):
747    v1: Optional[ConfiguredTableAnalysisRulePolicyV1Model] = Field(
748        default=None, alias="v1"
749    )
750
751
752class CreateConfiguredTableOutputModel(BaseModel):
753    configured_table: ConfiguredTableModel = Field(alias="configuredTable")
754    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
755
756
757class GetConfiguredTableOutputModel(BaseModel):
758    configured_table: ConfiguredTableModel = Field(alias="configuredTable")
759    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
760
761
762class UpdateConfiguredTableOutputModel(BaseModel):
763    configured_table: ConfiguredTableModel = Field(alias="configuredTable")
764    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
765
766
767class StartProtectedQueryInputRequestModel(BaseModel):
768    type: Literal["SQL"] = Field(alias="type")
769    membership_identifier: str = Field(alias="membershipIdentifier")
770    sql_parameters: ProtectedQuerySQLParametersModel = Field(alias="sqlParameters")
771    result_configuration: ProtectedQueryResultConfigurationModel = Field(
772        alias="resultConfiguration"
773    )
774
775
776class ProtectedQueryModel(BaseModel):
777    id: str = Field(alias="id")
778    membership_id: str = Field(alias="membershipId")
779    membership_arn: str = Field(alias="membershipArn")
780    create_time: datetime = Field(alias="createTime")
781    sql_parameters: ProtectedQuerySQLParametersModel = Field(alias="sqlParameters")
782    status: Literal[
783        "CANCELLED",
784        "CANCELLING",
785        "FAILED",
786        "STARTED",
787        "SUBMITTED",
788        "SUCCESS",
789        "TIMED_OUT",
790    ] = Field(alias="status")
791    result_configuration: ProtectedQueryResultConfigurationModel = Field(
792        alias="resultConfiguration"
793    )
794    statistics: Optional[ProtectedQueryStatisticsModel] = Field(
795        default=None, alias="statistics"
796    )
797    result: Optional[ProtectedQueryResultModel] = Field(default=None, alias="result")
798    error: Optional[ProtectedQueryErrorModel] = Field(default=None, alias="error")
799
800
801class AnalysisRuleModel(BaseModel):
802    collaboration_id: str = Field(alias="collaborationId")
803    type: Literal["AGGREGATION", "LIST"] = Field(alias="type")
804    name: str = Field(alias="name")
805    create_time: datetime = Field(alias="createTime")
806    update_time: datetime = Field(alias="updateTime")
807    policy: AnalysisRulePolicyModel = Field(alias="policy")
808
809
810class ConfiguredTableAnalysisRuleModel(BaseModel):
811    configured_table_id: str = Field(alias="configuredTableId")
812    configured_table_arn: str = Field(alias="configuredTableArn")
813    policy: ConfiguredTableAnalysisRulePolicyModel = Field(alias="policy")
814    type: Literal["AGGREGATION", "LIST"] = Field(alias="type")
815    create_time: datetime = Field(alias="createTime")
816    update_time: datetime = Field(alias="updateTime")
817
818
819class CreateConfiguredTableAnalysisRuleInputRequestModel(BaseModel):
820    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
821    analysis_rule_type: Literal["AGGREGATION", "LIST"] = Field(alias="analysisRuleType")
822    analysis_rule_policy: ConfiguredTableAnalysisRulePolicyModel = Field(
823        alias="analysisRulePolicy"
824    )
825
826
827class UpdateConfiguredTableAnalysisRuleInputRequestModel(BaseModel):
828    configured_table_identifier: str = Field(alias="configuredTableIdentifier")
829    analysis_rule_type: Literal["AGGREGATION", "LIST"] = Field(alias="analysisRuleType")
830    analysis_rule_policy: ConfiguredTableAnalysisRulePolicyModel = Field(
831        alias="analysisRulePolicy"
832    )
833
834
835class GetProtectedQueryOutputModel(BaseModel):
836    protected_query: ProtectedQueryModel = Field(alias="protectedQuery")
837    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
838
839
840class StartProtectedQueryOutputModel(BaseModel):
841    protected_query: ProtectedQueryModel = Field(alias="protectedQuery")
842    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
843
844
845class UpdateProtectedQueryOutputModel(BaseModel):
846    protected_query: ProtectedQueryModel = Field(alias="protectedQuery")
847    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
848
849
850class GetSchemaAnalysisRuleOutputModel(BaseModel):
851    analysis_rule: AnalysisRuleModel = Field(alias="analysisRule")
852    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
853
854
855class CreateConfiguredTableAnalysisRuleOutputModel(BaseModel):
856    analysis_rule: ConfiguredTableAnalysisRuleModel = Field(alias="analysisRule")
857    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
858
859
860class GetConfiguredTableAnalysisRuleOutputModel(BaseModel):
861    analysis_rule: ConfiguredTableAnalysisRuleModel = Field(alias="analysisRule")
862    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")
863
864
865class UpdateConfiguredTableAnalysisRuleOutputModel(BaseModel):
866    analysis_rule: ConfiguredTableAnalysisRuleModel = Field(alias="analysisRule")
867    response_metadata: ResponseMetadataModel = Field(alias="ResponseMetadata")