관리-도구
편집 파일: symbol_database.cpython-311.pyc
� P�Dg� � � � d Z ddlZddlmZ ddlmZ ddlmZ G d� d� � Z e ej � � �� � Z d � ZdS ) aS A database of Python protocol buffer generated symbols. SymbolDatabase is the MessageFactory for messages generated at compile time, and makes it easy to create new instances of a registered type, given only the type's protocol buffer symbol name. Example usage:: db = symbol_database.SymbolDatabase() # Register symbols of interest, from one or multiple files. db.RegisterFileDescriptor(my_proto_pb2.DESCRIPTOR) db.RegisterMessage(my_proto_pb2.MyMessage) db.RegisterEnumDescriptor(my_proto_pb2.MyEnum.DESCRIPTOR) # The database can be used as a MessageFactory, to generate types based on # their name: types = db.GetMessages(['my_proto.proto']) my_message_instance = types['MyMessage']() # The database's underlying descriptor pool can be queried, so it's not # necessary to know a type's filename to be able to generate it: filename = db.pool.FindFileContainingSymbol('MyMessage') my_message_instance = db.GetMessages([filename])['MyMessage']() # This functionality is also provided directly via a convenience method: my_message_instance = db.GetSymbol('MyMessage')() � N)�api_implementation)�descriptor_pool)�message_factoryc �Z � e Zd ZdZi Zdd�Zd� Zd� Zd� Zd� Z d� Z d � Z d � Z d � Z d � Zd � ZdS )�SymbolDatabasez'A database of Python generated symbols.Nc �: � |pt j � � | _ dS )z!Initializes a new SymbolDatabase.N)r �DescriptorPool�pool)�selfr s ��/builddir/build/BUILD/imunify360-venv-2.4.0/opt/imunify360/venv/lib64/python3.11/site-packages/google/protobuf/symbol_database.py�__init__zSymbolDatabase.__init__I s � ��8��6�8�8�D�I�I�I� c �R � t j d� � t j |� � S )Nz�SymbolDatabase.GetPrototype() is deprecated. Please use message_factory.GetMessageClass() instead. SymbolDatabase.GetPrototype() will be removed soon.)�warnings�warnr �GetMessageClass�r � descriptors r � GetPrototypezSymbolDatabase.GetPrototypeM s2 � � �M� H� I� I� I� � *�:� 6� 6�6r c �R � t j d� � t j |� � S )Nz�Directly call CreatePrototype() is wrong. Please use message_factory.GetMessageClass() instead. SymbolDatabase.CreatePrototype() will be removed soon.)r r r �_InternalCreateMessageClassr s r �CreatePrototypez SymbolDatabase.CreatePrototypeS s2 � � �M� K� L� L� L� � 6�z� B� B�Br c �^ � t j d� � t j || j � � S )Nz�SymbolDatabase.GetMessages() is deprecated. Please use message_factory.GetMessageClassedForFiles() instead. SymbolDatabase.GetMessages() will be removed soon.)r r r �GetMessageClassedForFilesr )r �filess r � GetMessageszSymbolDatabase.GetMessagesY s6 � � �M� G� H� H� H� � 4�U�D�I� F� F�Fr c �R � |j }|| j |< | � |� � |S )a@ Registers the given message type in the local database. Calls to GetSymbol() and GetMessages() will return messages registered here. Args: message: A :class:`google.protobuf.message.Message` subclass (or instance); its descriptor will be registered. Returns: The provided message. )� DESCRIPTOR�_classes�RegisterMessageDescriptor)r �message�descs r �RegisterMessagez SymbolDatabase.RegisterMessage_ s1 � � � �D�!�D�M�$���"�"�4�(�(�(� �Nr c �l � t j � � dk r | j � |� � dS dS )z�Registers the given message descriptor in the local database. Args: message_descriptor (Descriptor): the message descriptor to add. �pythonN)r �Typer �_AddDescriptor)r �message_descriptors r r z(SymbolDatabase.RegisterMessageDescriptorq s= � � � � � �H�,�,� �i� � � 1�2�2�2�2�2� -�,r c �h � t j � � dk r| j � |� � |S )z�Registers the given enum descriptor in the local database. Args: enum_descriptor (EnumDescriptor): The enum descriptor to register. Returns: EnumDescriptor: The provided descriptor. r% )r r& r �_AddEnumDescriptor)r �enum_descriptors r �RegisterEnumDescriptorz%SymbolDatabase.RegisterEnumDescriptor{ s4 � � � � � �H�,�,� �i�"�"�?�3�3�3� �r c �l � t j � � dk r | j � |� � dS dS )z�Registers the given service descriptor in the local database. Args: service_descriptor (ServiceDescriptor): the service descriptor to register. r% N)r r& r �_AddServiceDescriptor)r �service_descriptors r �RegisterServiceDescriptorz(SymbolDatabase.RegisterServiceDescriptor� s= � � � � � �H�,�,� �i�%�%�&8�9�9�9�9�9� -�,r c �l � t j � � dk r | j � |� � dS dS )z�Registers the given file descriptor in the local database. Args: file_descriptor (FileDescriptor): The file descriptor to register. r% N)r r& r �_InternalAddFileDescriptor)r �file_descriptors r �RegisterFileDescriptorz%SymbolDatabase.RegisterFileDescriptor� s< � � � � � �H�,�,� �i�*�*�?�;�;�;�;�;� -�,r c �L � | j | j � |� � S )aw Tries to find a symbol in the local database. Currently, this method only returns message.Message instances, however, if may be extended in future to support other symbol types. Args: symbol (str): a protocol buffer symbol. Returns: A Python class corresponding to the symbol. Raises: KeyError: if the symbol could not be found. )r r �FindMessageTypeByName)r �symbols r � GetSymbolzSymbolDatabase.GetSymbol� s! � � �=���8�8��@�@� A�Ar c �� �� �fd��i }|D ]l}| j � |� � }|j � � � D ]6} �|� � D ](} | j | ||j < �# t $ r Y �%w xY w�7�m|S )a Gets all registered messages from a specified file. Only messages already created and registered will be returned; (this is the case for imported _pb2 modules) But unlike MessageFactory, this version also returns already defined nested messages, but does not register any message extensions. Args: files (list[str]): The file names to extract messages from. Returns: A dictionary mapping proto names to the message classes. Raises: KeyError: if a file could not be found. c 3 �L �K � | V � | j D ]} �|� � D ]}|V � ��dS )zCWalk a message Descriptor and recursively yields all message names.N)� nested_types)r" �msg_desc� nested_desc�_GetAllMessagess �r r> z3SymbolDatabase.GetMessages.<locals>._GetAllMessages� sY �� � � � �j�j�j��'� � �(�*�?�8�4�4� � �K�� � � � � � � r )r �FindFileByName�message_types_by_name�valuesr � full_name�KeyError)r r �result� file_name� file_descr< r" r> s @r r zSymbolDatabase.GetMessages� s� �� �& � � � � � �F�� � � ��)�*�*�9�5�5�i� �5�<�<�>�>� � �(�#�O�H�-�-� � �D� �%)�]�4�%8�F�4�>� "� "��� � � � �D� ���� �� �Ms �A&�& A3 �2A3 )N)�__name__� __module__� __qualname__�__doc__r r r r r r# r r, r0 r4 r8 � r r r r C s� � � � � � �/�/� �(�9� 9� 9� 9�7� 7� 7� C� C� C� G� G� G� � � �$3� 3� 3� � � � :� :� :�<� <� <�B� B� B�$$� $� $� $� $r r )r c � � t S )z#Returns the default SymbolDatabase.)�_DEFAULTrK r r �DefaultrN � s � � �/r ) rJ r �google.protobuf.internalr �google.protobufr r r rN rM rK r r �<module>rQ s� ��>� �: ���� 7� 7� 7� 7� 7� 7� +� +� +� +� +� +� +� +� +� +� +� +�Q� Q� Q� Q� Q� Q� Q� Q�h �>� 6�� 6� 8� 8� 9� 9� 9��� � � � r