3 (\@sddlmZddlZddlZddlmZddlmZmZej dej dej dej dd Z dd d Z dd dZ ej dZ ddZddZddZdS))print_functionN) Identifier)TokenErrorz(\w+)$z ([^():,\s]+)$z([^\.():,\s]+)$z ([^\s]+)$)alphanum_underscoreZmany_punctuationsZmost_punctuationsZall_punctuationsrcCs@|sdS|djrdSt|}|j|}|r8|jdSdSdS)a Find the last word in a sentence. >>> last_word('abc') 'abc' >>> last_word(' abc') 'abc' >>> last_word('') '' >>> last_word(' ') '' >>> last_word('abc ') '' >>> last_word('abc def') 'def' >>> last_word('abc def ') '' >>> last_word('abc def;') '' >>> last_word('bac $def') 'def' >>> last_word('bac $def', include='most_punctuations') '$def' >>> last_word('bac \def', include='most_punctuations') '\\\\def' >>> last_word('bac \def;', include='most_punctuations') '\\\\def;' >>> last_word('bac::def', include='most_punctuations') 'def' >>> last_word('"foo*bar', include='most_punctuations') '"foo*bar' rN)isspace cleanup_regexsearchgroup)textincludeZregexZmatchesrI/usr/share/pgadmin4/web/pgadmin/utils/sqlautocomplete/parseutils/utils.py last_words"   rcCs|js d Stj|d}t|j}|dt||}d }x`t|D]T}|jdksj|jrH|jj |krH|j |}dj d d |d|d D}||fSqHWdS)z Find the last sql keyword in an SQL statement Returns the value of the last keyword, and the text of the query with everything after the last keyword stripped NrrANDORNOTBETWEEN(css|] }|jVqdS)N)value).0tokrrr asz$find_prev_keyword..r)Nr)rrrr)Nr) stripsqlparseparselistflattenlenreversedrZ is_keywordupperindexjoin)sqlZn_skipparsedZ flattenedZlogical_operatorstidxrrrrfind_prev_keywordCs    r*z ^\$[^$]*\$$cCstj|}tdd|DS)z4Returns true if the query contains an unclosed quotecss|]}t|VqdS)N)_parsed_is_open_quote)rprrrrpsz is_open_quote..)rrany)r&r'rrr is_open_quoteks r.cCstdd|jDS)Ncss|]}|jtjdVqdS)'$N)r/r0)matchrr)rrrrrrusz(_parsed_is_open_quote..)r-r )r'rrrr+ssr+cCsbtj|d}t|j}|dkr:t|jdtr:|jdS|jtdfddrZt|dSdSdS)asAttempt to parse a (partially typed) word as an identifier word may include a schema qualification, like `schema_name.partial_name` or `schema_name.` There may also be unclosed quotation marks, like `"schema`, or `schema."partial_name` :param word: string representing a (partially complete) identifier :return: sqlparse.sql.Identifier, or None rr")mN) rrr!tokens isinstancerZ token_next_byrparse_partial_identifier)Zwordr,Zn_tokrrrr6xs    r6)r)r)Z __future__rrerZ sqlparse.sqlrZsqlparse.tokensrrcompiler rr*Zdollar_quote_regexr.r+r6rrrrs   0 %