o
    !§Ñhï  ã                   @   sL   d dl mZmZmZ d dlmZ d dlmZ ddlm	Z
 G dd„ de
ƒZ	dS )	é    )Úabsolute_importÚdivisionÚunicode_literals)Ú	text_type)Úbisect_lefté   )ÚTriec                   @   sF   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zddd„Zdd„ Z	dS )r   c                 C   sJ   t dd„ | ¡ D ƒƒstdƒ‚|| _t| ¡ ƒ| _d| _dt|ƒf| _d S )Nc                 s   s   � | ]}t |tƒV  qd S ©N)Ú
isinstancer   )Ú.0Úx© r   ú?/usr/lib/python3/dist-packages/pip/_vendor/html5lib/_trie/py.pyÚ	<genexpr>   s   € z Trie.__init__.<locals>.<genexpr>zAll keys must be stringsÚ r   )	ÚallÚkeysÚ	TypeErrorÚ_dataÚsortedÚ_keysÚ	_cachestrÚlenÚ_cachepoints)ÚselfÚdatar   r   r   Ú__init__
   s   zTrie.__init__c                 C   s
   || j v S r	   ©r   ©r   Úkeyr   r   r   Ú__contains__   ó   
zTrie.__contains__c                 C   ó
   t | jƒS r	   )r   r   ©r   r   r   r   Ú__len__   r!   zTrie.__len__c                 C   r"   r	   )Úiterr   r#   r   r   r   Ú__iter__   r!   zTrie.__iter__c                 C   s
   | j | S r	   r   r   r   r   r   Ú__getitem__   r!   zTrie.__getitem__Nc                 C   sÀ   |d u s|dks| j st| j ƒS | | j¡r&| j\}}t| j |||ƒ }}nt| j |ƒ }}tƒ }|t| j ƒkr:|S | j |  |¡rV| | j | ¡ |d7 }| j |  |¡sB|| _||f| _|S )Nr   r   )r   ÚsetÚ
startswithr   r   r   r   Úadd)r   ÚprefixÚloÚhiÚstartÚir   r   r   r   r      s    

þ
z	Trie.keysc                 C   sd   || j v rdS | | j¡r| j\}}t| j|||ƒ}nt| j|ƒ}|t| jƒkr*dS | j|  |¡S )NTF)r   r)   r   r   r   r   r   )r   r+   r,   r-   r/   r   r   r   Úhas_keys_with_prefix6   s   

zTrie.has_keys_with_prefixr	   )
Ú__name__Ú
__module__Ú__qualname__r   r    r$   r&   r'   r   r0   r   r   r   r   r   	   s    	
r   N)Ú
__future__r   r   r   Úpip._vendor.sixr   Úbisectr   Ú_baser   ÚABCTrier   r   r   r   Ú<module>   s
    