3
�Pf� � @ sL d dl mZmZmZ d dlmZ d dlmZ ddl mZ
G dd� de
�ZdS )� )�absolute_import�division�unicode_literals)�Trie)� text_type� c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zdd d
�Zdd� Z dd� Z
dd� Z d S )r c C sv t � }x:|j� D ].}t|t�s&td��x|D ]}|j|� q,W qW tdj|��| _x |j � D ]\}}|| j|< q\W d S )NzAll keys must be strings� )
�set�keys�
isinstancer � TypeError�add�DATrie�join�_data�items)�self�data�chars�key�char�value� r � /usr/lib/python3.6/datrie.py�__init__
s
z
Trie.__init__c C s
|| j kS )N)r )r r r r r � __contains__ s zTrie.__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 zTrie.__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 zTrie.has_keys_with_prefixc C s | j j|�S )N)r �longest_prefix)r r! r r r r# ( s zTrie.longest_prefixc C s | j j|�S )N)r �longest_prefix_item)r r! r r r r$ + s zTrie.longest_prefix_item)N) �__name__�
__module__� __qualname__r r r r r r
r" r# r$ r r r r r s
r N) Z
__future__r r r Zdatrier r Zpip._vendor.sixr Z_baseZABCTrier r r r � s