hjSUdZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZdZded<d Zd ed <egZd ed < dd lZdeDZdZn #e$rdZYnwxYwddlmZmZmZ ddlmZddlmZm Z m!Z!m"Z"ddl#m$Z$ ddl%m&Z&e'e&n #e$rYnwxYwe'e$e(edrej)Zddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6iZ7ded<e2j8j 9D] \Z:Z;e;e7de:<e2j<j 9D] \Z:Z;e;e7de:<e2j=j>9D] \Z:Z;e;e7de:<e,j?9D] \Z:Z;e;e7de:<e,j@jA9D] \Z:Z;e;e7de:<e,jBjC9D] \Z:Z;e;e7d e:<e,jDjE9D] \Z:Z;e;e7d!e:<e.jFj 9D] \Z:Z;e;e7d"e:<e.jj 9D] \Z:Z;e;e7d#e:<e/jGjHjI9D] \Z:Z;e;e7d$e:<e/jGjHjJ9D] \Z:Z;e;e7d%e:<e/jGjK9D] \Z:Z;e;e7d&e:<e0jLj 9D] \Z:Z;e;e7d'e:<e0jMj 9D] \Z:Z;e;e7d(e:<e1jNjO9D] \Z:Z;e;e7d)e:<e1jPjQ9D] \Z:Z;e;e7d*e:<e1jRjS9D] \Z:Z;e;e7d+e:<e3jTjUj 9D] \Z:Z;e;e7d,e:<e4jVjW9D] \Z:Z;e;e7d-e:<ejXZYe!jZeYZ[eY\e j]^e+j_j`eY\e j]^e,j_j`eY\e j]^e-j_j`eY\e j]^e.j_j`eY\e j]^e/j_j`eY\e j]^e0j_j`eY\e j]^e1j_j`eY\e j]^e2j_j`eY\e j]^e3j_j`eY\e j]^e4j_j`eY\e j]^e5j_j`eY\e j]^e6j_j`d.eDZdZad ZbGd/d0Zcn.#eedeef$r#ZfdZaeefnd1Zbesed2ebefYd Zf[fnd Zf[fwwxYwd3d4d5d6d7d8Zgd9ed:<Gd;d<ZhGd=d>ZidDdCZjd S)Ez?Implementation of a helper class for MySQL X Protobuf messages.) annotations)AnyDictListOptionalTupleTypeUnion) MessageTypeProtobufMessageCextTypeProtobufMessageType))zMysqlx.ServerMessages.Type.OKz Mysqlx.Ok)z Mysqlx.ServerMessages.Type.ERRORz Mysqlx.Error)z,Mysqlx.ServerMessages.Type.CONN_CAPABILITIESzMysqlx.Connection.Capabilities)z5Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_CONTINUEz#Mysqlx.Session.AuthenticateContinue)z/Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_OKzMysqlx.Session.AuthenticateOk)z!Mysqlx.ServerMessages.Type.NOTICEzMysqlx.Notice.Frame)z5Mysqlx.ServerMessages.Type.RESULTSET_COLUMN_META_DATAzMysqlx.Resultset.ColumnMetaData)z(Mysqlx.ServerMessages.Type.RESULTSET_ROWzMysqlx.Resultset.Row)z/Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONEzMysqlx.Resultset.FetchDone)z4Mysqlx.ServerMessages.Type.RESULTSET_FETCH_SUSPENDEDzMysqlx.Resultset.FetchSuspended)z?Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONE_MORE_RESULTSETSz(Mysqlx.Resultset.FetchDoneMoreResultsets)z.Mysqlx.ServerMessages.Type.SQL_STMT_EXECUTE_OKzMysqlx.Sql.StmtExecuteOk)z?Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONE_MORE_OUT_PARAMSz'Mysqlx.Resultset.FetchDoneMoreOutParams)z&Mysqlx.ServerMessages.Type.COMPRESSIONzMysqlx.Connection.CompressionzTuple[Tuple[str, str], ...]_SERVER_MESSAGES_TUPLESN Optional[str]PROTOBUF_VERSIONzpList[Union[Type[List[Dict[str, Any]]], Type[RepeatedCompositeContainer], Type[RepeatedCompositeFieldContainer]]]PROTOBUF_REPEATED_TYPEScXi|]'\}}ttj||(S)int _mysqlxpb enum_value.0keyvals `/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/mysqlx/protobuf/__init__.py rds>08SI  % %&&TF) BYTE_TYPES NUMERIC_TYPESencode_to_bytes)protobuf)descriptor_databasedescriptor_pb2descriptor_poolmessage_factory)RepeatedCompositeFieldContainer)RepeatedCompositeContainer __version__) mysqlx_connection_pb2mysqlx_crud_pb2mysqlx_cursor_pb2mysqlx_datatypes_pb2mysqlx_expect_pb2mysqlx_expr_pb2mysqlx_notice_pb2 mysqlx_pb2mysqlx_prepare_pb2mysqlx_resultset_pb2mysqlx_session_pb2mysqlx_sql_pb2zDict[str, int] _MESSAGESzMysqlx.ClientMessages.Type.zMysqlx.ServerMessages.Type.zMysqlx.Error.Severity.zMysqlx.Crud.DataModel.zMysqlx.Crud.Find.RowLock.zMysqlx.Crud.Order.Direction.z'Mysqlx.Crud.UpdateOperation.UpdateType.zMysqlx.Datatypes.Scalar.Type.zMysqlx.Datatypes.Any.Type.z0Mysqlx.Expect.Open.Condition.ConditionOperation.z!Mysqlx.Expect.Open.Condition.Key.z Mysqlx.Expect.Open.CtxOperation.zMysqlx.Expr.Expr.Type.z"Mysqlx.Expr.DocumentPathItem.Type.zMysqlx.Notice.Frame.Scope.zMysqlx.Notice.Warning.Level.z,Mysqlx.Notice.SessionStateChanged.Parameter.z)Mysqlx.Prepare.Prepare.OneOfMessage.Type.z*Mysqlx.Resultset.ColumnMetaData.FieldType.c0i|]\}}t||Srr7rs rrr s"SSSxsCy~sSSSrceZdZUdZejZded<eddZ edd Z eddZ eddZ eddZ eddZdS)_mysqlxpb_purez]This class implements the methods in pure Python used by the _mysqlxpb C++ extension.zmessage_factory.MessageFactoryfactorynamestrreturnrctjt|}|S)zCreate new Protobuf message. Args: name (str): Message type name. Returns: object: Protobuf message. )r;r< GetPrototype_DESCRIPTOR_POOLFindMessageTypeByName)r=clss r new_messagez_mysqlxpb_pure.new_messages:!(55 66t<<C355Lrenum_keyrct|S)zReturn enum value. Args: enum_key (str): Enum key. Returns: int: enum value. r9)rFs rrz_mysqlxpb_pure.enum_value%sX& &rmsgbytesc*|S)zSerialize message. Args: msg (object): Protobuf message. Returns: bytes: Serialized message bytes string. )SerializeToStringrHs rserialize_messagez _mysqlxpb_pure.serialize_message1s((** *rc*|S)zSerialize partial message. Args: msg (object): Protobuf message. Returns: bytes: Serialized partial message bytes string. )SerializePartialToStringrLs rserialize_partial_messagez(_mysqlxpb_pure.serialize_partial_message=s//11 1r msg_type_namepayloadcdt|}|||S)zSerialize partial message. Args: msg_type_name (str): Message type name. payload (bytes): Payload. Returns: object: Message parsed from string. )r;rEParseFromString)rQrRrHs r parse_messagez_mysqlxpb_pure.parse_messageIs0!,,];;C    ( ( (Jrmsg_typect|}|std|t|}|||S)zParse server message message. Args: msg_type (int): Message type. payload (bytes): Payload. Returns: object: Server message parsed from string. zUnknown msg_type: )SERVER_MESSAGESget ValueErrorr;rErT)rVrRrQrHs rparse_server_messagez#_mysqlxpb_pure.parse_server_messageXsc,//99M  B !@h!@!@AAA ,,];;C    ( ( (JrN)r=r>r?r)rFr>r?r)rHrr?rI)rQr>rRrIr?r)rVrrRrIr?r)__name__ __module__ __qualname____doc__r&MessageFactoryr<__annotations__ staticmethodrErrMrPrUr[rrrr;r;s $ $3Q/2P2R2RRRRR        ' ' '  '  + + +  +  2 2 2  2              rr;zProtobuf >=3.11.0 is requiredProtobuf is not available: )z-Mysqlx.Prepare.Prepare.OneOfMessage.Type.FINDfind)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.INSERTinsert)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.UPDATEupdate)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.DELETEdelete)z-Mysqlx.Prepare.Prepare.OneOfMessage.Type.STMT stmt_execute)z$Mysqlx.ClientMessages.Type.CRUD_FINDz&Mysqlx.ClientMessages.Type.CRUD_INSERTz&Mysqlx.ClientMessages.Type.CRUD_UPDATEz&Mysqlx.ClientMessages.Type.CRUD_DELETEz+Mysqlx.ClientMessages.Type.SQL_STMT_EXECUTEzDict[str, Tuple[str, str]]CRUD_PREPARE_MAPPINGcHeZdZUdZereneZe Zde d<e ddZ dS) ProtobufzProtobuf class acts as a container of the Protobuf message class. It allows the switch between the C extension and pure Python implementation message handlers, by patching the `mysqlxpb` class attribute. booluse_purer?Nonec|rtstdt|ststd|rtnt t _|t _dS)zSets whether to use the C extension or pure Python implementation. Args: use_pure (bool): `True` to use pure Python implementation. rcz-MySQL X Protobuf C extension is not availableN) HAVE_PROTOBUF ImportErrorHAVE_PROTOBUF_ERRORHAVE_MYSQLXPB_CEXTr;rrkmysqlxpbrm)rms r set_use_purezProtobuf.set_use_puresm  SM SQ> * *KD%   T5 ) ) ) ) * *rr=r>rc2tjr(t|tr%t |j|t |dSt|ttfrt |j||dSt|tr*t|j| |dSt|trsetattrrzr!r rlistgetattrextendrw MergeFrom get_messager~r=rs rr}zMessage.__setattr__sq   %%% : 4)?)?@@@@@EM:#>?? : 4/////E4(( : 4((//66666E7++ : 4((2253D3D3F3FGGGGG 4((22599999%%% "1%"8"8 $,6eW+E+EPE%%'''5 $rc tjs |j|nt|j|S#t$r t dwxYwr)rkrmrzrKeyErrorAttributeErrorr~r=s r __getattr__zMessage.__getattr__sS +'/'8V $gdiQU>V>V  + + + d * +s -0Ac2|||dSr)r}rs r __setitem__zMessage.__setitem__s u%%%%%rc,||Sr)rrs r __getitem__zMessage.__getitem__s%%%rdefaultctjs!|jd||nt |jd||S)a Returns the value of an element of the message dictionary. Args: name (string): Key name. default (object): The default value if the key does not exists. Returns: object: The value of the provided key name. rz)rkrmr{rYr)r~r=rs rrYz Message.getsF$ ?DM& ! % %dG 4 4 4v.g>> rrH3Union[ProtobufMessageType, ProtobufMessageCextType]c||jd<dS)zdSets the message. Args: msg (dict): Dictionary representing a message. rzNr{)r~rHs r set_messagezMessage.set_messages!$ frc|jdS)zReturns the dictionary representing a message containing parsed data. Returns: dict: The dictionary representing a message containing parsed data. rzrr~s rrzMessage.get_messages}V$$rrIcJtj|jS)zSerializes a message to a string. Returns: str: A string representing a message containing parsed data. )rkrtrMrzrs rserialize_to_stringzMessage.serialize_to_strings  2249===rcJtj|jS)zSerializes the protocol message to a binary string. This method is similar to serialize_to_string but doesn't check if the message is initialized. Returns: str: A string representation of the partial message. )rkrtrPrzrs rserialize_partial_to_stringz#Message.serialize_partial_to_strings ::49EEErcVtjs |jdn|jjjS)zstring: Message type name._mysqlxpb_type_name)rkrmrz DESCRIPTOR full_namers rtypez Message.type s, $ 0DI+ , ,%/ rrRcBtj||S)a>Creates a new message, initialized with parsed data. Args: msg_type_name (string): Message type name. payload (string): Serialized message data. Returns: dict: The dictionary representing a message containing parsed data. .. versionadded:: 8.0.21 )rkrtrU)rQrRs rparsez Message.parses ..}gFFFrr rctjr|n-tt |S)zReturns the size of the message in bytes. Args: msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. Returns: int: Size of the message in bytes. .. versionadded:: 8.0.21 )rkrmByteSizelenr!rrLs r byte_sizezMessage.byte_size&s@  ACLLNNN_S%<%<%>%>??@@ rrVcBtj||S)aCreates a new server-side message, initialized with parsed data. Args: msg_type (int): Message type. payload (string): Serialized message data. Returns: dict: The dictionary representing a message containing parsed data. )rkrtr[)rVrRs rparse_from_serverzMessage.parse_from_server8s 55hHHHrc|}|tj|||S)aCreates a new message, initialized with parsed data and returns a :class:`mysqlx.protobuf.Message` object. Args: msg_type_name (string): Message type name. payload (string): Serialized message data. Returns: mysqlx.protobuf.Message: The Message representing a message containing parsed data. )rrkrtrU)rDrQrRrHs r from_messagezMessage.from_messageGs9cee )77 wOOPPP rc|}|tj|||S)aCreates a new server-side message, initialized with parsed data and returns a :class:`mysqlx.protobuf.Message` object. Args: msg_type (int): Message type. payload (string): Serialized message data. Returns: mysqlx.protobuf.Message: The Message representing a message containing parsed data. )rrkrtr[)rDrVrRrHs rfrom_server_messagezMessage.from_server_messageXs9cee )>>xQQRRR rr)rQrrxrr?rn)r=r>rrr?rn)r=r>r?r)r=r>rrr?r)rHrr?rn)r?r)r?rI)r?r>)rQr>rRrIr?r)rHr r?r)rVrrRrIr?r)rQr>rRrIr?r )rVrrRrIr?r )r\r]r^r_rr}rrrrYrrrrpropertyrrbrrr classmethodrrrrrrwrws*****(++++&&&&&&&&     $$$$%%%%>>>> F F F F   X GGG\G    \ " I I I\ I[ [rrwr=r>r?rc@tj|S)zReturns the value of a MySQL X Protobuf enumerator. Args: name (string): MySQL X Protobuf numerator name. Returns: int: Value of the enumerator. )rkrtr)r=s r mysqlxpb_enumrjs   ' ' - --r)r=r>r?r)kr_ __future__rtypingrrrrrr r typesr r rrrarrrrrXrsrqhelpersrr r!googler"google.protobufr#r$r%r&#google.protobuf.internal.containersr'google.protobuf.pyext._messager(appendhasattrr)r+r,r-r.r/r0r1r2r3r4r5r6r7ClientMessagesr|rrServerMessagesErrorSeverity DataModelFindRowLockOrder DirectionUpdateOperation UpdateTypeScalarOpen ConditionConditionOperationKey CtxOperationExprDocumentPathItemFrameScopeWarningLevelSessionStateChanged ParameterPrepare OneOfMessageColumnMetaData FieldTypeDescriptorDatabase_DESCRIPTOR_DBDescriptorPoolrBAddFileDescriptorProto FromStringr serialized_pbrprrr; SyntaxError TypeErrorerrrirkrwrrrrrs >FEE#"""""@@@@@@@@@@@@@@@@@@MMMMMMMMMM-8----^#'&&&& F     >S:= 66677#3>DDFFIISEH ACAABB)/4::<<??S;> 7#7788(,17799<<S8; 4s4455&*4GMMOORRSNQ JSJJKK%*48>>@@CCS?B ;c;;<<%*7==??BBS>A :S::;;$(-335588S47 030011#49??AADDS@C >S:= 66677%9CIIKKNNSJM FFFGG'.;@FFHHKKSGJ CcCCDD)7AGGIILLSHK DsDDEE<(;==N55nEE*55 ! , :   *55  & 4   *55  ( 6   *55 + 9   *55  ( 6   *55  & 4   *55  ( 6   *55  ! /   *55  ) 7   *55 + 9   *55  ) 7   *55  % 3   TS;RSSSOMXXXXXXXXXt [),XXXM+1P XkM8KMMNNTWWXXXXX X-///4#440%%%%%%%%2DDDDDDDDN . . . . . .sNAAA(\B\B%"\$B%%Y,\\=\88\=