"idZddlZddlZddlZddlmZ ddlZdZn #e $rdZYnwxYw ddl Z dZ n #e $rdZ YnwxYwddl mZmZmZmZddlmZmZmZmZmZmZdd lmZmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#d Z$ej%d Z&Gdde'Z(Gdde'Z)Gdde'Z*Gdde'Z+dS)z4Implementation of the X protocol for MySQL servers. N)BytesIOTF)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingError) ExprParser build_expr build_scalarbuild_bool_scalarbuild_int_scalarbuild_unsigned_int_scalar)encode_to_bytesget_item_or_attr)Column)CRUD_PREPARE_MAPPINGSERVER_MESSAGESPROTOBUF_REPEATED_TYPESMessage mysqlxpb_enumimysqlxc$eZdZdZdZdZdZdS) CompressorzImplements compression/decompression using `zstd_stream`, `lz4_message` and `deflate_stream` algorithms. Args: algorithm (str): Compression algorithm. .. versionadded:: 8.0.21 c||_|dkr2tj|_tj|_dS|dkr2t j|_t j|_dSd|_d|_dS)N zstd_streamdeflate_stream) _algorithmzstdZstdCompressor _compressobjZstdDecompressor_decompressobjzlib compressobj decompressobjself algorithms G/srv/django_bis/venv311/lib/python3.11/site-packages/mysqlx/protocol.py__init__zCompressor.__init__Js#  % % $ 3 5 5D "&"7"9"9D    * * * $ 0 2 2D "&"4"6"6D    $D "&D   c|jdkr|j|S|jdkr{tj5}|}|||z }||z }dddn #1swxYwY|S|j|}||jtj z }|S)zCompresses data and returns it. Args: data (str, bytes or buffer object): Data to be compressed. Returns: bytes: Compressed data. r lz4_messageN) rr compresslz4frameLZ4FrameCompressorbeginflushr# Z_SYNC_FLUSH)r'data compressor compresseds r)r.zCompressor.compressVs ?m + +$--d33 3 ?m + +--// 1:'--// j11$777 j..000  1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 &//55 d'--d.?@@@ sABB#&B#c|jdkr|j|S|jdkrMtj5}||}dddn #1swxYwY|S|j|}||jtjz }|S)zDecompresses a frame of data and returns it as a string of bytes. Args: data (str, bytes or buffer object): Data to be compressed. Returns: bytes: Decompresssed data. rr-N) rr" decompressr/r0LZ4FrameDecompressorr3r#r4)r'r5 decompressor decompresseds r)r9zCompressor.decompressms ?m + +&11$77 7 ?m + +//11 =\+66t<<  = = = = = = = = = = = = = = = *55d;; +11$2CDDD sA11A58A5N)__name__ __module__ __qualname____doc__r*r.r9r+r)rr@sK ' ' '.r+rc0eZdZdZdZdZdZdZdZdS) MessageReaderzImplements a Message Reader. Args: socket_stream (mysqlx.connection.SocketStream): `SocketStream` object. .. versionadded:: 8.0.21 c>||_d|_d|_g|_dSN)_stream _compressor_msg _msg_queuer' socket_streams r)r*zMessageReader.__init__s#$  r+c|jr|jdStjd|jd\}}|dkrt d|j|dz }|tvr"td ||dkr|d kr| Stj ||}|d kr|d }t|j|d }d}||kr~tjd|d\}} ||dz } |jtj | | ||d zz }||k~|jr|jdndS|S)aReads X Protocol messages from the stream and returns a :class:`mysqlx.protobuf.Message` object. Raises: :class:`mysqlx.ProgrammingError`: If e connected server does not have the MySQL X protocol plugin enabled. Returns: mysqlx.protobuf.Message: MySQL X Protobuf Message. r>zJJKK K     4 4%%'' '/ MJJ    )*= > T-8899MNNOOFO!$555M%Q88' h ++lQ&677&&/'BBDDD?? ? r+c6|rt|nd|_dS)zCreates a :class:`mysqlx.protocol.Compressor` object based on the compression algorithm. Args: algorithm (str): Compression algorithm. .. versionadded:: 8.0.21 NrrGr&s r)set_compressionzMessageReader.set_compressions$5>G:i0004r+N) r=r>r?r@r*r[rirkrnrAr+r)rCrCso ---^ $ $ $    H H H H Hr+rCc$eZdZdZdZdZdZdS) MessageWriterzImplements a Message Writer. Args: socket_stream (mysqlx.connection.SocketStream): `SocketStream` object. .. versionadded:: 8.0.21 c"||_d|_dSrE)rFrGrJs r)r*zMessageWriter.__init__s$ r+c||}|jrh|tkr\t|}t jd|dz|}|jd||g}td}||d<|dz|d<td}||d<dt| d d t| g} td } t jdt| dz| }|j d|| gd St|}t jd|dz|}|j d||gd S) zWrite message. Args: msg_type (int): The message type. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. rMrr+zMysqlx.Connection.Compressionclient_messagesrNrRrSNz&Mysqlx.ClientMessages.Type.COMPRESSION) byte_sizerG_COMPRESSION_THRESHOLDrserialize_to_stringrVpackr.joinrserialize_partial_to_stringrlenrFsendall) r'rerhmsg_sizemsg_strheaderr7msg_first_fields msg_payloadoutput msg_comp_ids r) write_messagezMessageWriter.write_messages==%%   >+A A A%c&=&=&?&?@@G[1 h??F)22388VWG:i0004r+N)r=r>r?r@r*rrnrAr+r)rprpsQ    > > >DHHHHHr+rpceZdZdZdZedZdZdZd$dZ dZ d Z d Z d Z d Zd%dZdZdZdZdZdZdZdZdZdZdZdZd&dZdZdZdZdZdZ d Z!d!Z"d"Z#d&d#Z$d S)'ProtocolzImplements the MySQL X Protocol. Args: read (mysqlx.protocol.MessageReader): A Message Reader object. writer (mysqlx.protocol.MessageWriter): A Message Writer object. .. versionchanged:: 8.0.21 c>||_||_d|_g|_dSrE)_reader_writer_compression_algorithm _warnings)r'readerwriters r)r*zProtocol.__init__(s#  &*#r+c|jS)z'str: The compresion algorithm. )r)r's r)compression_algorithmzProtocol.compression_algorithm.s **r+cR|jr||d<|jr-|d||jr-|d||jr||d<dSdS)zApply filter. Args: msg (mysqlx.protobuf.Message): The MySQL X Protobuf Message. stmt (Statement): A `Statement` based type object. criteriaordergroupinggrouping_criteriaN) has_whereget_where_exprhas_sortextend get_sort_expr has_group_by get_grouping has_having get_having)r'rhstmts r) _apply_filterzProtocol._apply_filter4s > 4"1133C O = 6 L   2 2 4 4 5 5 5   8  O " "4#4#4#6#6 7 7 7 ? 9'+'8'8C# $ $ $ 9 9r+c :t|tr5td|}tdd|}tdd|St|trtddt |St|t rD|d krtddt |Stddt|St|trst|d kr`|\}}td || | }td | g}tdd |St|ts7t|ttfrt|d trg}|D]} g} | D]Q\} }td | | | }| | Rtd | } tdd | } || td}||d<tdd|St|tr{g} |D]Q\} }td | | | }| | Rtd | } tdd | } | SdS)zCreate any. Args: arg (object): Arbitrary object. Returns: mysqlx.protobuf.Message: MySQL X Protobuf Message. zMysqlx.Datatypes.Scalar.String)valuezMysqlx.Datatypes.Scalar)typev_stringMysqlx.Datatypes.Anyr)rscalarr#Mysqlx.Datatypes.Object.ObjectFieldkeyrMysqlx.Datatypes.ObjectfldrobjzMysqlx.Datatypes.Arrayr)rarrayN) isinstancestrrboolr intr rtupler{ _create_any get_messagedictlistitemsr])r'argrrarg_key arg_valueobj_fldr array_valuesrobj_fldsrmsg_objmsg_anyrhs r)rzProtocol._create_anyDsq c3  - z,Protocol._get_binding_args..sG%DL$7$7$C$C$E$E$E"..u55AACCr+Nc&g|] }|SrArA).0r build_values r) z.Protocol._get_binding_args..s#===5KK&&===r+z;The number of bind parameters and placeholders do not matchz-Unable to find placeholder for parameter: {0}) get_bindingsget_binding_mapr{rrrZ) r'rrbindings binding_mapcountargsnamerposrs ` ` @r)_get_binding_argszProtocol._get_binding_args~sDDDDD $$&&**,,   ====H=== =K  v~ CMM ! !"$?@@ @#>>++ + +KD%;&&&(88>t FFFd#C# E**DII r+c|ddkrtjd|d}|j|jt d|j|j||j |j|jdS|ddkrtjd|ddS|ddkrtjd |d}|d td kr'| d |d DdSt|d ttr|d dn|d }|d tdkr%|t!|ddS|d tdkr'|t!|ddSdSdS)zProcess frame. Args: msg (mysqlx.protobuf.Message): A MySQL X Protobuf Message. result (Result): A `Result` based type object. rrzMysqlx.Notice.WarningrSz:Protocol.process_frame Received Warning Notice code %s: %srz$Mysqlx.Notice.SessionVariableChangedrz!Mysqlx.Notice.SessionStateChangedparamzBMysqlx.Notice.SessionStateChanged.Parameter.GENERATED_DOCUMENT_IDScng|]2}tt|dd3S)v_octetsr)rdecode)rrs r)rz+Protocol._process_frame..sL;;;&( ;;WFFFLfhh;;;r+rrz9Mysqlx.Notice.SessionStateChanged.Parameter.ROWS_AFFECTEDv_unsigned_intz?Mysqlx.Notice.SessionStateChanged.Parameter.GENERATED_INSERT_IDN)r from_messagerr]rh_LOGGERwarningcodeappend_warninglevelrset_generated_idsrrrset_rows_affectedrset_generated_insert_id)r'rhresultwarn_msgsess_state_msgsess_state_values r)_process_framezProtocol._process_framesn v;!  +,C,/ N<'#:1#=#=(0!"'*m(/)/))),,()9;KLLNNNNN#G, .1/1///223C(*:4<4<=====/ (//r+c |j}n]#t$rP}t|}|r#td||Yd}~nd}~wwxYw|jdkrt|d|d|jdkr |||nx#YxYw|jdkrdS|jd kr| dnC|jd kr| dn"|jd kr| dnn<|S) z`Read message. Args: result (Result): A `Result` based type object. Tz {} reason: {}N Mysqlx.ErrorrhrMysqlx.Notice.FramezMysqlx.Sql.StmtExecuteOkzMysqlx.Resultset.FetchDonez(Mysqlx.Resultset.FetchDoneMoreResultsetsMysqlx.Resultset.Row) rri RuntimeErrorrepr get_warningsrZrrr set_closedset_has_more_results set_has_data)r'rrherrwarningss r)r[zProtocol._read_messages   ?l//11 ? ? ? 3 3 5 566?&'..sH==???????? ? x>))&s5z3v;???222''V4444H777t999!!$''''GGG++D1111333##D)))3 4 s! A7AA22A7-CCc|||_|j||j|dS)zSets the compression algorithm to be used by the compression object, for uplink and downlink. Args: algorithm (str): Algorithm to be used in compression/decompression. .. versionadded:: 8.0.21 N)rrrnrr&s r)rnzProtocol.set_compressions>'0# $$Y/// $$Y/////r+cRtd}|jtd||j}|jdkr$|j}|jdk$|jdkrt|d|d|S)zkGet capabilities. Returns: mysqlx.protobuf.Message: MySQL X Protobuf Message. z!Mysqlx.Connection.CapabilitiesGetz/Mysqlx.ClientMessages.Type.CON_CAPABILITIES_GETrrrhr)rrrrrrirrrgs r)get_capabiliteszProtocol.get_capabilitess 9:: "" K L L    l''))h///,++--Ch/// 8~ % %"3u:s6{;; ; r+c V|sdStd}|D] \}}td}||d<t|tr|}g}|D]T}td||||} || Utd|} td d | } | |d <n|||d <|d |g td} || d <|j td|  | S#t$r} | j dkrYd} ~ nd} ~ wwxYwdS)zSet capabilities. Args: **kwargs: Arbitrary keyword arguments. Returns: mysqlx.protobuf.Message: MySQL X Protobuf Message. NzMysqlx.Connection.CapabilitieszMysqlx.Connection.Capabilityrrrrrrrrr capabilitiesz!Mysqlx.Connection.CapabilitiesSetz/Mysqlx.ClientMessages.Type.CON_CAPABILITIES_SETi)rrrrrr]rrrrrread_okrerrno)r'kwargsrrr capabilityrritemrrrrhrs r)set_capabilitieszProtocol.set_capabilitiess  F?@@  ,,.. L LJC !?@@J!$Jv %&& >!;;D%&K*.,0,<,> !   yD  !      ts2F F& F!!F&Nctd}||d<|||d<|||d<|jtd|dS)zSend authenticate start. Args: method (str): Message method. auth_data (Optional[str]): Authentication data. initial_response (Optional[str]): Initial response. z Mysqlx.Session.AuthenticateStart mech_nameN auth_datainitial_responsez2Mysqlx.ClientMessages.Type.SESS_AUTHENTICATE_STARTrrrr)r'methodrrrhs r)send_auth_startzProtocol.send_auth_start=sz899!K  (C   '&6C" # ""= @$B$BCF H H H H Hr+c|j}|jdkr$|j}|jdk$|jdkrtd|dS)zRead authenticate continue. Raises: :class:`InterfaceError`: If the message type is not `Mysqlx.Session.AuthenticateContinue` Returns: str: The authentication data. r#Mysqlx.Session.AuthenticateContinuez>Unexpected message encountered during authentication handshaker)rrirrrgs r)read_auth_continuezProtocol.read_auth_continueNsxl''))h///,++--Ch/// 8< < < "<== =;r+cxtd|}|jtd|dS)zeSend authenticate continue. Args: auth_data (str): Authentication data. r )rz5Mysqlx.ClientMessages.Type.SESS_AUTHENTICATE_CONTINUENr)r'rrhs r)send_auth_continuezProtocol.send_auth_continue`s[ ; )+++ ""= C$E$EFI K K K K Kr+c |j}|jdkrdS|jdkrt|jF)z~Read authenticate OK. Raises: :class:`mysqlx.InterfaceError`: If message type is `Mysqlx.Error`. TzMysqlx.Session.AuthenticateOkrN)rrirrrhrgs r) read_auth_okzProtocol.read_auth_okksO  .,++--Cx:::x>))$SW---  .r+c|jr|jdkr |jdkr||\}}nj|jdkr||\}}nF|jdkr||\}}n"t d|t|}td}td}td|| |d <|jdkrtd||d z |d <||d <t|\}} td} t|| d<|| | <td} |j | d<| | d<|j td|  |dS#t $rt"wxYw)a Send prepare statement. Args: msg_type (str): Message ID string. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. stmt (Statement): A `Statement` based type object. Raises: :class:`mysqlx.NotSupportedError`: If prepared statements are not supported. .. versionadded:: 8.0.16 Mysqlx.Crud.InsertMysqlx.Crud.FindMysqlx.Crud.UpdateMysqlx.Crud.DeletezInvalid message type: {}z!Mysqlx.Expr.Expr.Type.PLACEHOLDERzMysqlx.Crud.LimitExprzMysqlx.Expr.Expr)rposition row_countroffset limit_expr#Mysqlx.Prepare.Prepare.OneOfMessagerzMysqlx.Prepare.Preparestmt_idrz*Mysqlx.ClientMessages.Type.PREPARE_PREPAREN) has_limitr build_find build_update build_deleterYrZr{rrrrrrrrrr) r'rerhr_r placeholdermsg_limit_expr oneof_typeoneof_op msg_oneof msg_prepares r)send_prepare_preparezProtocol.send_prepare_preparexs > /ch*>>>x---..33111**40033111**40033 !;!B!B8!L!LMMM4,,..//H'(KLLK$%<==N*12D7B;C+E+E+EN; 'x---+23E8C    & &  !9!9!;!;<<HHJJ  !6!6!8!899EEGG(    "" F G G      r+ctd}||d<|jtd||dS)z Send prepare deallocate statement. Args: stmt_id (int): Statement ID. .. versionadded:: 8.0.16 zMysqlx.Prepare.Deallocaterz-Mysqlx.ClientMessages.Type.PREPARE_DEALLOCATEN)rrrrr)r'r msg_deallocs r)send_prepare_deallocatez Protocol.send_prepare_deallocatesX9:: !( I "" I J J     r+cV|jrMtd}||d<|jdkr||d<||d<|dkrdnd}||| }|r|d ||||d S) a) Send a message without prepared statements support. Args: msg_type (str): Message ID string. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. stmt (Statement): A `Statement` based type object. .. versionadded:: 8.0.16 zMysqlx.Crud.Limitrrrlimit+Mysqlx.ClientMessages.Type.SQL_STMT_EXECUTEFTr'rN)rrr(rr)rrsend_msg)r'rerhr msg_limitrrs r)send_msg_without_pszProtocol.send_msg_without_pss > % 344I%)%=%=%?%?Ik "x---&*&;&;&=&= (#$CLHHHE %%di%@@  % K  t $ $ $ h$$$$$r+cV|jt||dS)z Send a message. Args: msg_type (str): Message ID string. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. .. versionadded:: 8.0.16 N)rrr)r'rerhs r)r2zProtocol.send_msgs* ""=#:#:C@@@@@r+ct|rdnd}td|jj|jj}td||}|jr||d<|||| rtd|d <n&| rtd |d <|j d kr |j |d <d |fS)aBuild find/read message. Args: stmt (Statement): A :class:`mysqlx.ReadStatement` or :class:`mysqlx.FindStatement` object. Returns: (tuple): Tuple containing: * `str`: Message ID string. * :class:`mysqlx.protobuf.Message`: MySQL X Protobuf Message. .. versionadded:: 8.0.16 Mysqlx.Crud.DataModel.DOCUMENTMysqlx.Crud.DataModel.TABLEMysqlx.Crud.Collectionrschemar data_model collection projectionz'Mysqlx.Crud.Find.RowLock.EXCLUSIVE_LOCKlockingz$Mysqlx.Crud.Find.RowLock.SHARED_LOCKrlocking_optionsz$Mysqlx.ClientMessages.Type.CRUD_FIND) r is_doc_basedrtargetrr;has_projectionget_projection_exprris_lock_exclusiveis_lock_sharedlock_contentionr'rr=r>rhs r)rzProtocol.build_finds0#&*&7&7&9&9$A#C#C#@BB 5"&+"2$(K$4666 (Z!+---   ; $ 8 8 : :C  3%%%  ! ! # # FGHH  NN  " " FDEE  N  ! # #%)%9C! "5s::r+c&t|rdnd}td|jj|jj}td||}||||D]t\}}td}|j |d<|j |d <|j t|j |d <|d |gud |fS) aBuild update message. Args: stmt (Statement): A :class:`mysqlx.ModifyStatement` or :class:`mysqlx.UpdateStatement` object. Returns: (tuple): Tuple containing: * `str`: Message ID string. * :class:`mysqlx.protobuf.Message`: MySQL X Protobuf Message. .. versionadded:: 8.0.16 r7r8r9r:rr<zMysqlx.Crud.UpdateOperation operationsourceNrz&Mysqlx.ClientMessages.Type.CRUD_UPDATE)rrBrrCrr;rget_update_opsr update_typerLrr rr)r'rr=r>rhr update_oprKs r)rzProtocol.build_update+s5#&*&7&7&9&9$A#C#C#@BB 5"&+"2$(K$4666 *z!+--- 3%%% //117799 ? ?LAy =>>I%.%:Ik ""+"2Ih *%/ %@%@ '"   # #Y%:%:%<%<$= > > > >7<n;|d  t| g|d  | gt |d r||d <d|fS)aBuild insert statement. Args: stmt (Statement): A :class:`mysqlx.AddStatement` or :class:`mysqlx.InsertStatement` object. Returns: (tuple): Tuple containing: * `str`: Message ID string. * :class:`mysqlx.protobuf.Message`: MySQL X Protobuf Message. .. versionadded:: 8.0.16 r7r8r9r:rr<_fieldsr?zMysqlx.Crud.Insert.TypedRowfieldrow is_upsertupsertz&Mysqlx.ClientMessages.Type.CRUD_INSERT)rrBrrCrr;hasattrrWr parse_table_insert_fieldrr get_valuesrrr rZ) r'rr=r>rhrXexprrrYvals r) build_insertzProtocol.build_inserts#&*&7&7&9&9$A#C#C#@BB 5"&+"2$(K$4666 *z!+--- 4 # # ? ? ?!%T->->-@-@)@AA--//L!(($*:*:*<*<)=>>>>__&& 3 3E788C%&& G IICL''C)D)D)F)F(GHHHHIG ##Z%6%6%B%B%D%D$EFFF J  s001 2 2 2 2 4 % % - NN,,CM7< ||d}n#t$rd}YnwxYw|rd|d<|jt d|||rdSdS)zSend reset session message. Returns: boolean: ``True`` if the server will keep the session open, otherwise ``False``. zMysqlx.Session.ResetNTF keep_openz%Mysqlx.ClientMessages.Type.SESS_RESET)rrrrrrr)r'rrhs r) send_resetzProtocol.send_resets,--   "%%'''  ! " " "!  "  $#C  ""= 3$5$569 ; ; ;   4us*> A  A )T)NNrE)%r=r>r?r@r*propertyrrrrrr[rnrrrr r rr%r+r.r4r2rrrrUrardrfrvrryr{rrrAr+r)rrs< ++X+ 999 888t""""H)=)=)=V   D 0 0 0&+++ZHHHH"   $ K K K . . .5$5$5$n@ %%%2 A A A%;%;%;N = = =D===2BBBB<*=*=*=X C C C   < C C C::: ;;; ? ? ?r+r),r@loggingrVr#ior lz4.framer/HAVE_LZ4 ImportError zstandardr HAVE_ZSTDerrorsrrrrr_r r r r r rhelpersrrrrprotobufrrrrrrv getLoggerrobjectrrCrprrAr+r)rs: HHHHHIIIII''''''''''''@@@@@@@@@@@@@@@@66666666HHHHHHHHHHHHHH ' H % %@@@@@@@@F`H`H`H`H`HF`H`H`HF6H6H6H6H6HF6H6H6HrI I I I I vI I I I I s''2<<