o RDiC,@sddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z e dZ e dZeeZGdddZ d1d2ddZd3ddZd4ddZddZd d!d!ejfd5d(d)Zd6d*d+Zd7d/d0ZdS)8) annotationsN)StringIO)Sequencez^\s*\-\-z^\s*$c@s&eZdZdZd ddZd dd d Zd S) SQLDelimiterzClass that wraps a SQL delimiter string. Since split_statements is a generator this mutable object will allow it change while executing. returnstrcCs|jS)N sql_delimiter)selfr V/var/www/Datamplify/venv/lib/python3.10/site-packages/snowflake/connector/util_text.py__str__szSQLDelimiter.__str__;r NonecCs ||_dS)z'Initializes SQLDelimiter with a string.Nr)r r r r r __init__s zSQLDelimiter.__init__N)rr)r)r rrr)__name__ __module__ __qualname____doc__r rr r r r rs rFbufrremove_commentsbool delimiterSQLDelimiter | Noneccsv|durt}d}d}d}d}d}|}t|trtdg} |dkr+d} d} t|} |j} |r7| |krpt| }t d| drMt d|}nt d|}t d| d rft |d }nt |d }| } | | kr| | kr|s|s|s| || | d ft| d kr| dddkrg} n|s|s|r| || | d fn |s| || | dfnf|r|| d drd}|s| || | ddf| d7} | } nA| d 7} n;|r|| d dr d}| || | ddf| d7} | } n| d 7} n|rt|| dkr3| | d kr3|| d |dfvr3| d7} n|| |krn| | d krJ|| d |ksQ| | d krhd}| || | d d f| d 7} | } n| d7} n| d 7} n|| dvrd }|| }| d 7} n|| dvr| || | d d f| d 7} | } n|| d dr| || | d f|s| || ddfn| d| d } | } nN|| d dr|| d ds|s| || | ddfn | || | df| d7} | } d }n|| d dr.| || | dd f| d7} | } d }n| dkrD| || d | t| rn| || | t| r| || | t| d r| || | dd f| t| 7} z|| dkr| d 7} | d dd| d d f| d <Wn t yYnwt || dst || dr|s| || ddf| } | | kr|| dvr| d 7} | | kr|| dvst| V| } g} n;| dkr|| dkrt| dkrt| Vg} || r|dt|  n|dfVn| d 7} qq|}|dks&t| dkr9t| VdSdS)aaSplits a stream into SQL statements (ends with a semicolon) or commands (!...). Args: buf: Unicode data stream. remove_comments: Whether or not to remove all comments (Default value = False). delimiter: The delimiter string that separates SQL commands from each other. Yields: A SQL statement or a command. NFz#Input data must not be binary type.rz\wz^[^\w$]?z^.?z[^\w$]?$z.?$Tz*/z$$\)'")  z--) Tz/*zfile://r>!)rreadline isinstancebytes TypeErrorlenr reescapematchcompileappend startswith IndexErrorCOMMENT_PATTERN_RE EMPTY_LINE_RE_concatenate_statementsstripendswith)rrrin_quotech_quote in_commentin_double_dollarsprevious_delimiterline statementcolcol0len_liner escaped_delimRE_STARTRE_ENDr r r split_statements!s$                     {rDstatement_listlist[tuple[str, bool]]rtuple[str, bool | None]cCsbg}d}|D]!\}}|||dur'|r't|dkr'|dddv}qd||fS)aConcatenate statements. Each statement should be a tuple of statement and is_put_or_get. The is_put_or_get is set to True if the statement is PUT or GET otherwise False for valid statement. None is set if the statement is empty or comment only. Args: statement_list: List of statement parts. Returns: Tuple of statements and whether they are PUT or GET. N)PUTGETr)r/r*r5upperjoin)rEvalid_statement_list is_put_or_gettext is_statementr r r r4s  r4region str | NoneaccountrcCsddd}|dkr d}|r1|d d kr|d |d }||r#d nd }|d |d |}|Sd }|d d krE||d drEd }|d |}|S)z,Constructs hostname from region and account.rrrrcSs|dS)Nzcn-)lowerr0)rTr r r _is_china_regionsz,construct_hostname.._is_china_regionz us-west-2r.rcncomz.snowflakecomputing.rN)rTrrr)findsplit)rQrSrVtop_level_domainhostr r r construct_hostnames  r^cCsT|d}t|dkr&|ddkr |dd|dd}|S|d}|S|}|S)NrWrglobalr-)r[r*rfind)rS url_partsparsed_accountr r r parse_accounts   rd rlengthintprefixsuffixchoices Sequence[str]cs,dfddt|D}d|||gS)aBOur convenience function to generate random string for object names. Args: length: How many random characters to choose from choices. prefix: Prefix to add to random string generated. suffix: Suffix to add to random string generated. choices: A generator of things to choose from. rcsg|] }tqSr )randomRandomchoice).0_rjr r sz!random_string..)rLrange)rfrhrirj random_partr rqr random_stringsrucCs|r t|dSdS)Nutf-8)base64 b64encodedecode)xr r r _base64_bytes_to_str$sr{rO str | bytesr(cCs2t|tr |d}tjdd}|||S)NrvF)usedforsecurity)r'rencodehashlibmd5updatedigest)rOrr r r get_md5_for_integrity(s    r)FN)rrrrrr)rErFrrG)rQrRrSrrr) rfrgrhrrirrjrkrr)rrR)rOr|rr() __future__rrwrloggingrlr+stringiortypingrr.r2r3 getLoggerr_loggerrrDr4r^rdascii_lowercaserur{rr r r r s6        1