3 (h @s8dZddlmZddlmZddZGdddeZdS) z7Contains _ExtensionDict class to represent extensions. ) type_checkers)FieldDescriptorcCsjt|tstd||js*td|j|js>td|j|j|jk rftd|j|jj|jjfdS)z0Verify that the given extension handle is valid.z3HasExtension() expects an extension handle, got: %sz"%s" is not an extension.z""%s" is missing a containing_type.zKExtension "%s" extends message type "%s", but this message is of type "%s".N) isinstancerKeyError is_extension full_nameZcontaining_typeZ DESCRIPTOR)messageextension_handler M/tmp/pip-install-q3hcpn_q/protobuf/google/protobuf/internal/extension_dict.py_VerifyExtensionHandle&s   r c@speZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZdS)_ExtensionDictzDict-like container for Extension fields on proto instances. Note that in all cases we expect extension handles to be FieldDescriptors. cCs ||_dS)z^ Args: extended_message: Message instance for which we are the Extensions dict. N)_extended_message)selfZextended_messager r r __init__Gsz_ExtensionDict.__init__c Cst|j||jjj|}|dk r&|S|jtjkr@|j|j}n|jtj kr|j }t |dsj|jj j |t|j ddstd|j|j jf|j j}y|j|jjWqtk rYqXn|jS|jjj||}|S)z8Returns the current value of the given extension handle.N_concrete_classzAUninitialized concrete class found for field %r (message type %r))r r_fieldsgetlabelrLABEL_REPEATEDZ_default_constructorcpp_typeCPPTYPE_MESSAGE message_typehasattrZ_FACTORYZ GetPrototypegetattrAssertionErrorrrZ _SetListenerZ_listener_for_childrenReferenceError default_value setdefault)rr resultrr r r __getitem__Ns.     z_ExtensionDict.__getitem__cCsHt||jsdS|jj}|jj}dd|D}dd|D}||kS)NFcSsg|]}|jr|qSr )r).0fieldr r r sz)_ExtensionDict.__eq__..cSsg|]}|jr|qSr )r)r!r"r r r r#s)r __class__r ListFields)rotherZ my_fieldsZ other_fieldsr r r __eq__ws   z_ExtensionDict.__eq__cCs ||k S)Nr )rr&r r r __ne__sz_ExtensionDict.__ne__cCs |jj}dd|D}t|S)NcSsg|]}|djr|qS)r)r)r!r"r r r r#sz*_ExtensionDict.__len__..)rr%len)rfieldsZextension_fieldsr r r __len__s z_ExtensionDict.__len__cCs tddS)Nzunhashable object) TypeError)rr r r __hash__sz_ExtensionDict.__hash__cCs\t|j||jtjks$|jtjkr2td|jt j |}|j ||jj |<|jj dS)zlIf extension_handle specifies a non-repeated, scalar extension field, sets the value of that field. zKCannot assign to extension "%s" because it is a repeated or composite type.N)r rrrrrrr,rrZGetTypeCheckerZ CheckValuerZ _Modified)rr valueZ type_checkerr r r __setitem__s     z_ExtensionDict.__setitem__cCs|jj|dS)N)rZClearExtension)rr r r r __delitem__sz_ExtensionDict.__delitem__cCs|jjj|dS)zTries to find a known extension with the specified name. Args: name: Extension full name. Returns: Extension field descriptor. N)rZ_extensions_by_namer)rnamer r r _FindExtensionByNames z#_ExtensionDict._FindExtensionByNamecCs|jjj|dS)zTries to find a known extension with the field number. Args: number: Extension field number. Returns: Extension field descriptor. N)rZ_extensions_by_numberr)rnumberr r r _FindExtensionByNumbers z%_ExtensionDict._FindExtensionByNumbercCsdd|jjDS)Ncss |]}|djr|dVqdS)rN)r)r!fr r r sz*_ExtensionDict.__iter__..)rr%)rr r r __iter__sz_ExtensionDict.__iter__cCsft|j|||jjkrdS|jtjkr:t|jjj|S|jtj krb|jjj|}|dk o`|j SdS)NFT) r rrrrrboolrrrZ_is_present_in_parent)rr r.r r r __contains__s    z_ExtensionDict.__contains__N)__name__ __module__ __qualname____doc__rr r'r(r+r-r/r0r2r4r7r9r r r r r ?s)   r N)r=Zgoogle.protobuf.internalrZgoogle.protobuf.descriptorrr objectr r r r r  s