o /i0 @sdZddlZddlZddlZddlZddlmZddlmZddlmZe Z ej ej ej ejejejejejejf ZddZdd Zd d Zd"d dZe fddZddZddZddZddZd#ddZddZddZ d d!Z!dS)$zHelpers for :mod:`protobuf`.N)field_mask_pb2)message) wrappers_pb2cCsR|}tt|ddr||}n|}||s'td|d|jjd|S)a~Converts an ``Any`` protobuf to the specified message type. Args: pb_type (type): the type of the message that any_pb stores an instance of. any_pb (google.protobuf.any_pb2.Any): the object to be converted. Returns: pb_type: An instance of the pb_type message. Raises: TypeError: if the message could not be converted. pbNzCould not convert `z9` with underlying type `google.protobuf.any_pb2.Any` to ``)callablegetattrrUnpack TypeErrorTypeName DESCRIPTOR full_name)pb_typeany_pbmsgmsg_pbrT/var/www/passon-env/lib/python3.10/site-packages/google/api_core/protobuf_helpers.py from_any_pb)s  rcKsH|sdSdd|D}t|dkr"tdjdt|ddS)zRaise ValueError if more than one keyword argument is not ``None``. Args: kwargs (dict): The keyword arguments sent to the function. Raises: ValueError: If more than one entry in ``kwargs`` is not ``None``. NcSsg|]}|dur|qSNr).0valrrr Vszcheck_oneof..z#Only one of {fields} should be set.z, )fields)valueslen ValueErrorformatjoinsortedkeys)kwargs not_nonesrrr check_oneofIs  r$cCsBt}t|D]}t||}t|rt|tjr|||<q|S)aDiscovers all protobuf Message classes in a given import module. Args: module (module): A Python module; :func:`dir` will be run against this module to find Message subclasses. Returns: dict[str, google.protobuf.message.Message]: A dictionary with the Message class names as keys, and the Message subclasses themselves as values. ) collections OrderedDictdirrinspectisclass issubclassrMessage)moduleanswername candidaterrr get_messages_s  r0.cCs(||d}t|dkr|S|ddfS)abResolve a potentially nested key. If the key contains the ``separator`` (e.g. ``.``) then the key will be split on the first instance of the subkey:: >>> _resolve_subkeys('a.b.c') ('a', 'b.c') >>> _resolve_subkeys('d|e|f', separator='|') ('d', 'e|f') If not, the subkey will be :data:`None`:: >>> _resolve_subkeys('foo') ('foo', None) Args: key (str): A string that may or may not contain the separator. separator (str): The namespace separator. Defaults to `.`. Returns: Tuple[str, str]: The key and subkey(s). rrN)splitr)key separatorpartsrrr_resolve_subkeysss   r6cCst|\}}t|tjrt|||}nt|tjjr!|||}n t d t ||t ur2t ||durA||urAt|||dS|S)aRetrieve a key's value from a protobuf Message or dictionary. Args: mdg_or_dict (Union[~google.protobuf.message.Message, Mapping]): the object. key (str): The key to retrieve from the object. default (Any): If the key is not present on the object, and a default is set, returns that default instead. A type-appropriate falsy default is generally recommended, as protobuf messages almost always have default values for unset values and it is not always possible to tell the difference between a falsy value and an unset one. If no default is set then :class:`KeyError` will be raised if the key is not present in the object. Returns: Any: The return value from the underlying Message or dict. Raises: KeyError: If the key is not found. Note that, for unset values, messages and dictionaries may not have consistent behavior. TypeError: If ``msg_or_dict`` is not a Message or Mapping. z4get() expected a dict or protobuf message, got {!r}.Ndefault)r6 isinstancerr+rr%abcMappinggetr rtype _SENTINELKeyError) msg_or_dictr3r8subkeyr-rrrr<s  r<cCst|tjjtfr=t||rt||t||s|D]}t|tjjr1t||jdi|qt|| |gqdSt|tjjrX| D] \}}t t||||qHdSt|t j rht|||dSt|||dS)z!Set helper for protobuf Messages.Nr)r9r%r:MutableSequencetuplerpopr;addextenditemssetrr+CopyFromsetattr)rr3valueitemitem_key item_valuerrr_set_field_on_messages    rOcCst|tjjtjfstdt|t |\}}|dur5t|tjjr*| |it t ||||dSt|tjjrB|||<dSt |||dS)aCSet a key's value on a protobuf Message or dictionary. Args: msg_or_dict (Union[~google.protobuf.message.Message, Mapping]): the object. key (str): The key to set. value (Any): The value to set. Raises: TypeError: If ``msg_or_dict`` is not a Message or dictionary. z4set() expected a dict or protobuf message, got {!r}.N)r9r%r:MutableMappingrr+r rr=r6 setdefaultrHr<rO)r@r3rKbasekeyrArrrrHs    rHcCs"t||ddst|||dSdS)a~Set the key on a protobuf Message or dictionary to a given value if the current value is falsy. Because protobuf Messages do not distinguish between unset values and falsy ones particularly well (by design), this method treats any falsy value (e.g. 0, empty list) as a target to be overwritten, on both Messages and dictionaries. Args: msg_or_dict (Union[~google.protobuf.message.Message, Mapping]): the object. key (str): The key on the object in question. value (Any): The value to set. Raises: TypeError: If ``msg_or_dict`` is not a Message or dictionary. Nr7)r<rH)r@r3rKrrrrQsrQcCs|dur |dur tS|dur|durt|}||dur.|dur.t|}|t|t|sAtdt|t|tjt ||dS)aCreate a field mask by comparing two messages. Args: original (~google.protobuf.message.Message): the original message. If set to None, this field will be interpreted as an empty message. modified (~google.protobuf.message.Message): the modified message. If set to None, this field will be interpreted as an empty message. Returns: google.protobuf.field_mask_pb2.FieldMask: field mask that contains the list of field names that have different values between the two messages. If the messages are equivalent, then the field mask is empty. Raises: ValueError: If the ``original`` or ``modified`` are not the same type. Nz`expected that both original and modified should be of the same type, received "{!r}" and "{!r}".)paths) r FieldMaskcopydeepcopyClearr9r=rr_field_mask_helper)originalmodifiedrrr field_masks    r[cCsg}|jjD]H}t||}t||}t||}t|st|rE||krDt|s+t|r1||q|s;||q|t |||q||krN||q|Sr) r fields_by_name _get_pathr _is_message _is_wrapperappend ListFieldsrFrX)rYrZcurrentr-r. field_path original_val modified_valrrrrXEs&        rXcCs|d}|s |Sd||fS)N_z%s.%s)rstrip)rcr.rrrr^as  r^cCs t|tjSr)r9rr+rKrrrr_n r_cCs t|tvSr)r=_WRAPPER_TYPESrirrrr`rrjr`)r1)r\)"__doc__r%collections.abcrUr(google.protobufrrrobjectr> BoolValue BytesValue DoubleValue FloatValue Int32Value Int64Value StringValue UInt32Value UInt64Valuerkrr$r0r6r<rOrHrQr[rXr^r_r`rrrrs@       3% )