o 8D©iÅã@s„ddlZddlmZmZejejZGdd„deƒZd dej edej ej dej fdd „Z dej d edej eej ffd d „ZdS)éN)ÚEnumÚautoc@seZdZeƒZeƒZeƒZdS)Ú TrieResultN)Ú__name__Ú __module__Ú __qualname__rÚFAILEDÚPREFIXÚEXISTS©r r úE/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/trie.pyrs rÚkeywordsÚtrieÚreturncCs@|durin|}|D]}|}|D]}| |i¡}qd|d<q |S)a< Creates a new trie out of a collection of keywords. The trie is represented as a sequence of nested dictionaries keyed by either single character strings, or by 0, which is used to designate that a keyword is in the trie. Example: >>> new_trie(["bla", "foo", "blab"]) {'b': {'l': {'a': {0: True, 'b': {0: True}}}}, 'f': {'o': {'o': {0: True}}}} Args: keywords: the keywords to create the trie from. trie: a trie to mutate instead of creating a new one Returns: The trie corresponding to `keywords`. NTr)Ú setdefault)r rÚkeyÚcurrentÚcharr r r Únew_trie s rrcCsV|stj|fS|}|D]}||vrtj|fS||}q d|vr&tj|fStj|fS)a& Checks whether a key is in a trie. Examples: >>> in_trie(new_trie(["cat"]), "bob") (, {'c': {'a': {'t': {0: True}}}}) >>> in_trie(new_trie(["cat"]), "ca") (, {'t': {0: True}}) >>> in_trie(new_trie(["cat"]), "cat") (, {0: True}) Args: trie: The trie to be searched. key: The target key. Returns: A pair `(value, subtrie)`, where `subtrie` is the sub-trie we get at the point where the search stops, and `value` is a TrieResult value that can be one of: - TrieResult.FAILED: the search was unsuccessful - TrieResult.PREFIX: `value` is a prefix of a keyword in `trie` - TrieResult.EXISTS: `key` exists in `trie` r)rrr r )rrrrr r r Úin_trie+s    r)N)ÚtypingÚtÚenumrrÚSequenceÚHashablerrÚIterableÚOptionalÚDictrÚTuplerr r r r Ús  ((