@ijS UdZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZdZded<d Zd ed <egZd ed < dd lZeDcic]\}}eej,|| c}}ZdZddlmZmZmZ ddlmZddl m!Z!m"Z"m#Z#m$Z$ddl%m&Z& ddl'm(Z(ejSe(ejSe&e*edr ejVZddl,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8iZ9ded<e4jtjjwD] \Zd=Zly cc}}w#e$rdZYwxYw#e$rYwxYwcc}}w#eefegf$r ZhdZceehnd/Zdes ed0edehYd Zh[hod Zh[hwwxYw)?z?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_TYPESTF) 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.ceZdZUdZej Zded<ed dZ ed dZ ed dZ ed dZ eddZ edd Zy )_mysqlxpb_purez]This class implements the methods in pure Python used by the _mysqlxpb C++ extension.zmessage_factory.MessageFactoryfactorycttjjtj |}|S)zCreate new Protobuf message. Args: name (str): Message type name. Returns: object: Protobuf message. )r-r. GetPrototype_DESCRIPTOR_POOLFindMessageTypeByName)nameclss R/var/www/html/aiagents/fe/lib/python3.12/site-packages/mysqlx/protobuf/__init__.py new_messagez_mysqlxpb_pure.new_messages1!((55 66t<C5Lct|S)zReturn enum value. Args: enum_key (str): Enum key. Returns: int: enum value. )r+)enum_keys r5 enum_valuez_mysqlxpb_pure.enum_value%sX& &r7c"|jS)zSerialize message. Args: msg (object): Protobuf message. Returns: bytes: Serialized message bytes string. )SerializeToStringmsgs r5serialize_messagez _mysqlxpb_pure.serialize_message1s((* *r7c"|jS)zSerialize partial message. Args: msg (object): Protobuf message. Returns: bytes: Serialized partial message bytes string. )SerializePartialToStringr=s r5serialize_partial_messagez(_mysqlxpb_pure.serialize_partial_message=s//1 1r7cRtj|}|j||S)zSerialize partial message. Args: msg_type_name (str): Message type name. payload (bytes): Payload. Returns: object: Message parsed from string. )r-r6ParseFromString) msg_type_namepayloadr>s r5 parse_messagez_mysqlxpb_pure.parse_messageIs'!,,];C    (Jr7ctj|}|std|tj |}|j ||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-r6rD)msg_typerFrEr>s r5parse_server_messagez#_mysqlxpb_pure.parse_server_messageXsM,//9M  #5hZ!@AA ,,];C    (Jr7N)r3strreturnr)r9rNrOint)r>rrObytes)rErNrFrQrOr)rLrPrFrQrOr)__name__ __module__ __qualname____doc__rMessageFactoryr.__annotations__ staticmethodr6r:r?rBrGrMr7r5r-r-s $3Q/2P2P2R/R      '  '  +  +  2  2        r7r-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_MAPPINGc@eZdZUdZereneZe Zde d<e ddZ y)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_purec|rtstdt|sts td|rtnt t _|t _y)zSets whether to use the C extension or pure Python implementation. Args: use_pure (bool): `True` to use pure Python implementation. rZz-MySQL X Protobuf C extension is not availableN) HAVE_PROTOBUF ImportErrorHAVE_PROTOBUF_ERRORHAVE_MYSQLXPB_CEXTr- _mysqlxpbrbmysqlxpbrd)rds r5 set_use_purezProtobuf.set_use_puresD M ; *KD%   T5 ) *r7ctjrt|tr!t |j |t |yt|ttfrt |j ||yt|tr&t|j |j|yt|tr4t|j |j|jyt|j |j|yt|trt ||j |<yt|tr|jn||j |<yN)rbrd isinstancerNsetattrrqrrrlistgetattrextendro MergeFrom get_messagerur3rws r5rtzMessage.__setattr__s   %% 4)?@EM:#>? 4/E4( 4(//6E7+ 4(2253D3D3FG 4(2259%%"1%"8 $,6eW+EE%%'5 $r7c tjs|j|St|j|S#t$rt dwxYwrz)rbrdrqr~KeyErrorAttributeErrorrur3s r5 __getattr__zMessage.__getattr__sL +'/'8'8 $ >EdiiQU>V  + d * +s 77Ac(|j||yrz)rtrs r5 __setitem__zMessage.__setitem__s u%r7c$|j|Srz)rrs r5 __getitem__zMessage.__getitem__s%%r7ctjs|jdj||St |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. rq)rbrdrrrJr~)rur3defaults r5rJz Message.getsI$$ MM& ! % %dG 4 v.g> r7c"||jd<y)zdSets the message. Args: msg (dict): Dictionary representing a message. rqNrr)rur>s r5 set_messagezMessage.set_messages!$ fr7c |jdS)zReturns the dictionary representing a message containing parsed data. Returns: dict: The dictionary representing a message containing parsed data. rqrrus r5rzMessage.get_messages}}V$$r7cTtjj|jS)zSerializes a message to a string. Returns: str: A string representing a message containing parsed data. )rbrkr?rqrs r5serialize_to_stringzMessage.serialize_to_strings   22499==r7cTtjj|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. )rbrkrBrqrs r5serialize_partial_to_stringz#Message.serialize_partial_to_strings  ::499EEr7ctjs|jdS|jjjS)zstring: Message type name._mysqlxpb_type_name)rbrdrq DESCRIPTOR full_namers r5typez Message.type s; $$ II+ , %%// r7cBtjj||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 )rbrkrG)rErFs r5parsez Message.parses  ..}gFFr7ctjr|jStt |j 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 )rbrdByteSizelenrrr=s r5 byte_sizezMessage.byte_size&s:   LLN _S%<%<%>?@ r7cBtjj||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. )rbrkrM)rLrFs r5parse_from_serverzMessage.parse_from_server8s  55hHHr7cr|}|jtjj|||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. )rrbrkrG)r4rErFr>s r5 from_messagezMessage.from_messageGs/e ))77 wOP r7cr|}|jtjj|||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. )rrbrkrM)r4rLrFr>s r5from_server_messagezMessage.from_server_messageXs/e ))>>xQR r7rz)rErrvrrOrm)r3rNrwrrOrm)r3rNrOr)r3rNrrrOr)r>3Union[ProtobufMessageType, ProtobufMessageCextType]rOrm)rOr)rOrQ)rOrN)rErNrFrQrOr)r>r rOrP)rLrPrFrQrOr)rErNrFrQrOr )rLrPrFrQrOr )rRrSrTrUrxrtrrrrJrrrrpropertyrrXrrr classmethodrrrYr7r5roros*(+&& $F$ $%> F  GG%*G <GG   " I I % I < I I r7roc@tjj|S)zReturns the value of a MySQL X Protobuf enumerator. Args: name (string): MySQL X Protobuf numerator name. Returns: int: Value of the enumerator. )rbrkr:)r3s r5 mysqlxpb_enumrjs    ' ' --r7)r3rNrOrP)mrU __future__rtypingrrrrrr r typesr r rrrWrr}rrjrPr:rIrirghelpersrrrgooglergoogle.protobufrrrr#google.protobuf.internal.containersrgoogle.protobuf.pyext._messagerappendhasattrrrr r!r"r#r$r%r&r'r(r)r*r+ClientMessagesrskeyvalServerMessagesErrorSeverity DataModelFindRowLockOrder DirectionUpdateOperation UpdateTypeScalarOpen ConditionConditionOperationKey CtxOperationExprDocumentPathItemFrameScopeWarningLevelSessionStateChanged ParameterPrepare OneOfMessageColumnMetaData FieldTypeDescriptorDatabase_DESCRIPTOR_DBDescriptorPoolr1AddFileDescriptorProto FromStringr serialized_pbrfrhr- SyntaxError TypeErrorerrr`rbror)rrs00r5rs >F#@@@MM-84-^#'-& F  =T08S I  %&+OA@CX T M&&'AB""#BCx'#//     !#I~"--2288:=S9< /u56=--2288:=S9< /u56=$$--3358S47 *3%018$--3358S47 *3%018#((00668;S7: -cU34;#))3399;>S:= 067>#33>>DDFISEH ;C5ABI)//44::<?S;> 1#78?(,,11779<S8; .se45<&**44GGMMORSNQ DSEJKR%**4488>>@CS?B 5cU;<C%**77==?BS>A 4SE:;B$((--3358S47 *3%018#4499??ADS@C 6se<=D&++11779<S8; .se45<%--3399;>S:= 067>%99CCIIKNSJM @FGN'..;;@@FFHKSGJ =cUCDK)77AAGGILSHK >seDEL<(;;=N555nE**55 ! , , : :  **55  & & 4 4  **55  ( ( 6 6  **55 + + 9 9  **55  ( ( 6 6  **55  & & 4 4  **55  ( ( 6 6  **55  ! ! / /  **55  ) ) 7 7  **55 + + 9 9  **55  ) ) 7 7  **55  % % 3 3