U
|
¤ý°djS ã
|
@ s8 U d Z ddlmZ ddlmZmZmZmZmZm Z m
|
Z
|
ddl m Z m Z mZ dZded< d Zd
|
ed < egZd ed < z dd lZdd eD ZdZW n ek
|
r¦ dZY nX ddlmZmZmZ zÚddl m Z ddl m Z m Z m!Z!m"Z" ddl#m$Z$ zddl%m&Z& e 'e&¡ W n ek
|
r Y nX e 'e$¡ e(e dr>e j)Zddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 i Z7ded< e2j8j 9¡ D ]\Z:Z;e;e7d e: < qe2j<j 9¡ D ]\Z:Z;e;e7d e: < q´e2j=j> 9¡ D ]\Z:Z;e;e7d e: < qÚe,j? 9¡ D ]\Z:Z;e;e7d e: < qþe,j@jA 9¡ D ]\Z:Z;e;e7d e: < q$e,jBjC 9¡ D ]\Z:Z;e;e7d!e: < qJe,jDjE 9¡ D ]\Z:Z;e;e7d"e: < qpe.jFj 9¡ D ]\Z:Z;e;e7d#e: < qe.jj 9¡ D ]\Z:Z;e;e7d$e: < q¼e/jGjHjI 9¡ D ]\Z:Z;e;e7d%e: < qäe/jGjHjJ 9¡ D ]\Z:Z;e;e7d&e: < q e/jGjK 9¡ D ]\Z:Z;e;e7d'e: < q2e0jLj 9¡ D ]\Z:Z;e;e7d(e: < qXe0jMj 9¡ D ]\Z:Z;e;e7d)e: < q~e1jNjO 9¡ D ]\Z:Z;e;e7d*e: < q¤e1jPjQ 9¡ D ]\Z:Z;e;e7d+e: < qÊe1jRjS 9¡ D ]\Z:Z;e;e7d,e: < qðe3jTjUj 9¡ D ]\Z:Z;e;e7d-e: < qe4jVjW 9¡ D ]\Z:Z;e;e7d.e: < q>e X¡ ZYe! ZeY¡Z[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/d eD ZdZad ZbG d0d1 d1ZcW nV eedeefk
|
rî Zf z0dZaed k rÄefnd2ZbesÞed3eb efW 5 d Zf[fX Y nX d4d5d6d7d8d9Zgd:ed;< G d<d= d=ZhG d>d? d?Zid@dAdBdCdDZjd S )Ez?Implementation of a helper class for MySQL X Protobuf messages.é )Ú annotations)ÚAnyÚDictÚListÚOptionalÚTupleÚTypeÚUnioné )Ú MessageTypeÚProtobufMessageCextTypeÚProtobufMessageType))z Mysqlx.ServerMessages.Type.OKz Mysqlx.Ok)z Mysqlx.ServerMessages.Type.ERRORz Mysqlx.Error)z,Mysqlx.ServerMessages.Type.CONN_CAPABILITIESz Mysqlx.Connection.Capabilities)z5Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_CONTINUEz#Mysqlx.Session.AuthenticateContinue)z/Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_OKz Mysqlx.Session.AuthenticateOk)z!Mysqlx.ServerMessages.Type.NOTICEzMysqlx.Notice.Frame)z5Mysqlx.ServerMessages.Type.RESULTSET_COLUMN_META_DATAz Mysqlx.Resultset.ColumnMetaData)z(Mysqlx.ServerMessages.Type.RESULTSET_ROWzMysqlx.Resultset.Row)z/Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONEzMysqlx.Resultset.FetchDone)z4Mysqlx.ServerMessages.Type.RESULTSET_FETCH_SUSPENDEDz Mysqlx.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.COMPRESSIONz Mysqlx.Connection.CompressionzTuple[Tuple[str, str], ...]Ú_SERVER_MESSAGES_TUPLESNú Optional[str]ÚPROTOBUF_VERSIONzpList[Union[Type[List[Dict[str, Any]]], Type[RepeatedCompositeContainer], Type[RepeatedCompositeFieldContainer]]]ÚPROTOBUF_REPEATED_TYPESc C s i | ]\}}t t |¡|qS © )ÚintÚ _mysqlxpbÚ
|
enum_value©Ú.0ÚkeyÚvalr r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\mysqlx/protobuf/__init__.pyÚ
|
<dictcomp>d s r TF)Ú
|
BYTE_TYPESÚ NUMERIC_TYPESÚencode_to_bytes)Úprotobuf)Údescriptor_databaseÚdescriptor_pb2Údescriptor_poolÚmessage_factory)Ú RepeatedCompositeFieldContainer)ÚRepeatedCompositeContainerÚ __version__é ) Úmysqlx_connection_pb2Úmysqlx_crud_pb2Úmysqlx_cursor_pb2Úmysqlx_datatypes_pb2Úmysqlx_expect_pb2Úmysqlx_expr_pb2Úmysqlx_notice_pb2Ú
|
mysqlx_pb2Úmysqlx_prepare_pb2Úmysqlx_resultset_pb2Úmysqlx_session_pb2Úmysqlx_sql_pb2zDict[str, int]Ú _MESSAGESzMysqlx.ClientMessages.Type.zMysqlx.ServerMessages.Type.zMysqlx.Error.Severity.zMysqlx.Crud.DataModel.zMysqlx.Crud.Find.RowLock.z Mysqlx.Crud.Order.Direction.z'Mysqlx.Crud.UpdateOperation.UpdateType.z Mysqlx.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.z Mysqlx.Notice.Warning.Level.z,Mysqlx.Notice.SessionStateChanged.Parameter.z)Mysqlx.Prepare.Prepare.OneOfMessage.Type.z*Mysqlx.Resultset.ColumnMetaData.FieldType.c C s i | ]\}}t | |qS r ©r4 r r r r r s c @ s e Zd ZU dZe ¡ Zded< 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.z message_factory.MessageFactoryÚfactoryÚstrr ©ÚnameÚreturnc C s t j t | ¡¡}| S )z«Create new Protobuf message.
|
|
Args:
|
name (str): Message type name.
|
|
Returns:
|
object: Protobuf message.
|
)r6 r7 Z GetPrototypeÚ_DESCRIPTOR_POOLZFindMessageTypeByName)r: Úclsr r r Ú new_message s
|
ÿz_mysqlxpb_pure.new_messager )Úenum_keyr; c C s t | S )zReturn enum value.
|
|
Args:
|
enum_key (str): Enum key.
|
|
Returns:
|
int: enum value.
|
r5 )r? r r r r % s
|
z_mysqlxpb_pure.enum_valueÚbytes©Úmsgr; c C s | ¡ S )z°Serialize message.
|
|
Args:
|
msg (object): Protobuf message.
|
|
Returns:
|
bytes: Serialized message bytes string.
|
)ZSerializeToString©rB r r r Úserialize_message1 s
|
z _mysqlxpb_pure.serialize_messagec C s | ¡ S )zÀSerialize partial message.
|
|
Args:
|
msg (object): Protobuf message.
|
|
Returns:
|
bytes: Serialized partial message bytes string.
|
)ZSerializePartialToStringrC r r r Úserialize_partial_message= s
|
z(_mysqlxpb_pure.serialize_partial_message©Ú msg_type_nameÚpayloadr; c C s t | ¡}| |¡ |S )zæSerialize partial message.
|
|
Args:
|
msg_type_name (str): Message type name.
|
payload (bytes): Payload.
|
|
Returns:
|
object: Message parsed from string.
|
)r6 r> ÚParseFromString)rG rH rB r r r Ú parse_messageI s
|
|
z _mysqlxpb_pure.parse_message©Úmsg_typerH r; c C s4 t | ¡}|s td| t |¡}| |¡ |S )zæParse server message message.
|
|
Args:
|
msg_type (int): Message type.
|
payload (bytes): Payload.
|
|
Returns:
|
object: Server message parsed from string.
|
zUnknown msg_type: )ÚSERVER_MESSAGESÚgetÚ
|
ValueErrorr6 r> rI )rL rH rG rB r r r Úparse_server_messageX s
|
|
|
z#_mysqlxpb_pure.parse_server_messageN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r# ZMessageFactoryr7 Ú__annotations__Ú staticmethodr> r rD rE rJ rP r r r r r6 s
|
r6 z Protobuf >=3.11.0 is requiredúProtobuf is not available: )z-Mysqlx.Prepare.Prepare.OneOfMessage.Type.FINDÚfind)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.INSERTÚinsert)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.UPDATEÚupdate)z/Mysqlx.Prepare.Prepare.OneOfMessage.Type.DELETEÚdelete)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@ e Zd ZU dZereneZe Zde d< e
|
dddddZ dS ) ÚProtobufzÙProtobuf 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.
|
ÚboolÚuse_pureÚNone)r_ r; c C s> | rt stdt | s&ts&td| r.tntt_| t_dS )z¡Sets whether to use the C extension or pure Python implementation.
|
|
Args:
|
use_pure (bool): `True` to use pure Python implementation.
|
rW z-MySQL X Protobuf C extension is not availableN) Ú HAVE_PROTOBUFÚ ImportErrorÚHAVE_PROTOBUF_ERRORÚHAVE_MYSQLXPB_CEXTr6 r r] Úmysqlxpbr_ )r_ r r r Ú set_use_pure s zProtobuf.set_use_pureN) rQ rR rS rT rd r r6 re r_ rU rV rf r r r r r] s
|
|
r] c @ s4 e Zd ZdZd3ddddddZd ddd
|
d d Zd dd ddZd ddd
|
ddZd dd ddZd4d dddddZ dddddZ
|
ddd d Z d dd d Z d dd!d"Z ed dd#d$Zed d dd%d&d'Zed(d)dd*d+Zed)d dd,d-d.Zed d d(d%d/d0Zed)d d(d,d1d2ZdS )5ÚMessagezÐHelper 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.
|
Nr r r` )rG Úkwargsr; c K s< |rt j |¡nd | jd< | ¡ D ]\}}| ||¡ q"d S )NÚ_msg)r] re r> Ú__dict__ÚitemsÚ __setattr__)ÚselfrG rh r: Úvaluer r r Ú__init__« s ÿzMessage.__init__r8 )r: rn r; c C sÐ t jrt|tr$t| j|t| qÌt|ttfrBt| j|| qÌt|t r`t
|
| j| |¡ qÌt|t rt
|
| j| | ¡ ¡ qÌt
|
| j| |¡ n6t|tr°t|| j|< n t|t rÂ| ¡ n|| j|< d S ©N)r] r_ Ú
|
isinstancer8 Úsetattrri r r r ÚlistÚgetattrÚextendrg Z MergeFromÚ get_message©rm r: rn r r r rl ´ s
|
|
|
|
ÿzMessage.__setattr__r9 c C s> z t js| j| n
|
t| j|W S tk
|
r8 td Y nX d S rp )r] r_ ri rt ÚKeyErrorÚAttributeError©rm r: r r r Ú __getattr__È s
|
ÿzMessage.__getattr__c C s | ||¡ d S rp )rl rw r r r Ú __setitem__Ð s zMessage.__setitem__c C s
|
| |¡S rp )r{ rz r r r Ú __getitem__Ó s zMessage.__getitem__)r: Údefaultr; c C s* t js| jd ||¡S t| 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_ rj rN rt )rm r: r~ r r r rN Ö s ÿýz Message.getz3Union[ProtobufMessageType, ProtobufMessageCextType]rA c C s || j d< dS )zdSets the message.
|
|
Args:
|
msg (dict): Dictionary representing a message.
|
ri N©rj )rm rB r r r Ú set_messageæ s zMessage.set_message)r; c C s
|
| j d S )z¸Returns the dictionary representing a message containing parsed
|
data.
|
|
Returns:
|
dict: The dictionary representing a message containing parsed data.
|
ri r ©rm r r r rv ð s zMessage.get_messager@ c C s t j | j¡S )z
Serializes a message to a string.
|
|
Returns:
|
str: A string representing a message containing parsed data.
|
)r] re rD ri r r r r Úserialize_to_stringù s zMessage.serialize_to_stringc C s t j | j¡S )zÿSerializes 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] re rE ri r r r r Úserialize_partial_to_string s z#Message.serialize_partial_to_stringc C s t js| jd S | jjjS )zstring: Message type name.Z_mysqlxpb_type_name)r] r_ ri Ú
|
DESCRIPTORZ full_namer r r r Útype s ÿ ýz Message.typerF c C s t j | |¡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] re rJ )rG rH r r r Úparse s z Message.parser r c C s t jr| ¡ S tt| ¡ S )zéReturns 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_ ZByteSizeÚlenr r rC r r r Ú byte_size& s ÿ
|
ýzMessage.byte_sizerK c C s t j | |¡S )a Creates 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] re rP )rL rH r r r Úparse_from_server8 s zMessage.parse_from_serverc C s | }| tj ||¡¡ |S )a Creates 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.
|
)r r] re rJ )r= rG rH rB r r r Ú from_messageG s zMessage.from_messagec C s | }| tj ||¡¡ |S )a Creates 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.
|
)r r] re rP )r= rL rH rB r r r Úfrom_server_messageX s zMessage.from_server_message)N)N)rQ rR rS rT ro rl r{ r| r} rN r rv r r Úpropertyr
rV r r r Ú classmethodr r r r r r rg £ s.
|
rg r8 r r9 c C s t j | ¡S )z¯Returns the value of a MySQL X Protobuf enumerator.
|
|
Args:
|
name (string): MySQL X Protobuf numerator name.
|
|
Returns:
|
int: Value of the enumerator.
|
)r] re r )r: r r r Ú mysqlxpb_enumj s r )krT Ú
|
__future__r Útypingr r r r r r r Útypesr r r r rU r rs r r rM rd rb Zhelpersr r r Zgoogler Zgoogle.protobufr r! r" r# Z#google.protobuf.internal.containersr$ Z google.protobuf.pyext._messager% ÚappendÚhasattrr& Ú r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 ZClientMessagesrk r r ZServerMessagesÚErrorZSeverityZ DataModelZFindZRowLockZOrderZ DirectionZUpdateOperationZ
|
UpdateTypeZScalarZOpenÚ ConditionZConditionOperationZKeyZ CtxOperationZExprZDocumentPathItemZFrameZScopeÚWarningZLevelZSessionStateChangedÚ ParameterZPrepareZ OneOfMessageZColumnMetaDataZ FieldTypeZDescriptorDatabaseZ_DESCRIPTOR_DBZDescriptorPoolr< ZAddZFileDescriptorProtoZ
|
FromStringr Z serialized_pbra rc r6 Ú SyntaxErrorÚ TypeErrorÚerrr\ r] rg r r r r r Ú<module> s, $ / ú
|
ÿ
|
|
8
|
ÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿZÿ"ï H
|
