B
    n&=[ã  ã               @   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   ú0lib/python3.7/site-packages/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 kS )N)r   )r   Úkeyr   r   r   Ú__contains__   s    zTrie.__contains__c             C   s
   t | jƒS )N)r   r   )r   r   r   r   Ú__len__   s    zTrie.__len__c             C   s
   t | jƒS )N)Úiterr   )r   r   r   r   Ú__iter__   s    zTrie.__iter__c             C   s
   | j | S )N)r   )r   r   r   r   r   Ú__getitem__   s    zTrie.__getitem__Nc             C   s¶   |d ks|dks| j s t| j ƒS | | j¡rL| j\}}t| j |||ƒ }}nt| j |ƒ }}tƒ }|t| j ƒkrt|S x,| j |  |¡r | | j | ¡ |d7 }qvW || _||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 krdS | | j¡r6| j\}}t| j|||ƒ}nt| j|ƒ}|t| jƒkrTd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_prefix)N)
Ú__name__Ú
__module__Ú__qualname__r   r   r   r    r!   r   r*   r   r   r   r   r   	   s   	
r   N)Z
__future__r   r   r   Zsixr   Zbisectr   Z_baser   ZABCTrier   r   r   r   Ú<module>   s   