ɯeiC,VddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ejdZ ejdZejeZGddZ d dd Z dd Zdd Zd Zd ddej.f ddZddZddZy)) annotationsN)StringIO)Sequencez^\s*\-\-z^\s*$c"eZdZdZddZdddZy) SQLDelimiterzClass that wraps a SQL delimiter string. Since split_statements is a generator this mutable object will allow it change while executing. c|jS)N sql_delimiter)selfs _/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/connector/util_text.py__str__zSQLDelimiter.__str__s!!!c||_y)z'Initializes SQLDelimiter with a string.Nr )r r s r __init__zSQLDelimiter.__init__s *rN)returnstr);)r rrNone)__name__ __module__ __qualname____doc__r rrr rrs "+rrc# K| t}d}d}d}d}d}|j}t|tr t dg} |dk7rWd} d} t |} |j } |r| |k7rtj| }tjd| drtjd|}ntjd|}tjd| d rtj|d }ntj|d }| } | | k\ru| | krn|s7|s5|s3| j|| | d ft | d k(r | dddk(rg} n/|s|s|r| j|| | d fn|s| j|| | dfn|rA|| djdr&d}|s| j|| | dzdf| dz } | } n| d z } n|r?|| djdr$d}| j|| | dzdf| dz } | } ny| d z } nr|rx|| dk(r| | d z kr|| d z|dfvr| dz } nM|| |k(rF| | d z kr || d z|k7s| | d z k(r$d}| j|| | d zd f| d z } | } n| dz } n| d z } n|| dvrd }|| }| d z } n|| dvr"| j|| | d zd f| d z } | } n|| djdrI| j|| | d f|s| j|| ddfn| jd| d z} | } n]|| djdrQ|| djds=|s| j|| | dzdfn| j|| | df| dz } | } d }n|| djdr$| j|| | dzd f| dz } | } d }n| dkDr%j|| d z | t | zr n j|| | t | zrj|| | t | zd zr| j|| | dzd f| t | z } || dk(r| d z } | d ddz| d d f| d <tj|| dstj|| dr|s| j|| ddf| } | | kr|| dvr| d z } | | kr|| dvrt!| | } g} n| dk(r|| dk(r|t | dkDrt!| g} |j#j%| r|j#dt |  n|j#j#dfn| d z } r|j}|dk7rWt | dkDrt!| yy#t$rYJwxYww)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)bufremove_comments delimiterin_quotech_quote in_commentin_double_dollarsprevious_delimiterline statementcolcol0len_liner escaped_delimRE_STARTRE_ENDs r split_statementsrH!s N HHJ <<>D$=>>I "*t9!// !]6H%HIIm4Mxx}Q/0::-&AB::]O&<=xx}R01 h$?@ c$:;!. h#:%h?P!(($tC.$)?@y>Q.9Q<?b3H(*I'X9J!(($tC.$)?@,!(($tC.%)@A:((.!&J*!(($tcAg*>)FG1HCD1HC":((.(-%$$d4#'&:E%BC1HCD1HCI%hl*S1W (D)991HC#Y(*hl* qMX5(Q,.$)!(($tcAg*>)EFq"q1HC9 *#H#CyH1HC#Y+-$$d4#'&:D%AB1HCD#$Z**40$$d4nd%;<*!(($st*e)<=!((6"Q,CD#$Z**40de9O9O:+!(($tcAg*>)FG!(($tC.%)@A1HCD!%J#$Z**40$$d4#'&:D%AB1HCD(,%QwNN4a#M8J2J#KL"..cC#m:L4L)MN||DsS5G/G!/K$LM$$d4ns&:D%AB3}--C9+1HC-6r]1-=-CYr]STEU,VIbM*//ST ;}?R?RST @ /%,,d34j%-@A&.T#Y+-Eq.T#Y+-E1)<<D "IAX$s)s"29~)5i@@$&  ::<00? %:M(:':;!ZZ\egu %% 1HCuv||~] "*` 9~%i009&s8QW#WA'W*B7W#W W  W W  Wcg}d}|D]N\}}|j|||st|jdk\s:|ddjdv}Pdj |j|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)r0r+r6upperjoin)statement_listvalid_statement_list is_put_or_gettext is_statements r r5r5s M,?l##D)  \c$**,6G16L !HNN,>M? 77' ( . . 0- ??rcd d}|dk(rd}|r@|jddkDr|d|jd}||rdnd}|d|d|}|Sd}|jddkDr||jdd rd}|d|}|S) z,Constructs hostname from region and account.c@|jjdS)Nzcn-)lowerr1)rs r _is_china_regionz,construct_hostname.._is_china_regionswwy##E**rz us-west-2r.rcncomz.snowflakecomputing.r)rWrrbool)findsplit)regionaccountrXtop_level_domainhosts r construct_hostnamercs+  << q a',,s"34G#3F#;4!F8#78H7IJ K ! << q %5gmmC6H6K%L# ./?.@A Krc|jd}t|dkDr+|ddk(r|dd|djd}|S|d}|S|}|S)NrYrglobalr-)r^r+rfind)r` url_partsparsed_accounts r parse_accountrjsq c"I 9~ Q<8 #&q\!il.@.@.EFN  'q\N ! r rcdjt|Dcgc]%}tjj |'c}}dj|||gScc}w)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. r)rNrangerandomRandomchoice)lengthprefixsuffixchoices_ random_parts r random_stringrwsN''E&MRq6==?11':RSK 77FK0 11Ss*AcR|r$tj|jdSdS)Nutf-8)base64 b64encodedecode)xs r _base64_bytes_to_strr~$s%236  A  % %g .==rct|tr|jd}tjd}|j ||j S)NryF)usedforsecurity)r(rencodehashlibmd5updatedigest)rRrs r get_md5_for_integrityr(s@${{7# ++e ,CJJt ::<r)FN)r8rr9r\r:zSQLDelimiter | None)rOzlist[tuple[str, bool]]rztuple[str, bool | None])r_ str | Noner`rrr) rqintrrrrsrrtz Sequence[str]rr)rr)rRz str | bytesrr)) __future__rrzrloggingrnr,stringiortypingrr/r3r4 getLoggerr_loggerrrHr5rcrjascii_lowercaserwr~rrrr rs"  RZZ , 8$ '  H % + + "%)m1 m1m1#m1`@*@@2.$#33 2 2 2 2 2  2$>r