dZddlmZddlmZddlmZddlmZddlmZdZ dZ d Z d Z dd Z dd ZGddeZy )z(Utilities for Cloud Pub/Sub Schemas API.)absolute_import)division)unicode_literals)apisInvalidArgumentErrorctdS)NzRThe schema name must include a revision-id of the format: SCHEMA_NAME@REVISION_ID.r,lib/googlecloudsdk/api_lib/pubsub/schemas.pyNoRevisionIdSpecifiedr s " r c@|jd}t|dkDS)N@)splitlen) schema_reffind_ids r CheckRevisionIdInSchemaPathr s!   S !' W r ctdS)Nz7The schema type must be either AVRO or PROTOCOL-BUFFER.rr r r InvalidSchemaTyper%s ? r c|j}|dk(s|dk(r |jjjS|dk(r |jjjSt )Nzprotocol-bufferprotocol_bufferavro)lowerSchemaTypeValueValuesEnumPROTOCOL_BUFFERAVROr)messages schema_typetype_strs r ParseSchemaTyper#+s^    ( ""h2C&C ?? . . > >>6 ?? . . 3 33r c2tjdd|S)Npubsubv1no_http)rGetClientInstancer's r r)r)4s  $ @@r Nc6|xs t}|jSN)r)MESSAGES_MODULE)clients r GetMessagesModuler.8s  (&(&   r c*eZdZdZddZdZdZdZy) SchemasClientz4Client for schemas service in the Cloud Pub/Sub API.Nc|xs t|_|xs t||_|jj|_yr+)r)r-r.r projects_schemas_service)selfr-r s r __init__zSchemasClient.__init__@s5/-/DK9 1& 9DMKK00DMr c |jj|t|j||}|jj|jj ||}|j j |S)aCommits a revision for a Schema. Args: schema_ref: The full schema_path. schema_definition: The new schema definition to commit. schema_type: The type of the schema (avro or protocol-buffer). Returns: Schema: the committed Schema revision )nametype definition)schema)commitSchemaRequestr7)r rr#"PubsubProjectsSchemasCommitRequestCommitSchemaRequestr3Commit)r4rschema_definitionr!r: commit_reqs r r>zSchemasClient.CommitEs|]] ! !  T]]K 8$"F AA MM==V=L BJ ==   ++r c|jj|jj||}|jj |S)a5Rolls back to a previous schema revision. Args: schema_ref: The path of the schema to rollback. revision_id: The revision_id to rollback to. Returns: Schema: the new schema revision you have rolled back to. Raises: InvalidArgumentError: If no revision_id is provided. ) revisionId)rollbackSchemaRequestr7)r $PubsubProjectsSchemasRollbackRequestRollbackSchemaRequestr3Rollback)r4r revision_id rollback_reqs r rFzSchemasClient.Rollback[sU==EE"mmAA"B  FL == ! !, //r ct|s t|jj|}|jj |S)zDeletes a schema revision. Args: schema_ref: The path of the schema, with the revision_id. Returns: Schema: the deleted schema revision. )r7)rr r *PubsubProjectsSchemasDeleteRevisionRequestr3DeleteRevision)r4rdelete_revision_reqs r rKzSchemasClient.DeleteRevisionpsO 'z 2 ! ## @@ A  == ' '(; <rFrKr r r r0r0=s<1 ,,0*=r r0)Fr+)rP __future__rrrgooglecloudsdk.api_lib.utilr&googlecloudsdk.command_lib.pubsub.utilrr rrr#r)r.objectr0r r r rUsF/&',G  A D=FD=r