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)ÚTrie)Ú	text_typeé   c               @   sV   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d„ Z
dd„ ZdS )r   c             C   sv   t ƒ }x:|jƒ D ].}t|tƒs&tdƒ‚x|D ]}|j|ƒ q,W qW tdj|ƒƒ| _x|j	ƒ D ]\}}|| j|< q\W d S )NzAll keys must be stringsÚ )
ÚsetÚkeysÚ
isinstancer   Ú	TypeErrorÚaddÚDATrieÚjoinÚ_dataÚitems)ÚselfÚdataÚcharsÚkeyÚcharÚvalue© r   ú;/tmp/pip-install-q3hcpn_q/html5lib/html5lib/_trie/datrie.pyÚ__init__
   s    

zTrie.__init__c             C   s
   || j kS )N)r   )r   r   r   r   r   Ú__contains__   s    zTrie.__contains__c             C   s
   t | jƒS )N)Úlenr   )r   r   r   r   Ú__len__   s    zTrie.__len__c             C   s
   t ƒ ‚d S )N)ÚNotImplementedError)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   | j j|ƒS )N)r   r
   )r   Úprefixr   r   r   r
   "   s    z	Trie.keysc             C   s   | j j|ƒS )N)r   Úhas_keys_with_prefix)r   r!   r   r   r   r"   %   s    zTrie.has_keys_with_prefixc             C   s   | j j|ƒS )N)r   Úlongest_prefix)r   r!   r   r   r   r#   (   s    zTrie.longest_prefixc             C   s   | j j|ƒS )N)r   Úlongest_prefix_item)r   r!   r   r   r   r$   +   s    zTrie.longest_prefix_item)N)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r    r
   r"   r#   r$   r   r   r   r   r   	   s   
r   N)Ú
__future__r   r   r   Zdatrier   r   Úsixr   Ú_baseÚABCTrier   r   r   r   Ú<module>   s   