o
    `^h	                     @   sr   d dl Z d dlmZ d dlmZ d dlmZ d dlmZ G dd de jZ	G dd	 d	e j
e	d
ZG dd dZdS )    N)descriptor_pb2)
_file_info)_package_info)EnumRulec                       s    e Zd ZdZ fddZ  ZS )ProtoEnumMetaz8A metaclass for building and registering protobuf enums.c                    s  |d t jkrt | |||S t||\}}t|d|d}d|v r<|	d}|d |d  ||d d   }d
|f| d}tj|d| }	d}
||
i }|
|jv rwt|jtrs|j	|
}|j| n|j|
= tj|tdd	 | D d
d d|d}tj|	|}t|dkr|jj | n||j|< t | |||}t|d d|_ ||j!|< |"|t#| |j$|dr|j%||d |S )Nr   __qualname__.z<locals>   
__module___pb_optionsc                 s   s,    | ]\}}t |trtj||d V  qdS ))namenumberN)
isinstanceintr   EnumValueDescriptorProto).0r   r    r   I/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/proto/enums.py	<genexpr>I   s    
z(ProtoEnumMeta.__new__.<locals>.<genexpr>c                 S   s   | j S N)r   )vr   r   r   <lambda>O   s    z'ProtoEnumMeta.__new__.<locals>.<lambda>)key)r   valueoptions	full_namepb)	new_class)r   fallback_salt)&enumIntEnumsuper__new__r   compiletuplegetsplitindexjoinlstripr   	_FileInfoproto_file_namelowerpop_member_namesr   listr   EnumDescriptorProtosorteditemsmaybe_add_descriptorlen
descriptor	enum_typeadd	MergeFromnested_enum	_EnumInfo_metaenumsregisterr   readygenerate_file_pb)mclsr   basesattrspackagemarshal
local_pathixr   filename
pb_optionsoptsidx	enum_desc	file_infocls	__class__r   r   r#      sN   
 
	

zProtoEnumMeta.__new__)__name__r
   r   __doc__r#   __classcell__r   r   rO   r   r      s    r   c                   @   sP   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd ZdS )Enumz:A enum object that also builds a protobuf enum descriptor.c                 C   s   t |t | tfv S r   )typer   selfotherr   r   r   _comparabler   s   zEnum._comparablec                 C   s
   t | jS r   )hashr   )rW   r   r   r   __hash__v   s   
zEnum.__hash__c                 C   s   |  |stS | jt|kS r   rY   NotImplementedr   r   rV   r   r   r   __eq__y      
zEnum.__eq__c                 C   s   |  |stS | jt|kS r   r\   rV   r   r   r   __ne__   r_   zEnum.__ne__c                 C   s   |  |stS | jt|k S r   r\   rV   r   r   r   __lt__   r_   zEnum.__lt__c                 C   s   |  |stS | jt|kS r   r\   rV   r   r   r   __le__   r_   zEnum.__le__c                 C   s   |  |stS | jt|kS r   r\   rV   r   r   r   __ge__   r_   zEnum.__ge__c                 C   s   |  |stS | jt|kS r   r\   rV   r   r   r   __gt__   r_   zEnum.__gt__N)rQ   r
   r   rR   rY   r[   r^   r`   ra   rb   rc   rd   r   r   r   r   rT   o   s    rT   )	metaclassc                   @   s   e Zd ZdefddZdS )r;   r   c                C   s   || _ || _d S r   r   )rW   r   r   r   r   r   __init__   s   
z_EnumInfo.__init__N)rQ   r
   r   strrf   r   r   r   r   r;      s    r;   )r    google.protobufr   protor   r   proto.marshal.rules.enumsr   EnumMetar   r!   rT   r;   r   r   r   r   <module>   s   W/