.0dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd l m Z dd lmZd ZddZddZdZddZ ddZdZ ddZdZddZGddeZy )z%Spanner database sessions API helper.)absolute_import)division)unicode_literals)encoding) exceptions) extra_types) http_wrapper) list_pager)apis) QueryHasDmlc|jdk(rtjj|t j |S)z5Wrap http_wrapper.CheckResponse to skip retry on 501.i) status_codeapitools_exceptions HttpError FromResponser CheckResponse)responses 7lib/googlecloudsdk/api_lib/spanner/database_sessions.pyrrs8 S  ' ' 4 4X >>  # #H --NcJtddd}tjdd}|!|j|j }nC|j |j |}|j||j }|jj|S)zCreate a database session. Args: database_ref: String, The database in which the new session is created. creator_role: String, The database role which created this session. Returns: Newly created session. spannerv1N)database) creatorRole)session)createSessionRequestr) _GetClientInstancer GetMessagesModule6SpannerProjectsInstancesDatabasesSessionsCreateRequest RelativeNameCreateSessionRequestSession%projects_instances_databases_sessionsCreate) database_ref creator_roleclientmsgsreqcreate_session_requests rr$r$%s it 4&   4 0$  E E**, F .C"66  678  E E3**, F .C  5 5 <  @&,& -rctdd|}tjdd}t|t ||||||} |j |j | } |jj| } t|rK|durG|j| j} t|g| jjj| S)aExecute an SQL command. Args: sql: String, The SQL to execute. query_mode: String, The mode in which to run the query. Must be one of 'NORMAL', 'PLAN', 'PROFILE', 'WITH_STATS', or 'WITH_PLAN_AND_STATS'. session_ref: Session, Indicates that the repo should be created if it does not exist. read_only_options: The ReadOnly message for a read-only request. It is ignored in a DML request. request_options: The RequestOptions message that contains the priority. enable_partitioned_dml: Boolean, whether partitioned dml is enabled. http_timeout_sec: int, Maximum time in seconds to wait for the SQL query to complete. Returns: (Repo) The capture repository. rr)rexecuteSqlRequestF)metadata)rr r_RegisterCustomMessageCodec_GetQueryRequest:SpannerProjectsInstancesDatabasesSessionsExecuteSqlRequestr r# ExecuteSqlr ResultSetrACommit transactionid) sql query_moder9read_only_optionsrequest_optionsenable_partitioned_dmlr;r'r(execute_sql_requestr)resp result_sets rrErE]s* i/? @&   4 0$d#(   GG&&(>? +rcxd}fd}tj||jjy)zHRegister custom message code. Args: msgs: Spanner v1 messages. chtjtj|jSN)entriesrJsonProtoEncoder JsonArrayentrymsgs r_ToJsonz,_RegisterCustomMessageCodec.._ToJsons'  ' 'cii0 22rcvjjtj|jS)NrY)rFRowsValueListEntryrJsonProtoDecoderrU)datar(s r _FromJsonz._RegisterCustomMessageCodec.._FromJsons3 >> , ,**4088 - ::rencoderdecoderN)rRegisterCustomMessageCodecrFr_)r(r\rbs` rrBrBs62:*(%%y* .. + +-rcntjdd}|dur t|}nbt|r3|j |j }|j |}n$|j |}|j |}|j|||jj||S) a_Formats the request based on whether the statement contains DML. Args: sql: String, The SQL to execute. query_mode: String, The mode in which to run the query. Must be one of 'NORMAL', 'PLAN', 'PROFILE', 'WITH_STATS', or 'WITH_PLAN_AND_STATS'. session_ref: Reference to the session. read_only_options: The ReadOnly message for a read-only request. It is ignored in a DML request. request_options: The RequestOptions message that contains the priority. enable_partitioned_dml: Boolean, whether partitioned dml is enabled. Returns: ExecuteSqlRequest parameters rrT readWrite)begin)readOnly) singleUse)rJrequestOptions queryModerH) r r_GetPartitionedDmlTransactionr TransactionOptions ReadWriteTransactionSelectorExecuteSqlRequestQueryModeValueValuesEnum) rJrKr9rLrMrNr(rHtransaction_optionss rrCrCs*    4 0$t#/   ??rcZ|jj|j||S)a*Constructs an UPDATE mutation, which updates a row in a table. Args: table: String, the name of the table. data: An ordered dictionary where the keys are the column names and values are the column values to be updated. Returns: An update mutation operation. )updaterrs rUpdatezMutationFactory.UpdaterrcZ|jj|j||S)zConstructs a DELETE mutation, which deletes a row in a table. Args: table: String, the name of the table. keys: String list, the primary key values of the row to delete. Returns: A delete mutation operation. )delete)r(r _GetDelete)rrkeyss rr8zMutationFactory.Deletes' 88  CNN5$$?  @@rcd}tj|d|jjj|j |}|Dcgc]}|j }}|Dcgc]}|j}}|jj||j|jjj |gScc}wcc}w)zFConstructs Write object, which is needed for insert/update operations.chtjtj|jSrTrVrZs rr\z*MutationFactory._GetWrite.._ToJson0'  ) )    244rNrcr^)columnsrvalues) rrfr(WriteValuesValueListEntry GetJsonDatacol_name col_valuer6)rrrar\ json_columnscoljson_column_namesjson_column_valuess rrzMutationFactory._GetWrite,s4'H''' HHNN / /1$$T*L1=>#>3?@>!jj33:L3MN  PP?@s C-Ccld}tj|d|jjj|jj|jjj |j |g}|jj |j|S)z?Constructs Delete object, which is needed for delete operation.chtjtj|jSrTrVrZs rr\z+MutationFactory._GetDelete.._ToJsonFrrNrcr^)r)rkeySet)rrfr(KeySetKeysValueListEntry GetJsonKeysr8r6)rrrr\key_sets rrzMutationFactory._GetDeleteAs 4'H''' HHOO . .0hhoo **1B1B41H*I$oG 88??G? <rs,&'%>()',>.B0  B@D