o 8D©i=8ã@s^ddlmZddlmZddlmZmZmZmZm Z ddl m Z dd d „Z Gd d „d e ƒZ d S)é)Ú annotations)Úexp)Úapprox_count_distinct_sqlÚbuild_timestamp_truncÚ rename_funcÚ time_formatÚ unit_to_str)ÚMySQLÚ expressionúexp.Lag | exp.LeadÚreturnÚstrcCsF| t|tjƒr dnd|j|j d¡ptj d¡|j d¡p!t  ¡¡S)NÚLAGÚLEADÚoffsetéÚdefault) ÚfuncÚ isinstancerÚLagÚthisÚargsÚgetÚLiteralÚnumberÚnull)Úselfr ©rúO/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/dialects/doris.pyÚ _lag_lead_sqls ürc@s<eZdZdZdZdZGdd„dejƒZGdd„dejƒZdS) ÚDorisz 'yyyy-MM-dd'z 'yyyyMMdd'z'yyyy-MM-dd HH:mm:ss'c@sNeZdZiejj¥ejje ej jej jej jdœ¥Zejj  ¡Z e  d¡dS)z Doris.Parser)Ú COLLECT_SETÚ DATE_TRUNCÚ MONTHS_ADDÚREGEXPÚTO_DATEÚ GROUP_CONCATN)Ú__name__Ú __module__Ú __qualname__r ÚParserÚ FUNCTIONSrÚArrayUniqueAggÚ from_arg_listrÚ AddMonthsÚ RegexpLikeÚ TsOrDsToDateÚFUNCTION_PARSERSÚcopyÚpoprrrrr*sÿú r*c@s‚eZdZdZdZiejj¥ej j j dej j j dej j j di¥ZiZeƒZiejj¥iejedƒ“eje“ejedƒ“ejedƒ“ejedƒ“ejedƒ“ejed ƒ“ejd d „“ejd d „“ejd d „“ejdd „“eje “ej!e “ej"edƒ“ej#edƒ“ej$edƒ“ej%edƒ“¥ej&edƒej'dd „ej(edƒej)dd „ej*dd „ej+edƒej,dd „ej-dd „ej.edƒi ¥Zhd£Z/dS)zDoris.GeneratorFÚSTRINGÚDATETIMEr#ÚMAX_BYÚMIN_BYÚ COLLECT_LISTÚ ARRAY_JOINr!cCs | d¡S)NÚNOW)r)rÚ_rrrÚ@s zDoris.Generator.cCó| d|jt|ƒ¡S©Nr"©rrr©rÚerrrr<AócCs$| d|j|j d¡ptj d¡¡S)Nr&Ú separatorú,)rrrrrrÚstringr@rrrr<BsÿcCó| d|j|j¡S)NÚ JSON_EXTRACT©rrr r@rrrr<EóÚ ARRAY_MAPr$ÚSPLIT_BY_STRINGcCs| d|j| |¡¡S)NÚUNIX_TIMESTAMP)rrÚ format_timer@rrrr<Msr%cCrF)NÚDATE_ADDrHr@rrrr<OrIcCs| d|j¡S)Nr%)rrr@rrrr<PsrLcCr=r>r?r@rrrr<RrBcCs| d|jtdƒ||ƒ¡S)NÚ FROM_UNIXTIMEÚdoris)rrrr@rrrr<SsÿrO>ÖÚasÚatÚbyÚdoÚifÚinÚisÚnoÚofÚonÚorÚs3ÚtoÚaddÚallÚandÚascÚbinÚdayÚdivÚendÚforÚhllÚhubÚintÚjobÚkeyÚmapÚmaxÚminÚnotÚrowÚsetÚsumÚuseÚxorÚantiÚautoÚblobÚbulkÚcallÚcaseÚcastÚcharr2ÚcronÚcubeÚdataÚdateÚdescÚdiskÚdropÚdualÚelseÚendsÚfastÚfileÚfreeÚfromÚfullÚhashÚhdfsÚhelpÚhourÚintoÚipv4Úipv6ÚjobsÚjoinÚjsonÚkeysÚkillÚlastÚldapÚleftÚlessÚlikeÚlinkÚlistÚloadÚlockÚmemoÚmtmvÚnameÚnextrÚonlyÚopenÚoverÚpathÚplanÚprocÚreadÚrealÚroleÚrowsÚsemiÚsetsÚshowÚskewÚstopÚsyncÚtaskÚtextÚthanÚthenÚtimeÚtreeÚtrimÚtrueÚtypeÚuserÚviewÚweekÚwhenÚwithÚworkÚyearÚadminÚafterÚaliasÚalterÚarrayÚbeginÚbitorÚbriefÚbuildÚchainÚcheckÚcleanÚcountÚcrossÚdroppÚenterÚeveryÚfalseÚfirstÚfloatÚforceÚgrantÚgraphÚgroupÚindexÚinnerÚjsonbÚlabelÚlevelÚlimitÚlinesÚlocalÚmatchÚmergeÚminusÚmonthÚnamesÚneverÚnullsÚorderÚouterÚpauseÚqueryÚquotaÚrangeÚrightÚrlikeÚrolesÚshapeÚsplitÚstartÚstatsÚtableÚtasksÚtrashÚtypesÚunionÚusingÚvalueÚvaultÚwhereÚwriteÚappendÚbackupÚbelongÚbigintÚbinaryÚbinlogÚbitandÚbitmapÚbitxorÚbrokerÚcachedÚcancelÚcolumnÚcommitÚconfigÚcreateÚdatev1Údatev2ÚdeleteÚdemandÚdoubleÚenableÚengineÚerrorsÚeventsÚexceptÚexistsÚexportÚfieldsÚfilterÚformatÚglobalÚgrantsÚgroupsÚhavingÚignoreÚinfileÚinsertÚisnullÚmanualÚminuteÚmodifyrÚparsedÚperiodÚpluginÚpolicyÚrandomÚregexpÚrenameÚrepairÚresumeÚrevokeÚrollupÚsampleÚschemaÚsecondÚselectÚsignedÚsonameÚstartsÚstatusÚstreamrEÚstructÚswitchÚsystemÚtablesÚtabletÚuniqueÚunlockÚupdateÚvaluesÚadddateÚanalyzeÚauthorsÚbackendÚbetweenÚbooleanÚbucketsÚbuiltinÚcatalogÚcharsetÚclusterÚcollateÚcollectÚcolumnsÚcommentÚcompactÚconvertÚcurrentÚdateaddÚdecimalrÚdynamicÚenginesÚexcludeÚexecuteÚexpiredÚexplainÚextractÚfeatureÚforeignÚgenericÚindexesÚinstallÚintegerÚlateralÚlogicalÚmigrateÚnaturalÚoutfileÚpercentÚpluginsÚprepareÚprimaryÚprocessÚprofileÚrecoverÚrecycleÚrefreshÚreleaseÚreplaceÚreplicaÚrestoreÚreturnsÚroutineÚschemasÚsessionÚstorageÚsubdateÚtabletsÚtinyintÚvarcharÚvariantÚverboseÚversionÚanalyzedÚbackendsÚcatalogsÚclustersÚcompleteÚcreationÚdatabaseÚdate_addÚdate_subÚdatediffÚdatetimeÚdays_addÚdays_subÚdeferredÚdescribeÚdiagnoseÚdistinctÚextendedÚexternalÚfollowerÚfrontendÚfunctionÚgroupingÚhostnameÚintervalÚinvertedÚlargeintÚlocationÚmaxvalueÚnegativeÚngram_bfÚobserverÚpasswordÚphysicalÚpropertyÚresourceÚrollbackÚscheduleÚsequenceÚsmallintÚsnapshotÚtriggersÚtruncateÚunsignedÚwarningsÚworkloadÚ agg_stateÚ aggregateÚ characterÚ collationÚ committedÚ databasesÚ date_ceilÚ date_diffÚ decimalv2Ú decimalv3Ú duplicateÚ followingÚ frontendsÚ functionsÚ histogramÚ hll_unionÚ immediateÚ intersectÚ isolationÚ localtimeÚ match_allÚ match_anyÚ optimizedÚ overwriteÚ parameterÚ partitionÚ precedingÚ procedureÚ rebalanceÚ resourcesÚ rewrittenÚ schedulerÚ streamingÚ superuserÚ temporaryÚ timestampÚ type_castÚ unboundedÚ uninstallÚ variablesÚ whitelistÚ connectionÚ consistentÚ constraintÚ date_floorÚ datetimev1Ú datetimev2Ú distinctpcÚ encryptkeyÚ identifiedÚ migrationsÚ partitionsÚ permissiveÚ propertiesÚ referencesÚ repeatableÚ repositoryÚ terminatedÚ array_rangeÚ constraintsÚ distributedÚ encryptkeysÚ incrementalÚ processlistÚ restrictiveÚ tablesampleÚ transactionÚ uncommittedÚ account_lockÚ bitmap_unionÚ current_dateÚ current_timeÚ current_userÚ decommissionÚ distinctpcsaÚ distributionÚ intermediateÚ is_null_predÚ low_priorityÚ match_phraseÚ match_regexpÚ materializedÚ non_nullableÚ repositoriesÚ serializableÚ timestampaddÚ connection_idÚ timestampdiffÚaccount_unlockÚauto_incrementÚlocaltimestampÚpassword_reuseÚquantile_stateÚquantile_unionÚsql_block_ruleÚcurrent_catalogÚpassword_expireÚis_not_null_predÚpassword_historyÚcurrent_timestampÚmatch_phrase_edgeÚpassword_lock_timeÚldap_admin_passwordÚmatch_phrase_prefixÚreplace_if_not_nullÚfailed_login_attemptsÚdoris_internal_table_idN)0r'r(r)ÚLAST_DAY_SUPPORTS_DATE_PARTÚVARCHAR_REQUIRES_SIZEr Ú GeneratorÚ TYPE_MAPPINGrÚDataTypeÚTypeÚTEXTÚ TIMESTAMPÚ TIMESTAMPTZÚ CAST_MAPPINGrqÚTIMESTAMP_FUNC_TYPESÚ TRANSFORMSr.rÚApproxDistinctrÚArgMaxÚArgMinÚArrayAggÚ ArrayToStringr,ÚCurrentTimestampÚ DateTruncÚ GroupConcatÚJSONExtractScalarrrÚLeadÚMapr/Ú RegexpSplitÚSplitÚ StringToArrayÚ StrToUnixÚ TimeStrToDateÚ TsOrDsAddr0Ú TimeToUnixÚTimestampTruncÚ UnixToStrÚ UnixToTimeÚRESERVED_KEYWORDSrrrrr$)sxÿ   üÿ þý ü û ú ù ø ÷ ö õ òñð ï î í ì         á #r$N) r'r(r)Ú DATE_FORMATÚDATEINT_FORMATÚ TIME_FORMATr r*r$rrrrr s  r N)r r r r )Ú __future__rÚsqlglotrÚsqlglot.dialects.dialectrrrrrÚsqlglot.dialects.mysqlr rr rrrrÚs