o SD©i~Tã@s<ddlmZddlmZddlmZmZddlmZddl m Z ddlmZddl mZdd lmZed ƒZdd„ZGd d„de ƒZGdd„dƒZGdd„de ƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deeƒZGdd„deƒZGdd „d eƒZGd!d"„d"eƒZGd#d$„d$eƒZeZ d S)%é)ÚSequence)ÚList)ÚfalseÚtrue)Ú DDLElement)Ú UpdateBase)Ú ClauseElement)ÚFromClauseRoleé)Ústring_typesNcCs|rtƒStƒS©N)rr)Úbln©rú]/var/www/Datamplify/venv/lib/python3.10/site-packages/snowflake/sqlalchemy/custom_commands.pyÚtranslate_boolsrc@sLeZdZdZdZdd„ZGdd„deƒZdd„Zd d „Z dd„Z d d„ZdS)Ú MergeIntoÚ merge_intoNcCs||_||_||_g|_dSr)ÚtargetÚsourceÚonÚclauses)ÚselfrrrrrrÚ__init__ó zMergeInto.__init__c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S)zMergeInto.clauseÚmerge_into_clausecCsi|_d|_||_dSr)ÚsetÚ predicateÚcommand)rrrrrr&s zMergeInto.clause.__init__c Cs¨|jdur dt|jƒ›nd}|jdkr/t|j ¡Ž\}}d ||jd |¡d tt|ƒ¡¡S|jr?d dd„|j ¡Dƒ¡nd}d ||j|jrQd t|ƒ›¡Sd¡S) Nz AND ÚÚINSERTz+WHEN NOT MATCHED{} THEN {} ({}) VALUES ({})z, cSs"g|] }|d›d|d›‘qS)rz = r r)Ú.0rrrrÚ :s"z-MergeInto.clause.__repr__..zWHEN MATCHED{} THEN {}{}z SET ) rÚstrrÚziprÚitemsÚformatÚjoinÚmap)rÚcase_predicateÚsetsÚsets_tosrrrÚ__repr__+s*ÿ ü ÿýýýzMergeInto.clause.__repr__cKó ||_|Sr)r)rÚkwargsrrrÚvaluesDózMergeInto.clause.valuescCr,r)r)rÚexprrrrÚwhereHr/zMergeInto.clause.whereN)Ú__name__Ú __module__Ú__qualname__Ú__visit_name__rr+r.r1rrrrÚclause#sr6cCsFd dd„|jDƒ¡}d|j›d|j›d|j›|r d|›SdS)Nú cSsg|]}t|ƒ‘qSr)Úrepr)r r6rrrr!Msz&MergeInto.__repr__..zMERGE INTO z USING z ON r)r&rrrr)rrrrrr+LsÿÿzMergeInto.__repr__cCó| d¡}|j |¡|S)NÚUPDATE©r6rÚappend©rr6rrrÚwhen_matched_then_updateRó z"MergeInto.when_matched_then_updatecCr9)NÚDELETEr;r=rrrÚwhen_matched_then_deleteWr?z"MergeInto.when_matched_then_deletecCr9)Nrr;r=rrrÚwhen_not_matched_then_insert\r?z&MergeInto.when_not_matched_then_insert)r2r3r4r5Ú_bindrrr6r+r>rArBrrrrrs)rc@s*eZdZdZdeefdd„Zdd„ZdS)ÚFilesOptionzO Class to represent FILES option for the snowflake COPY INTO statement Ú file_namescCs ||_dSr)rE©rrErrrrgs zFilesOption.__init__cCs"dd„|jDƒ}dd |¡›dS)NcSs g|]}d| dd¡d‘qS)ú'z\')Úreplace)r Úfrrrr!ks z'FilesOption.__str__..ú(ú,ú))rEr&)rÚ the_filesrrrÚ__str__jszFilesOption.__str__N)r2r3r4Ú__doc__rr"rrNrrrrrDbsrDc@sZeZdZdZdZdZddd„Zdd„Zdd „Zd d„Z dd „Z dd„Zdd„Zdd„Z dS)ÚCopyIntoz…Copy Into Command base class, for documentation see: https://docs.snowflake.net/manuals/sql-reference/sql/copy-into-location.htmlÚ copy_intoNcCs"||_||_||_i|_||_dSr)Úfrom_ÚintoÚ formatterÚcopy_optionsÚpartition_by)rrRrSrVrTrrrrvó zCopyInto.__init__cCsPd|j›dt|jƒ›}|jdur|d|j›7}|dt|jƒ›d|j›dS)ú… repr for debugging / logging purposes only. For compilation logic, see the corresponding visitor in base.py z COPY INTO z FROM Nz PARTITION BY r7ú (rL)rSr8rRrVrTrU)rÚvalrrrr+}s zCopyInto.__repr__cCsdSrr©rrrrÚbindˆsz CopyInto.bindcCó*t|tƒs tdƒ‚|j dt|ƒi¡|S)Nz)Parameter force should be a boolean valueÚFORCE©Ú isinstanceÚboolÚ TypeErrorrUÚupdater)rÚforcerrrrd‹ó zCopyInto.forcecCr])Nz0Parameter single_file should be a boolean valueÚSINGLEr_)rÚsingle_filerrrÚsingle‘rezCopyInto.singlecCs&t|tƒs tdƒ‚|j d|i¡|S)Nz-Parameter max_size should be an integer valueÚ MAX_FILE_SIZE)r`ÚintrbrUrc)rÚmax_sizerrrÚmaxfilesize—s zCopyInto.maxfilesizecCs|j dt|ƒi¡|S)NÚFILES)rUrcrDrFrrrÚfilesszCopyInto.filescCs|j d|i¡|S)NÚPATTERN)rUrc)rÚpatternrrrrp¡szCopyInto.pattern)NN)r2r3r4rOr5rCrr+r\rdrhrlrnrprrrrrPos rPc@s2eZdZdZdZd dd„Zdd„Zedd „ƒZdS)Ú CopyFormatterz~ Base class for Formatter specifications inside a COPY INTO statement. May also be used to create a named format. Úcopy_formatterNcCstƒ|_|r ||jd<dSdS)NÚformat_name)ÚdictÚoptions)rrsrrrr®sÿzCopyFormatter.__init__cCsd|j›dS)rXz FILE_FORMAT=(rL)rur[rrrr+³szCopyFormatter.__repr__cCsP|dkr|St|tƒrd|›dSt|tƒr$t|ƒdkr$d|d›dSt|ƒS)a$ Make a SQL-suitable representation of "value". This is called from the corresponding visitor function (base.py/visit_copy_formatter()) - in case of a format name: return it without quotes - in case of a string: enclose in quotes: "value" - in case of a tuple of length 1: enclose the only element in brackets: (value) Standard stringification of Python would append a trailing comma: (value,) which is not correct in SQL - otherwise: just convert to str as is: value rsrGr z('rz'))r`r"ÚtupleÚlen)ÚnameÚvaluerrrÚ value_reprºs zCopyFormatter.value_reprr) r2r3r4rOr5rr+Ústaticmethodrzrrrrrq¦s rqc@seZdZdZdd„Zdd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"S)#ÚCSVFormatterÚcsvcCó>t|tƒr | ¡}gd¢}||vrtd|›ƒ‚||jd<|S©zqString (constant) that specifies to compresses the unloaded data files using the specified compression algorithm.)ÚautoÚgzipÚbz2ÚbrotliÚzstdÚdeflateÚraw_deflateNz$Compression type should be one of : ÚCOMPRESSION©r`rÚlowerrbru©rÚ comp_typeÚ_available_optionsrrrÚcompressionÓó zCSVFormatter.compressioncCsRt|tƒrdSt|tƒr| ¡ d¡}t|ƒdkrdSt|tƒr"dSt|›dƒ‚)z÷ Check if a delimiter is either a string of length 1 or an integer. In case of a string delimiter, take into account that the actual string may be longer, but still evaluate to a single character (like "\n" or r" " NÚunicode_escaper zC should be a single character, that is either a string, or a number)r`ÚNoneTyperÚencodeÚdecoderwrjrb)rÚ delimiterÚ delimiter_txtÚdelimiter_processedrrrÚ_check_delimiteræs ÿzCSVFormatter._check_delimitercCó6| |d¡t|tƒrt|ƒ|jd<|S||jd<|S)z5Character that separates records in an unloaded file.zRecord delimiterÚRECORD_DELIMITER©r–r`rjÚhexru©rÚ deli_typerrrÚrecord_delimiterøó ÿzCSVFormatter.record_delimitercCr—)z4Character that separates fields in an unloaded file.zField delimiterÚFIELD_DELIMITERr™r›rrrÚfield_delimiterržzCSVFormatter.field_delimitercCó$t|ttfƒstdƒ‚||jd<|S©zÝString that specifies the extension for files unloaded to a stage. Accepts any extension. The user is responsible for specifying a valid file extension that can be read by the desired software or service. z!File extension should be a stringÚFILE_EXTENSION©r`rrrbru©rÚextrrrÚfile_extension ó zCSVFormatter.file_extensioncCó t|tƒs tdƒ‚||jd<|S)zIString that defines the format of date values in the unloaded data files.zDate format should be a stringÚDATE_FORMAT©r`rrbru)rÚdt_frmtrrrÚdate_formató zCSVFormatter.date_formatcCr©)zIString that defines the format of time values in the unloaded data files.zTime format should be a stringÚTIME_FORMATr«)rÚtm_frmtrrrÚtime_formatr®zCSVFormatter.time_formatcCr©)zNString that defines the format of timestamp values in the unloaded data files.z#Timestamp format should be a stringÚTIMESTAMP_FORMATr«)rÚtmstmp_frmtrrrÚtimestamp_format!r®zCSVFormatter.timestamp_formatcCr~)zCharacter used as the escape character for any field values. The option can be used when unloading data from binary columns in a table.)ršÚbase64Úutf8z!Binary format should be one of : Ú BINARY_FORMATrˆ)rÚbin_fmtrŒrrrÚ binary_format(s zCSVFormatter.binary_formatcCr—)z.rcsrrrrrrrr\rz''azure://{}.blob.core.windows.net/{}{}'rçrr)r%r&rr$rr9rýrérrrrr+Ws$ÿ þÿÿýýzAzureContainer.__repr__cCsd|i|_|S)NÚAZURE_SAS_TOKEN)r)rÚazure_sas_tokenrrrrjr2zAzureContainer.credentialscCr&)NÚ AZURE_CSEr(r+r,rrrÚencryption_azure_csenr/z#AzureContainer.encryption_azure_cser)r2r3r4rOr5rrðr r+rr?rrrrr7=s r7)!Úcollections.abcrÚtypingrÚ sqlalchemyrrÚsqlalchemy.sql.ddlrÚsqlalchemy.sql.dmlrÚsqlalchemy.sql.elementsrÚsqlalchemy.sql.rolesr ÚcompatrÚtyperrrrDrPrqr|rÓrÖràrñrúrÿr7ÚCopyIntoStoragerrrrÚs0I 7*#"%H6