o 8Di=@sddlmZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZddlmZejrIdd lmZdddZGdddeZdS)) annotationsN)exp transforms) NormalizationStrategyconcat_to_dpipe_sqlconcat_ws_to_dpipe_sqldate_delta_sql'generatedasidentitycolumnconstraint_sqljson_extract_segmentsno_tablesample_sql rename_func map_date_part)Postgres)seq_get) TokenTypebuild_convert_timezone)E expr_type t.Type[E]returnt.Callable[[t.List], E]csdfdd }|S)Nargst.ListrrcsFt|dt|dtt|dd}tjur!|dtjd|S)Nr)this expressionunit return_type TIMESTAMP)rr r TsOrDsAddsetDataTypebuild)rexprrR/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/dialects/redshift.py_builders  z#_build_date_delta.._builder)rrrrr')rr)r'r&r(_build_date_deltas r*c@sveZdZejZdZdZdZdZ dZ dZ ie j ddddZ Gd d d e jZGd d d e jZGd dde jZdS)RedshiftFrTz'YYYY-MM-DD HH24:MI:SS'z%bz%Hz%I)MONHH24HHc seZdZiejjddddeejeejeej eej ej j ej j ddej j d ZiejjdddddZd Z d!d"fdd Z d#d$ddZd%dd ZZS)&zRedshift.ParsercCs,tjt|dt|dtdtjddS)Nrrmonthr )rrrr)rr!rvarr#r$rr'r'r(:s  zRedshift.Parser.cCs t|dS)NUTCrr1r'r'r(r2@ cCs&tjt|dt|dptjddS)Nrr,)rr)r StringToArrayrLiteralstringr1r'r'r(r2Gs) ADD_MONTHSCONVERT_TIMEZONEDATEADDDATE_ADDDATEDIFF DATE_DIFFGETDATELISTAGGSPLIT_TO_ARRAYSTRTOLcCs|SN)_parse_approximate_countselfr'r'r(r2OscCs|jtjddS)NT)sysdate)rrCurrentTimestamprEr'r'r(r2Ps) APPROXIMATESYSDATETFNschemabooljoins alias_tokens#t.Optional[t.Collection[TokenType]] parse_bracketis_db_referenceparse_partition consume_pipert.Optional[exp.Expression]c s<|tj}tj|||||d} |r|jtj| ddS| S)N)rKrMrNrPrQT)runpivot)_matchrUNPIVOTsuper _parse_tablerrPivot) rFrKrMrNrPrQrRrSrUtable __class__r'r(rYUs zRedshift.Parser._parse_tablestrictsafet.Optional[bool]cCs0|}|tj|}|jtj|||dS)N)rtor_) _parse_typesrVrCOMMA_parse_bitwiserrTryCast)rFr^r_rarr'r'r(_parse_convertks zRedshift.Parser._parse_convertt.Optional[exp.ApproxDistinct]cCsT|jd}|}t|tjr#t|jtjr#|jtjt |jj ddS| |dS)Nrr)r) _index_parse_function isinstancerCountrDistinctrApproxDistinctr expressions_retreat)rFindexfuncr'r'r(rDss  z(Redshift.Parser._parse_approximate_count)FFNFFFF)rKrLrMrLrNrOrPrLrQrLrRrLrSrLrrTrC)r^rLr_r`rrT)rrg)__name__ __module__ __qualname__rParser FUNCTIONSr*rr! TsOrDsDiffrH from_arg_list GroupConcatFromBaseNO_PAREN_FUNCTION_PARSERSSUPPORTS_IMPLICIT_UNNESTrYrfrD __classcell__r'r'r\r(ru7sB ruc @sneZdZgZgZddgZiejje j e j e j e j e je je je jdZedejjZeddS)zRedshift.Tokenizer\')z(+) HLLSKETCHMINUSSUPERTOPUNLOADVARBYTEzBINARY VARYINGVALUES#N)rrrsrt BIT_STRINGS HEX_STRINGSSTRING_ESCAPESr TokenizerKEYWORDSr JOIN_MARKERrEXCEPTrrCOMMAND VARBINARYpop SINGLE_TOKENScopyr'r'r'r(r|s& rcsreZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZiejjejjjdejjjdejjjdejjjd ejjjd ejjj dejjj!diZiejj"iej#d d ej$e%ej&e'ej(d d ej)dd ej*e+dej,e+dej-dd ej.dd ej/dd ej0e1dej2e3ej4e5dej6e5dej7e1dej8dd ej9e:;e:je:j?e:j@gejAdd ejBdd ejCe1dejDeEejFe+dejGe+dejHdd iZ"e"IejJe"IejKe"IejLe"IejMe"IejNhdZOd8d!d"ZPd9d:fd'd( ZQd;fd*d+ ZRdd3d4ZUd?d6d7ZVZWS)@zRedshift.GeneratorFTTO_HEX JSON_PARSETYPE rINTEGERTIMEr cCs|j|ddS)N ARRAY_CONCAT)name)arrayconcat_sqlrFer'r'r(r2zRedshift.Generator.cCsd||ddS)NzAPPROXIMATE COUNT(DISTINCT r))sqlrr'r'r(r2cCs|jdrdSdS)NrGrJz GETDATE())rgetrr'r'r(r2rr;r=cCs|d|jS)NDISTKEY)rqrrr'r'r(r2rcC ||SrC)naked_propertyrr'r'r(r2r4cCrrC) explode_sqlrr'r'r(r2r4rBJSON_EXTRACT_PATH_TEXTr@c Cs|d|d||dS)NUPPERrr)rqrrr'r'r(r2scCs&|jdrdndd|j|jdS)Ncompoundz COMPOUND zSORTKEY(r)r format_argsrrr'r'r(r2s&cCs ||jd||jdS)Nz LIKE z || '%')rrrrr'r'r(r2s rAcCrrC)_unix_to_time_sqlrr'r'r(r2r4> encrypt asdoinisonorraallandanyascendforlunlzonewnotoffoidoldrawtagtopaz64bothcasecastdescelsefromfullgzipintojoinleftlikelunslzopnullonlyopensometdesthentruetypeuserwhenwitharraybzip2checkcrossdeltafalsegrantgroupilikeinnerlimitminusnullsorderouterpivotrightr[unionusingwhereaes128aes256backupbinarycolumncreatedefragenableencodeexceptfreezehavingignoreisnulloffsetresortselectsystemuniqueunnestwalletanalyseanalyzebetweencollatedefaultdeflatedisableforeignleadingmostly8naturalnotnullofflinepercentplacingprimaryrecoverrespectrestoresimilarrGtext255text32krUverbosewithoutbytedictdelta32kdistinctexplicitidentityintervalmostly16mostly32overlapsparallelsnapshottrailing initially intersect localtime partition readratio rejectlog timestamp constraint deferrable encryption references credentials emptyasnull permissions blanksasnull current_date current_time current_user session_user authorization globaldict256 globaldict64kallowoverwritelocaltimestampcurrent_user_idtruncatecolumnscurrent_timestampr exp.UnnestrstrcCs|j}t|}|dkr|d|dSt|tjtjtjtjr*|ddS| t |d}|j|j dddd }|rG|d |S|S) Nrz+Unsupported number of arguments in UNNEST: rz5Unsupported UNNEST when not used in FROM/JOIN clausesraliascolumnsT)keyflatz AS ) rnlen unsupportedrj find_ancestorrFromJoinSelectrrrr)rFrrnum_argsargrIr'r'r( unnest_sqls zRedshift.Generator.unnest_sqlNexp.Cast safe_prefixt.Optional[str]cs,|tjjjr||dStj||dS)Nr)rW)is_typerr#TypeJSONrrXcast_sql)rFrrWr\r'r(r\s zRedshift.Generator.cast_sql exp.DataTypecsJ|dr|dtjjj|jd}|s|dt dt |S)a} Redshift converts the `TEXT` data type to `VARCHAR(255)` by default when people more generally mean VARCHAR of max length which is `VARCHAR(max)` in Redshift. Therefore if we get a `TEXT` data type without precision we convert it to `VARCHAR(max)` and if it does have precision then we just convert `TEXT` to `VARCHAR`. textrrnMAX) rYr"rr#rZVARCHARrrappendr0rX datatype_sql)rFr precisionr\r'r(rbs   zRedshift.Generator.datatype_sql exp.AlterSetcCsv|j|dd}|rd|dnd}||d}|rd|nd}|j|ddd d }|r0d |nd}d |||S) NT)rLz TABLE PROPERTIES (rrlocationz LOCATION file_formatr)rKrLsepz FILE FORMAT SET)rnr)rFrexprsrerfr'r'r( alterset_sqls zRedshift.Generator.alterset_sql exp.Arraycs&|jdr t|Std||S)Nbracket_notationARRAY)rrrX array_sqlr rFrr\r'r(rns  zRedshift.Generator.array_sql exp.ExplodecCs|ddS)NzUnsupported EXPLODE() functionr)rNror'r'r(rs zRedshift.Generator.explode_sqlexp.UnixToTimecCsT|jd}||j}|dur$|tjjkr$|jr$d|d|d}d|dS)Nscale(z / POWER(10, z))z(TIMESTAMP 'epoch' + z * INTERVAL '1 SECOND')) rrrrr UnixToTimeSECONDSis_intto_py)rFrrrrr'r'r(rs   z$Redshift.Generator._unix_to_time_sql)rrGrrHrC)rrVrWrXrrH)rr]rrH)rrdrrH)rrkrrH)rrprrH)rrqrrH)XrrrsrtLOCKING_READS_SUPPORTED QUERY_HINTSVALUES_AS_TABLETZ_TO_WITH_TIME_ZONENVL2_SUPPORTEDLAST_DAY_SUPPORTS_DATE_PARTCAN_IMPLEMENT_ARRAY_ANYMULTI_ARG_DISTINCTCOPY_PARAMS_ARE_WRAPPEDHEX_FUNCPARSE_JSON_NAMEARRAY_CONCAT_IS_VAR_LENSUPPORTS_CONVERT_TIMEZONE#EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSESUPPORTS_MEDIANALTER_SET_TYPESUPPORTS_DECODE_CASEWITH_PROPERTIES_PREFIXr Generator TYPE_MAPPINGrr#rZBINARYBLOBINTTIMETZ TIMESTAMPTZr ROWVERSION TRANSFORMS ArrayConcatConcatrConcatWsrrmrHDateAddrDateDiffDistKeyPropertyDistStylePropertyExploderzr #GeneratedAsIdentityColumnConstraintr JSONExtractr JSONExtractScalarryHexrRr preprocesseliminate_window_clauseeliminate_distinct_oneliminate_semi_and_anti_joinsunqualify_unnest.unnest_generate_date_array_using_recursive_cteSortKeyProperty StartsWithr6 TableSampler r!rwrtrrZ ParseJSONAnyValueLastDaySHA2RESERVED_KEYWORDSrUr\rbrjrnrrr}r'r'r\r(rs                            *     !  rN)rrrsrtrCASE_INSENSITIVENORMALIZATION_STRATEGYSUPPORTS_USER_DEFINED_TYPES INDEX_OFFSETCOPY_PARAMS_ARE_CSV HEX_LOWERCASEHAS_DISTINCT_ARRAY_CONSTRUCTORS TIME_FORMATr TIME_MAPPINGrurrr'r'r'r(r+)sEr+)rrrr) __future__rtypingtsqlglotrrsqlglot.dialects.dialectrrrrr r r r r sqlglot.dialects.postgresrsqlglot.helperrsqlglot.tokensrsqlglot.parserr TYPE_CHECKINGsqlglot._typingrr*r+r'r'r'r(s ,