U ڲg @sddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z e dZ Gddde dd d d d d dgZ dS)N)descriptor_pb2)descriptor_pool)message) reflection) MessageRule _FileInfoc@s^eZdZiZeddZeddZddZddZ d d Z dd d Z ddZ e ddZdS)rcCsF|j|}|sB|tj||ddtt|iid}|j|<|S)NZproto3)namepackageZsyntax) descriptorenumsmessagesrnested nested_enum)registrygetrZFileDescriptorProto collections OrderedDict)clsfilenamer r r4/tmp/pip-unpacked-wheel-zasgy6sz/proto/_file_info.pymaybe_add_descriptor%s  z_FileInfo.maybe_add_descriptorcCsd|ddS)Nz {0}.proto./)formatreplacerrrrproto_file_name8sz_FileInfo.proto_file_namecCs&t|}t|dr t|jjStS)N __protobuf__)inspect getmodulehasattr frozensetrmanifest)self new_classmodulerrr _get_manifest<s   z_FileInfo._get_manifestcCs|||jhSN)r'__name__)r$r%rrr_get_remaining_manifestCsz!_FileInfo._get_remaining_manifestcCsP||}|r6|j|kr6tdjt|j|jd|j|krDdS|pJdS)NzPproto-plus module {module} has a declared manifest but {class_name} is not in it)r& class_name)r'r)logwarningrrr lower)r$r%fallbackr#rrr_calculate_saltFs  z_FileInfo._calculate_saltr,c Cs0t}|||}djd|jjdd|gdd|j_||j|j D]\}}| |}t |jtjf|dd}||j_|jj|t|||jjD]F} | jrt| jtr|j | j| _q| jrt| jtr|j| j| _qqV|j D]\}} ||}|| j_q|j|jdS)aGenerate the descriptors for all protos in the file. This method takes the file descriptor attached to the parent message and generates the immutable descriptors for all of the messages in the file descriptor. (This must be done in one fell swoop for immutability and to resolve proto cross-referencing.) This is run automatically when the last proto in the file is generated, as determined by the module's __all__ tuple. z {name}.proto_Nir)Z DESCRIPTOR __module__)rZDefaultr1rjoinr rrstripZAddr itemsZFindMessageTypeByNamerZGeneratedProtocolMessageTyperMessage_metaZ_pbmarshalregisterrfieldsvalues isinstancestrenumr ZFindEnumTypeByNameZpbrpop) r$r%Z fallback_saltpoolZsaltZ full_nameproto_plus_messager Z pb_messagefieldZproto_plus_enumrrrgenerate_file_pbRs6        z_FileInfo.generate_file_pbcsxt|jst|jrdS|jD].}|jr4|j|jksF|jr|j|jkrdSqt || |}t fdd|DS)akReturn True if a file descriptor may added, False otherwise. This determine if all the messages that we plan to create have been created, as best as we are able. Since messages depend on one another, we create descriptor protos (which reference each other using strings) and wait until we have built everything that is going to be in the module, and then use the descriptor protos to instantiate the actual descriptors in one fell swoop. Args: new_class (~.MessageMeta): The new class currently undergoing creation. Fc3s|]}t|VqdSr()r!).0ir&rr sz"_FileInfo.ready..) lenr runresolved_fieldsrr r?r rr r*all)r$r%rCr#rrGrreadys    z_FileInfo.readyccsP|jD]@}|jjD].}|jr0t|jtsB|jrt|jtr|Vqq dS)z8Return fields with referencing message types as strings.N)r r<r8r;rr=r>r?)r$rBrCrrrrJs z_FileInfo.unresolved_fieldsN)r,)r)r3 __qualname__r classmethodr staticmethodrr'r*r1rDrLpropertyrJrrrrrs   A)r r r rr r)rrloggingZgoogle.protobufrrrrZproto.marshal.rules.messager getLoggerr- namedtuplerrrrrs