o hjS @s$UdZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZdZded<d Zd ed <egZd ed <zdd lZddeDZdZWn eyRdZYnwddlmZmZmZzfddlmZddlmZm Z m!Z!m"Z"ddl#m$Z$z ddl%m&Z&e'e&Wn eyYnwe'e$e(edrej)Zddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6iZ7ded<e2j8j 9D] \Z:Z;e;e7de:<qe2j9D] \Z:Z;e;e7de:<qe,j?9D] \Z:Z;e;e7de:<qe,j@jA9D] \Z:Z;e;e7d e:<q e,jBjC9D] \Z:Z;e;e7d!e:<qe,jDjE9D] \Z:Z;e;e7d"e:<q0e.jFj 9D] \Z:Z;e;e7d#e:<qCe.jj 9D] \Z:Z;e;e7d$e:<qVe/jGjHjI9D] \Z:Z;e;e7d%e:<qje/jGjHjJ9D] \Z:Z;e;e7d&e:<q~e/jGjK9D] \Z:Z;e;e7d'e:<qe0jLj 9D] \Z:Z;e;e7d(e:<qe0jMj 9D] \Z:Z;e;e7d)e:<qe1jNjO9D] \Z:Z;e;e7d*e:<qe1jPjQ9D] \Z:Z;e;e7d+e:<qe1jRjS9D] \Z:Z;e;e7d,e:<qe3jTjUj 9D] \Z:Z;e;e7d-e:<qe4jVjW9D] \Z:Z;e;e7d.e:<qeXZYe!ZeYZ[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/deDZdZad ZbGd0d1d1ZcWn,eedeefyZfzdZaed urefnd2Zbesed3ebefWYd Zf[fnd Zf[fwwd4d5d6d7d8d9Zgd:ed;<Gdd?d?ZidFdDdEZjd S)Gz?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_TYPEScCs i|] \}}tt||qS)int _mysqlxpb enum_value.0keyvalrrS/var/www/html/scripts/venv/lib/python3.10/site-packages/mysqlx/protobuf/__init__.py dsrTF) 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.cCsi|] \}}t||qSrr4rrrrr sc@sveZdZUdZeZded<eddd Z edd d Z edddZ edddZ edddZ ed ddZdS)!_mysqlxpb_purez]This class implements the methods in pure Python used by the _mysqlxpb C++ extension.zmessage_factory.MessageFactoryfactorynamestrreturnr cCstjt|}|S)zCreate new Protobuf message. Args: name (str): Message type name. Returns: object: Protobuf message. )r6r7 GetPrototype_DESCRIPTOR_POOLFindMessageTypeByName)r8clsrrr new_messages z_mysqlxpb_pure.new_messageenum_keyrcCst|S)zReturn enum value. Args: enum_key (str): Enum key. Returns: int: enum value. r5)r@rrrr% z_mysqlxpb_pure.enum_valuemsgbytescC|S)zSerialize message. Args: msg (object): Protobuf message. Returns: bytes: Serialized message bytes string. )SerializeToStringrBrrrserialize_message1rAz _mysqlxpb_pure.serialize_messagecCrD)zSerialize partial message. Args: msg (object): Protobuf message. Returns: bytes: Serialized partial message bytes string. )SerializePartialToStringrFrrrserialize_partial_message=rAz(_mysqlxpb_pure.serialize_partial_message msg_type_namepayloadcCst|}|||S)zSerialize partial message. Args: msg_type_name (str): Message type name. payload (bytes): Payload. Returns: object: Message parsed from string. )r6r?ParseFromString)rJrKrBrrr parse_messageIs z_mysqlxpb_pure.parse_messagemsg_typecCs4t|}|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 ValueErrorr6r?rL)rNrKrJrBrrrparse_server_messageXs   z#_mysqlxpb_pure.parse_server_messageN)r8r9r:r )r@r9r:r)rBr r:rC)rJr9rKrCr:r )rNrrKrCr:r )__name__ __module__ __qualname____doc__r#MessageFactoryr7__annotations__ staticmethodr?rrGrIrMrRrrrrr6s      r6zProtobuf >=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@s:eZdZUdZer eneZe Zde d<e d 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:NonecCs>|r ts tdt|ststd|rtntt_|t_dS)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_CEXTr6rramysqlxpbrc)rcrrr set_use_pures  zProtobuf.set_use_pureN)rcrbr:rd) rSrTrUrVrhrr6rircrXrYrjrrrrras  rac@seZdZdZd4d5d d Zd6ddZd7ddZd6ddZd7ddZd4d8ddZ d9ddZ d:ddZ d;d d!Z d;d"d#Z edd+d,Zed?d.d/Zed@d0d1ZedAd2d3ZdS)BMessagezHelper class for interfacing with the MySQL X Protobuf extension. Args: msg_type_name (string): Protobuf type name. **kwargs: Arbitrary keyword arguments with values for the message. NrJrkwargsrr:rdcKs<|rtj|nd|jd<|D] \}}|||qdS)N_msg)rarir?__dict__items __setattr__)selfrJrlr8valuerrr__init__s zMessage.__init__r8r9rrcCstjrPt|trt|j|t|dSt|ttfr#t|j||dSt|t r3t |j| |dSt|t rEt |j| |dSt |j| |dSt|tr^t||j|<dSt|t rg|n||j|<dSN)rarc isinstancer9setattrrmrrrlistgetattrextendrk MergeFrom get_messagerqr8rrrrrrps     zMessage.__setattr__cCs6ztjs |j|WSt|j|WStytdwrt)rarcrmrxKeyErrorAttributeErrorrqr8rrr __getattr__s  zMessage.__getattr__cCs|||dSrt)rpr|rrr __setitem__szMessage.__setitem__cCs ||Srt)rrrrr __getitem__s zMessage.__getitem__defaultcCs*tjs |jd||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. rm)rarcrnrPrx)rqr8rrrrrPs  z Message.getrB3Union[ProtobufMessageType, ProtobufMessageCextType]cCs||jd<dS)zdSets the message. Args: msg (dict): Dictionary representing a message. rmNrn)rqrBrrr set_messageszMessage.set_messagecCs |jdS)zReturns the dictionary representing a message containing parsed data. Returns: dict: The dictionary representing a message containing parsed data. rmrrqrrrr{s zMessage.get_messagerCcCtj|jS)zSerializes a message to a string. Returns: str: A string representing a message containing parsed data. )rarirGrmrrrrserialize_to_stringszMessage.serialize_to_stringcCr)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. )rarirIrmrrrrserialize_partial_to_strings z#Message.serialize_partial_to_stringcCstjs|jdS|jjjS)zstring: Message type name._mysqlxpb_type_name)rarcrm DESCRIPTOR full_namerrrrtype s  z Message.typerKcCtj||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 )rarirM)rJrKrrrparsesz Message.parser rcCstjr|Stt|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 )rarcByteSizelenrrrFrrr byte_size&s zMessage.byte_sizerNcCr)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. )rarirR)rNrKrrrparse_from_server8s zMessage.parse_from_servercC|}|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. )rrarirM)r>rJrKrBrrr from_messageG zMessage.from_messagecCr)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. )rrarirR)r>rNrKrBrrrfrom_server_messageXrzMessage.from_server_messagert)rJrrlrr:rd)r8r9rrrr:rd)r8r9r:r)r8r9rrr:r)rBrr:rd)r:r)r:rC)r:r9)rJr9rKrCr:r)rBr r:r)rNrrKrCr:r)rJr9rKrCr:r )rNrrKrCr:r )rSrTrUrVrsrprrrrPrr{rrpropertyrrYrrr classmethodrrrrrrrks0            rkr8r9r:rcCs 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. )rarir)r8rrr mysqlxpb_enumjs r)r8r9r:r)krV __future__rtypingrrrrrrr typesr r r rrXrrwrrrOrhrfhelpersrrrgooglergoogle.protobufr r!r"r##google.protobuf.internal.containersr$google.protobuf.pyext._messager%appendhasattrr&r(r)r*r+r,r-r.r/r0r1r2r3r4ClientMessagesrorrServerMessagesErrorSeverity DataModelFindRowLockOrder DirectionUpdateOperation UpdateTypeScalarOpen ConditionConditionOperationKey CtxOperationExprDocumentPathItemFrameScopeWarningLevelSessionStateChanged ParameterPrepare OneOfMessageColumnMetaData FieldTypeDescriptorDatabase_DESCRIPTOR_DBDescriptorPoolr<AddFileDescriptorProto FromStringr serialized_pbrergr6 SyntaxError TypeErrorerrr`rarkrrrrrs8 $  /        8  Z  H