3 v^ @sddlZddlZddlmZddlmZddlmZejdej Z ejdZ ddZ d d Z d d Zdd dZddZedddZedddZdS)N)deque)contextmanager) text_typeaZ ( (?: # Start of non-capturing group (?:\r\n|\r|\n) | # Match any single newline, or [^\r\n'"]+ | # Match any character series without quotes or # newlines, or "(?:[^"\\]|\\.)*" | # Match double-quoted strings, or '(?:[^'\\]|\\.)*' # Match single quoted strings ) ) z (\r\n|\r|\n)cCsXt|}tj|}dg}x:|D]2}|s*qqtj|r@|jdq|d|7<qW|S)zSplit a string on all unquoted newlines. Unlike str.splitlines(), this will ignore CR/LF/CR+LF if the requisite character is inside of a string.)r SPLIT_REGEXsplit LINE_MATCHmatchappend)stmttextlinesZ outputlinesliner2/usr/lib/python3.6/site-packages/sqlparse/utils.pysplit_unquoted_newlines&s    rcCs8|dkr dS|ddkr4|d|dkr4|dd}|S) z4Helper that removes surrounding quotes from strings.Nr"'r)rrrrr)valrrr remove_quotes8s  rcsfdd}|S)zqFunction decorator to help with recursion :param cls: Classes to not recurse over :return: function csfddS)Ncs0x"|jD]}t|s |q W|dS)N) get_sublists isinstance)ZtlistZsgroup)clsf wrapped_frrrHs  z(recurse..wrap..wrapped_fr)r)r)rrrwrapGszrecurse..wrapr)rrr)rrrecurseAs rcs|}|rt|t r|gn|}|r4t|t r4|gn|}dkrDdS|rVt|rVdS|rttfdd|DrtdS|rtfdd|DrdSdSdS)aHelper function to simplify comparisons Instance, Match and TokenType :param token: :param i: Class or Tuple/List of Classes :param m: Tuple of TokenType & Value. Can be list of Tuple for multiple :param t: TokenType or Tuple/List of TokenTypes :return: bool NFTc3s|]}j|VqdS)N)r ).0pattern)tokenrr cszimt..c3s|]}j|kVqdS)N)ttype)rr#)r!rrr"es)rlistany)r!imtZclsstypesZ mpatternsr)r!rimtSsr*cCsttj||dddS)zCAdvance the iterator n-steps ahead. If n is none, consume entirely.r)maxlenN)r itertoolsislice)iteratornrrrconsumeksr0ccs&|j|7_dV|j|8_dS)N)offset)filter_r/rrrr1psr1rccs&|j|7_dV|j|8_dS)N)indent)r2r/rrrr3wsr3)NNN)r)r)r,re collectionsr contextlibrsqlparse.compatrcompileVERBOSErr rrrr*r0r1r3rrrr s