3 S^A@s&dZdkZegZy"ddlZeddeDZdZWnek rHd ZYnXd!d"l m Z m Z m Z m Z d!d#lmZyPd$d%lmZd$d&lmZd$d'lmZd$d(lmZd$d)lmZd$d*lmZd$d+lmZd$d,lmZd$d-lmZd$d.lmZd$d/lmZdd0lmZdd1lmZdd2lmZdd3lm Z dd4l!m"Z"ydd5l#m$Z$ej%e$Wnek rbYnXej%e"iZ&x(ej'j(j)D]\Z*Z+e+e&d6j,e*<qWx(ej-j(j)D]\Z*Z+e+e&d7j,e*<qWx(ej.j/j)D]\Z*Z+e+e&d8j,e*<qWx&ej0j)D]\Z*Z+e+e&d9j,e*<qWx(ej1j2j)D]\Z*Z+e+e&d:j,e*<q&Wx(ej3j4j)D]\Z*Z+e+e&d;j,e*<qPWx(ej5j6j)D]\Z*Z+e+e&dj,e*<qWx*ej9j:j;j)D]\Z*Z+e+e&d?j,e*<qWx*ej9j:jj(j)D]\Z*Z+e+e&dBj,e*<qzWx(ej?j(j)D]\Z*Z+e+e&dCj,e*<qWx(ej@jAj)D]\Z*Z+e+e&dDj,e*<qWx(ejBjCj)D]\Z*Z+e+e&dEj,e*<qWx(ejDjEj)D]\Z*Z+e+e&dFj,e*<q"Wx*ejFjGj(j)D]\Z*Z+e+e&dGj,e*<qNWx(ejHjIj)D]\Z*Z+e+e&dHj,e*<qxWejJZKejLeKZMeKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjReKjNejOjPejQjRedIdeDZdZSGdJdKdKeTZUWn(ek rd ZSesedLYnXdldmdndodpdWZVGdXdYdYeTZWGdZd[d[eTZXd\d]ZYdS)qzXThis module contains the implementation of a helper class for MySQL X Protobuf messages.Mysqlx.ServerMessages.Type.OK Mysqlx.Ok Mysqlx.ServerMessages.Type.ERROR Mysqlx.Error,Mysqlx.ServerMessages.Type.CONN_CAPABILITIESMysqlx.Connection.Capabilities5Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_CONTINUE#Mysqlx.Session.AuthenticateContinue/Mysqlx.ServerMessages.Type.SESS_AUTHENTICATE_OKMysqlx.Session.AuthenticateOk!Mysqlx.ServerMessages.Type.NOTICEMysqlx.Notice.Frame5Mysqlx.ServerMessages.Type.RESULTSET_COLUMN_META_DATAMysqlx.Resultset.ColumnMetaData(Mysqlx.ServerMessages.Type.RESULTSET_ROWMysqlx.Resultset.Row/Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONEMysqlx.Resultset.FetchDone4Mysqlx.ServerMessages.Type.RESULTSET_FETCH_SUSPENDEDMysqlx.Resultset.FetchSuspended?Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONE_MORE_RESULTSETS(Mysqlx.Resultset.FetchDoneMoreResultsets.Mysqlx.ServerMessages.Type.SQL_STMT_EXECUTE_OKMysqlx.Sql.StmtExecuteOk?Mysqlx.ServerMessages.Type.RESULTSET_FETCH_DONE_MORE_OUT_PARAMS'Mysqlx.Resultset.FetchDoneMoreOutParamsNcCs"g|]\}}ttj||fqS)int _mysqlxpb enum_value).0keyvalrrJ/tmp/pip-build-8app2_gc/mysql-connector-python/mysqlx/protobuf/__init__.py Asr$TF)PY3 NUMERIC_TYPES STRING_TYPES BYTE_TYPES)encode_to_bytes)mysqlx_connection_pb2)mysqlx_crud_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_pb2)descriptor_database)descriptor_pb2)descriptor_pool)message_factory)RepeatedCompositeFieldContainer)RepeatedCompositeContainerzMysqlx.ClientMessages.Type.{0}zMysqlx.ServerMessages.Type.{0}zMysqlx.Error.Severity.{0}zMysqlx.Crud.DataModel.{0}zMysqlx.Crud.Find.RowLock.{0}zMysqlx.Crud.Order.Direction.{0}z*Mysqlx.Crud.UpdateOperation.UpdateType.{0}z Mysqlx.Datatypes.Scalar.Type.{0}zMysqlx.Datatypes.Any.Type.{0}z3Mysqlx.Expect.Open.Condition.ConditionOperation.{0}z$Mysqlx.Expect.Open.Condition.Key.{0}z#Mysqlx.Expect.Open.CtxOperation.{0}zMysqlx.Expr.Expr.Type.{0}z%Mysqlx.Expr.DocumentPathItem.Type.{0}zMysqlx.Notice.Frame.Scope.{0}zMysqlx.Notice.Warning.Level.{0}z/Mysqlx.Notice.SessionStateChanged.Parameter.{0}z,Mysqlx.Prepare.Prepare.OneOfMessage.Type.{0}z-Mysqlx.Resultset.ColumnMetaData.FieldType.{0}cCsg|]\}}t||fqSr) _MESSAGES)r r!r"rrr#r$sc@sTeZdZdZejZeddZeddZ eddZ edd Z ed d Z d S) _mysqlxpb_purez]This class implements the methods in pure Python used by the _mysqlxpb C++ extension.cCstjjtj|}|S)N)r>factoryZ GetPrototype_DESCRIPTOR_POOLZFindMessageTypeByName)nameclsrrr# new_messages z_mysqlxpb_pure.new_messagecCst|S)N)r=)r!rrr#rsz_mysqlxpb_pure.enum_valuecCs|jS)N)ZSerializeToString)msgrrr#serialize_messagesz _mysqlxpb_pure.serialize_messagecCstj|}|j||S)N)r>rCParseFromString) msg_type_namepayloadrDrrr# parse_messages  z_mysqlxpb_pure.parse_messagecCs4tj|}|stdj|tj|}|j||S)NzUnknown msg_type: {0})SERVER_MESSAGESget ValueErrorformatr>rCrF)msg_typerHrGrDrrr#parse_server_messages    z#_mysqlxpb_pure.parse_server_messageN) __name__ __module__ __qualname____doc__r:ZMessageFactoryr? staticmethodrCrrErIrOrrrr#r>s    r>zProtobuf is not available-Mysqlx.Prepare.Prepare.OneOfMessage.Type.FINDfind/Mysqlx.Prepare.Prepare.OneOfMessage.Type.INSERTinsert/Mysqlx.Prepare.Prepare.OneOfMessage.Type.UPDATEupdate/Mysqlx.Prepare.Prepare.OneOfMessage.Type.DELETEdelete-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_EXECUTEc@s4eZdZdZereneZer dndZe 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. FTcCs@|rt rtdn| r(t r(td|r0tntt_|t_dS)zSets whether to use the C extension or pure Python implementation. Args: use_pure (bool): `True` to use pure Python implementation. zProtobuf is not availablez-MySQL X Protobuf C extension is not availableN) HAVE_PROTOBUF ImportErrorHAVE_MYSQLXPB_CEXTr>rr_mysqlxpbuse_pure)rdrrr# set_use_pures    zProtobuf.set_use_pureN) rPrQrRrSrbrr>rcrdrTrerrrr#r_s  r_c@seZdZdZdddZddZddZd d Zd d Zd d dZ ddZ ddZ ddZ e ddZeddZeddZeddZeddZdS)!MessagezHelper 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. NcKs@|rtjj|nd|jd<x |jD]\}}|j||q$WdS)N_msg)r_rcrC__dict__items __setattr__)selfrGkwargsr!valuerrr#__init__szMessage.__init__cCstjrtr(t|tr(t|j|t|qt|ttt frHt|j||qt|t rft |j|j |qt|t rt |j|j|jqt |j|j|nt|t r|jn||j|<dS)N)r_rdr& isinstancer(setattrrgr*r'r)listgetattrextendrfZ MergeFrom get_message)rkrArmrrr#rjs  zMessage.__setattr__c Cs:ytjs|j|St|j|Stk r4tYnXdS)N)r_rdrgrrKeyErrorAttributeError)rkrArrr# __getattr__-s  zMessage.__getattr__cCs|j||dS)N)rj)rkrArmrrr# __setitem__4szMessage.__setitem__cCs |j|S)N)rw)rkrArrr# __getitem__7szMessage.__getitem__cCs*tjs|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. rg)r_rdrhrKrr)rkrAdefaultrrr#rK:s z Message.getcCs||jd<dS)zdSets the message. Args: msg (dict): Dictionary representing a message. rgN)rh)rkrDrrr# set_messageHszMessage.set_messagecCs |jdS)zReturns the dictionary representing a message containing parsed data. Returns: dict: The dictionary representing a message containing parsed data. rg)rh)rkrrr#rtPszMessage.get_messagecCstjj|jS)zSerializes a message to a string. Returns: string: A string representing a message containing parsed data. )r_rcrErg)rkrrr#serialize_to_stringYszMessage.serialize_to_stringcCstjs|jdS|jjjS)zstring: Message type name.Z_mysqlxpb_type_name)r_rdrg DESCRIPTOR full_name)rkrrr#typeasz Message.typecCstjj||S)aCreates 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. )r_rcrI)rGrHrrr#parsegs z Message.parsecCstjj||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_rcrO)rNrHrrr#parse_from_serverts zMessage.parse_from_servercCs|}|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. )r{r_rcrI)rBrGrHrDrrr# from_messages zMessage.from_messagecCs|}|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. )r{r_rcrO)rBrNrHrDrrr#from_server_messages zMessage.from_server_message)N)N)rPrQrRrSrnrjrwrxryrKr{rtr|propertyrrTrr classmethodrrrrrr#rfs     rfcCs 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. )r_rcr)rArrr# mysqlxpb_enums rrrrrrrrrr r r r r rrrrrrrrrrrrr) rrrrrrrrrrrrr)rUrV)rWrX)rYrZ)r[r\)r]r^)ZrSZ_SERVER_MESSAGES_TUPLESrqZPROTOBUF_REPEATED_TYPESrdictrJrbracompatr&r'r(r)helpersr*r,r-r.r/r0r1r2r3r4r5r6Zgoogle.protobufr7r8r9r:Z#google.protobuf.internal.containersr;Zgoogle.protobuf.pyext._messager<appendr=ZClientMessagesTyperir!r"rMZServerMessagesErrorZSeverityZ DataModelZFindZRowLockZOrderZ DirectionZUpdateOperationZ UpdateTypeZScalarZAnyZOpen ConditionZConditionOperationZKeyZ CtxOperationExprZDocumentPathItemZFrameZScopeWarningZLevelZSessionStateChanged ParameterZPrepareZ OneOfMessageZColumnMetaDataZ FieldTypeZDescriptorDatabaseZ_DESCRIPTOR_DBZDescriptorPoolr@AddZFileDescriptorProtoZ FromStringr}Z serialized_pbr`objectr>ZCRUD_PREPARE_MAPPINGr_rfrrrrr#s                                            "