3
œ(ƒ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„ |jƒ D ƒƒstdƒ‚|| _t|jƒ ƒ| _d| _dt|ƒf| _d S )Nc             s   s   | ]}t |tƒV  qd S )N)Ú
isinstancer   )Ú.0Úx© r   ú7/tmp/pip-install-q3hcpn_q/html5lib/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  r"t| j ƒS |j| jƒrN| j\}}t| j |||ƒ }}nt| j |ƒ }}tƒ }|t| j ƒkrv|S x,| j | j|ƒr¢|j| j | ƒ |d7 }qxW || _||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| jƒr6| j\}}t| j|||ƒ}nt| j|ƒ}|t| jƒkrTdS | j| 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)Ú
__future__r   r   r   Úsixr   Úbisectr   Ú_baser   ÚABCTrier   r   r   r   Ú<module>   s   