U 4JejS @s8UdZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZdZded<d Zd ed <egZd ed <zdd lZddeDZdZWnek rdZYnXddlmZmZmZzddlmZddlmZm Z m!Z!m"Z"ddl#m$Z$zddl%m&Z&e'e&Wnek r YnXe'e$e(edr>ej)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:<qJe,jDjE9D]\Z:Z;e;e7d"e:<qpe.jFj 9D]\Z:Z;e;e7d#e:<qe.jj 9D]\Z:Z;e;e7d$e:<qe/jGjHjI9D]\Z:Z;e;e7d%e:<qe/jGjHjJ9D]\Z:Z;e;e7d&e:<q e/jGjK9D]\Z:Z;e;e7d'e:<q2e0jLj 9D]\Z:Z;e;e7d(e:<qXe0jMj 9D]\Z:Z;e;e7d)e:<q~e1jNjO9D]\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:<q>eXZYe!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 ZbGd0d1d1ZcWnVeedeefk rZfz0dZaed k refnd2Zbesed3ebefW5d Zf[fXYnXd4d5d6d7d8d9Zgd:ed;<Gdd?d?Zid@dAdBdCdDZjd 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_TYPEScCs i|]\}}tt||qS)int _mysqlxpb enum_value.0keyvalrrds rTF) 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@seZdZUdZeZded<edddddZ edd d d d Z edd dddZ edd dddZ edd ddddZ ed d ddddZdS)_mysqlxpb_purez]This class implements the methods in pure Python used by the _mysqlxpb C++ extension.zmessage_factory.MessageFactoryfactorystrr namereturncCstjt|}|S)zCreate new Protobuf message. Args: name (str): Message type name. Returns: object: Protobuf message. )r6r7Z GetPrototype_DESCRIPTOR_POOLZFindMessageTypeByName)r:clsrrr new_messages z_mysqlxpb_pure.new_messager)enum_keyr;cCst|S)zReturn enum value. Args: enum_key (str): Enum key. Returns: int: enum value. r5)r?rrrr%s z_mysqlxpb_pure.enum_valuebytesmsgr;cCs|S)zSerialize message. Args: msg (object): Protobuf message. Returns: bytes: Serialized message bytes string. )ZSerializeToStringrBrrrserialize_message1s z _mysqlxpb_pure.serialize_messagecCs|S)zSerialize partial message. Args: msg (object): Protobuf message. Returns: bytes: Serialized partial message bytes string. )ZSerializePartialToStringrCrrrserialize_partial_message=s z(_mysqlxpb_pure.serialize_partial_message msg_type_namepayloadr;cCst|}|||S)zSerialize partial message. Args: msg_type_name (str): Message type name. payload (bytes): Payload. Returns: object: Message parsed from string. )r6r>ParseFromString)rGrHrBrrr parse_messageIs  z_mysqlxpb_pure.parse_messagemsg_typerHr;cCs4t|}|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>rI)rLrHrGrBrrrparse_server_messageXs   z#_mysqlxpb_pure.parse_server_messageN)__name__ __module__ __qualname____doc__r#ZMessageFactoryr7__annotations__ staticmethodr>rrDrErJrPrrrrr6s    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.STMTZ 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@eZdZUdZereneZe Zde d<e dddddZ 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_pureNone)r_r;cCs>|rtstdt|s&ts&td|r.tntt_|t_dS)zSets whether to use the C extension or pure Python implementation. Args: use_pure (bool): `True` to use pure Python implementation. rWz-MySQL X Protobuf C extension is not availableN) HAVE_PROTOBUF ImportErrorHAVE_PROTOBUF_ERRORHAVE_MYSQLXPB_CEXTr6rr]mysqlxpbr_)r_rrr set_use_pures zProtobuf.set_use_pureN) rQrRrSrTrdrr6rer_rUrVrfrrrrr]s  r]c@s4eZdZdZd3ddddddZd ddd d d Zd dd ddZd ddd ddZd dd ddZd4d dddddZ dddddZ ddddZ dddd Z ddd!d"Z ed dd#d$Zed ddd%d&d'Zed(d)dd*d+Zed)ddd,d-d.Zed dd(d%d/d0Zed)dd(d,d1d2ZdS)5MessagezHelper 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. Nrrr`)rGkwargsr;cKs<|rtj|nd|jd<|D]\}}|||q"dS)N_msg)r]rer>__dict__items __setattr__)selfrGrhr:valuerrr__init__szMessage.__init__r8)r:rnr;cCstjrt|tr$t|j|t|qt|ttfrBt|j||qt|t r`t |j| |qt|t rt |j| |qt |j| |n6t|trt||j|<nt|t r|n||j|<dSN)r]r_ isinstancer8setattrrirrrlistgetattrextendrgZ MergeFrom get_messagermr:rnrrrrls    zMessage.__setattr__r9cCs>ztjs|j|n t|j|WStk r8tdYnXdSrp)r]r_rirtKeyErrorAttributeErrorrmr:rrr __getattr__s zMessage.__getattr__cCs|||dSrp)rlrwrrr __setitem__szMessage.__setitem__cCs ||Srp)r{rzrrr __getitem__szMessage.__getitem__)r:defaultr;cCs*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. ri)r]r_rjrNrt)rmr:r~rrrrNs z Message.getz3Union[ProtobufMessageType, ProtobufMessageCextType]rAcCs||jd<dS)zdSets the message. Args: msg (dict): Dictionary representing a message. riNrj)rmrBrrr set_messageszMessage.set_message)r;cCs |jdS)zReturns the dictionary representing a message containing parsed data. Returns: dict: The dictionary representing a message containing parsed data. rirrmrrrrvszMessage.get_messager@cCstj|jS)zSerializes a message to a string. Returns: str: A string representing a message containing parsed data. )r]rerDrirrrrserialize_to_stringszMessage.serialize_to_stringcCstj|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. )r]rerErirrrrserialize_partial_to_strings z#Message.serialize_partial_to_stringcCstjs|jdS|jjjS)zstring: Message type name.Z_mysqlxpb_type_name)r]r_ri DESCRIPTORZ full_namerrrrtype s z Message.typerFcCstj||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 )r]rerJ)rGrHrrrparsesz 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 )r]r_ZByteSizelenrrrCrrr byte_size&s zMessage.byte_sizerKcCstj||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. )r]rerP)rLrHrrrparse_from_server8s zMessage.parse_from_servercCs|}|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. )rr]rerJ)r=rGrHrBrrr from_messageGs zMessage.from_messagecCs|}|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. )rr]rerP)r=rLrHrBrrrfrom_server_messageXs zMessage.from_server_message)N)N)rQrRrSrTrorlr{r|r}rNrrvrrpropertyrrVrrr classmethodrrrrrrrgs.    rgr8rr9cCs 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. )r]rer)r:rrr mysqlxpb_enumjs r)krT __future__rtypingrrrrrrr typesr r r rrUrrsrrrMrdrbZhelpersrrrZgooglerZgoogle.protobufr r!r"r#Z#google.protobuf.internal.containersr$Zgoogle.protobuf.pyext._messager%appendhasattrr&r(r)r*r+r,r-r.r/r0r1r2r3r4ZClientMessagesrkrrZServerMessagesErrorZSeverityZ DataModelZFindZRowLockZOrderZ DirectionZUpdateOperationZ UpdateTypeZScalarZOpen ConditionZConditionOperationZKeyZ CtxOperationZExprZDocumentPathItemZFrameZScopeWarningZLevelZSessionStateChanged ParameterZPrepareZ OneOfMessageZColumnMetaDataZ FieldTypeZDescriptorDatabaseZ_DESCRIPTOR_DBZDescriptorPoolr<ZAddZFileDescriptorProtoZ FromStringrZ serialized_pbrarcr6 SyntaxError TypeErrorerrr\r]rgrrrrrs, $ /        8  Z"H