9Qga5ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z n #e $rdZ YnwxYw ddl m Z n#e $r ddlm Z YnwxYw ddlmZn #e $rdZYnwxYwdZddlmZdZ ddlmZn #e $rdZYnwxYwd Zd Zd Zd Zd ZdZiZiZiZ Gdde!Z"e"Z#e"Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,e'edLdZ-e'edZ.e'edMdZ/e'edZ0e'edZ1ej2ddkr*e'edNd!Z3e'ed"Z4n)e'edNd#Z3e'ed$Z4e'ed%Z5e'ed&Z6e'edLd'Z7e'ed(Z8e'ed)Z9e'edOd*Z:e'ed+Z;e'ed,Z<e'ed-Z=e'ed.Z>e'ed/Z?d0Z@Gd1d2eAZBGd3d4eBZCej2ddd5krd6ZDnd7ZDe%eGd8d9eCZEe%eGd:d;eCZFe%eGd<d=eAZGe%eGd>d?eAZHe%eGd@dAeBZIe%eGdBdCeBZJe%eGdDdEeAZKe%eGdFdGeAZLede'eejMZMe'eejNZNe'eejOZOe%eejPZPe@e&eGdHdIeZQe&eGdJdKeZRdSdS)PN)Counter)urlparse) TableFunction)%Y-%m-%d %H:%M:%Sz%Y-%m-%d %H:%M:%S.%f%Y-%m-%d%H:%M:%Sz %H:%M:%S.%fz%H:%M)format_date_timec,t|tSN)r SQLITE_DATETIME_FORMATS) date_values t/builddir/build/BUILD/imunify360-venv-2.5.0/opt/imunify360/venv/lib/python3.11/site-packages/playhouse/sqlite_udf.pyformat_date_time_sqliter$s J(? @ @@) _sqlite_udf control_flowdatefilehelpersmathstringc<eZdZfdZfdZfdZfdZxZS)synchronized_dictcxtt|j|i|tj|_dSr )superr__init__ threadingLock_lock)selfargskwargs __class__s rrzsynchronized_dict.__init__;s9/&&/@@@@^%% rc|j5tt||cdddS#1swxYwYdSr )rrr __getitem__r keyr#s rr%zsynchronized_dict.__getitem__? Z C C*D11==cBB C C C C C C C C C C C C C C C C C C(>AAc|j5tt|||cdddS#1swxYwYdSr )rrr __setitem__)r r'valuer#s rr+zsynchronized_dict.__setitem__Cs Z J J*D11==c5II J J J J J J J J J J J J J J J J J Js)?AAc|j5tt||cdddS#1swxYwYdSr )rrr __delitem__r&s rr.zsynchronized_dict.__delitem__Gr(r))__name__ __module__ __qualname__rr%r+r. __classcell__)r#s@rrr:s&&&&&CCCCCJJJJJCCCCCCCCCrrcfd}|S)NcD]=}t|gt||>|Sr )AGGREGATE_COLLECTION setdefaultappendklassgroupgroupss r decoratorzaggregate..decoratorQsI 6 6E + +E2 6 6 6  ' . .u 5 5 5 5 rr;r<s` r aggregater?P$ rcfd}|S)NcD]=}t|gt||>|Sr )TABLE_FUNCTION_COLLECTIONr6r7r8s rr<z!table_function..decoratorYsI ; ;E % 0 0 ; ; ; %e , 3 3E : : : : rr=r>s` rtable_functionrDXr@rcfd}|S)NcD]=}t|gt||>|Sr )UDF_COLLECTIONr6r7)fnr:r;s rr<zudf..decoratorasI - -E  % %eR 0 0 0 5 ! ( ( , , , , rr=r>s` rudfrI`s$ rct}|D]g}t|d}|D]G}t|d|j}||vr+|||||HhdS)Nr=name)setr5getgetattrr/addregister_aggregate)dbr;seenr:klassesr9rKs rregister_aggregate_groupsrTis 55D33&**5"55 3 3E5&%.99D4%%eT222  333rct}|D]Z}t|d}|D]:}|j|vr/||j||;[dSNr=)rLrCrMrKrOregister_table_function)rQr;rRr:rSr9s rregister_table_function_groupsrXss 55D22+//r:: 2 2Ez%%$$$**5111 222rct}|D]X}t|d}|D]8}|j}||vr+|||||9YdSrV)rLrGrMr/rOregister_function)rQr;rRr: functionsfunctionrKs rregister_udf_groupsr]|s 55D55"&&ub11 ! 5 5H$D4$$Xt444  555rcTt|g|Rt|g|Rt|g|RdSr )rTrXr])rQr;s rregister_groupsr_sJb*6****"2/////$V$$$$$$rcrt|gtRt|gtRt |gt RdSr )rTr5rXrCr]rG)rQs r register_allrasLb8#78888"2B(ABBBB,^,,,,,,rc|r|S|Sr r=)condtruthyfalseys r if_then_elserfs  Mrc|dd}|d}|dkr |d|S|d}|dkr |d|S|S)NT +- )replacefind)date_strtz_idx1tz_idx2s rstrip_tzrssnS))HmmC  G"}}!!mmC  G||!! Or, cd}g}|D]@\}}t||\}}|r&|dkrdpd}||d||A|sdS||S)N))i3year)i'month)i: week)Qday)ihour)<minute)secondr~sriz 0 seconds)divmodr7join)nsecondsgluepartsaccumoffsetrKvalsuffixs r human_deltars E E:: x00 X  :AX%#+F LLccc448 9 9 9 { 99U  rcv tj|}n#t$rYdSwxYw|dSNr~)ospathsplitext ValueError)filenameress rfile_extrsIgx(( tt q6Ms " 00c t|5}|cdddS#1swxYwYdS#YdSxYwr )openread)rfhs r file_readrs (^^ r7799                    s!?2 ?6?6?A cFttj||Sr )bufferzlibcompressdata compressions rgziprsdmD+66777rc*tj|Sr r decompressrs rgunziprt$$$rct|tr"t|d}t j||S)Nraw_unicode_escape) isinstancestrbytesencoderrrs rrrs@ dC  <%9::;;D}T;///rc*tj|Sr rrs rrrrrc6t|}|r|jSdSr )rnetloc)url parse_results rhostnamers(C==L#""##rcx|}t| xt|<}|Sr )lowerSTATErM)r'rets rtogglers/ ))++C 99S>>))E#J JrcR|t|S|t|<|Sr )SETTINGSrM)r'r,s rsettingrs' }||C     rc8tdSr )rclearr=rrclear_settingsrs NNrc8tdSr )rrr=rr clear_togglesrs KKMMMMMrcD|d|}}n|d}tj|||SNrr~)random randrange)startendsteps r randomrangers1 {s   E3 - --rcP tj||S#t$rYdSwxYwr )rgaussr)meansigmas rgauss_distributionrs;|D%((( tts  %%cN tj|S#t$rYdSwxYwr )rsqrtr)ns rrrs7y|| tts  $$ct t|S#t$r t|cYS#YYdSxYwwxYwr )intrfloat)rs rtonumberrsS1vv  88OOO 444 s 7-7377c8|r|sdS||SNr)count)haystackneedles r substr_countrs( 6q >>& ! !!rc,||Sr )strip)rcharss r strip_charsrs >>%  rcr|}|D]}|||Sr )update hexdigest) constructorr!hash_objargs r_hashr#sC{}}H     rc eZdZdZdZdZdS) _heap_aggc"g|_d|_dSr)heapctr s rrz_heap_agg.__init__+s rc|Sr r=r r,s rprocessz_heap_agg.process/s rc|xjdz c_tj|j||dSr)rheapqheappushrrrs rrz_heap_agg.step2s7 1  ty$,,u"5"566666rN)r/r0r1rrrr=rrrr*sA77777rrceZdZdZdS)_datetime_heap_aggc t|Sr )rrs rrz_datetime_heap_agg.process7s&u---rN)r/r0r1rr=rrrr6s#.....rr)rc<|j|jdzz|jdz zS)Nryg.A)secondsdays microsecondstds r total_secondsr;s) 5"F+- .rc*|Sr )rrs rr@sr//11rceZdZdZdS)mintdiffcdx}}|jrO||tj|j}%tj|j}||z }|||kr|}|}|jO|t|SdSr )rrheappopr)r dtpmin_diffdtdiffs rfinalizezmintdiff.finalizeDshi ;- 22Cty))B8D8d??Ci    ** * rNr/r0r1r r=rrrrBs# + + + + +rrceZdZdZdS)avgtdiffc4|jdkrdS|jdkrdSdx}}d}|jr`|dkr|tj|j})tj|j}||z }|dz }|t |z }|}|j`t ||z SNr~r)rrrrrr)r totalrrrrs rr zavgtdiff.finalizeUs 7Q;; F W\\1i zz;- 22Cty))B8D !GB ]4(( (ECi U||b  rNr r=rrr r S#!!!!!rr c eZdZdZdZdZdS)durationc"dx|_|_dSr _min_maxrs rrzduration.__init__m $$ DIIIrct|}|j ||jkr||_|j ||jkr ||_dSdSr )rrr)r r,rs rrz duration.steppsL $U + + 9 TYDI 9 TYDIII!/rc^|jr%|jr|j|jz }t|SdSr )rrr)r rs rr zduration.finalizews6 9 % %)di'B $$ $trNr/r0r1rrr r=rrrrksA%%%rrc:eZdZer dZdZdZdSdZdZdZdS)modec,t|_dSr )ritemsrs rrz mode.__init__s DJJJrc:|j|dSr )rr)r r!s rrz mode.step J  d # # # # #rc`|jr&|jdddSdSr)r most_commonrs rr z mode.finalizes6z 7z--a003A66 7 7rcg|_dSr )rrs rrz mode.__init__s DJJJrc:|j|dSr )rr7)r items rrz mode.stepr rcn|jr-tt|j|jjSdS)N)r')rmaxrLrrs rr z mode.finalizes9z B3tz?? 0@AAAA B BrN)r/r0r1rrrr r=rrrr}sB # # # $ $ $ 7 7 7 7 7    $ $ $ B B B B BrrceZdZdZdS)minrangec|jdkrdS|jdkrdSdx}}|jrO||tj|j}%tj|j}||z }|||kr|}|}|jO|Sr)rrrr)r prevrcurrrs rr zminrange.finalizes 7a<< F W\\1xi < =33D=++D$;D8d??Di rNr r=rrr)r)s#rr)ceZdZdZdS)avgrangec|jdkrdS|jdkrdSdx}}d}|jrS|dkr|tj|j})tj|j}||z }|dz }||z }|}|jSt ||z Sr)rrrrr)r rrr+r,rs rr zavgrange.finalizes 7a<< F W\\1i zz< =33D=++D$;D !GB TMEDi U||b  rNr r=rrr.r.rrr.c$eZdZdZdZdZdZdS)_rangerangec"dx|_|_dSr rrs rrz_range.__init__rrcn|j ||jkr||_|j ||jkr ||_dSdSr rrs rrz _range.stepsD 9  1 1DI 9  1 1DIII!2 1rc@|j|j|j|jz SdSr rrs rr z_range.finalizes% 9 TY%:9ty( (trN)r/r0r1rKrrr r=rrr1r1sF D%%% rr1c eZdZdZdZdZdS)stddevc"d|_g|_dSr)rvaluesrs rrzstddev.__init__s rcZ|xjdz c_|j|dSr)rr9r7)r vs rrz stddev.steps- !  1rc|jdkrdSt|j|jz tjtfd|jD|jdz z S)Nr~rc3(K|] }|z dzV dS)rNr=).0irs r z"stddev.finalize..s+BBa$h1_BBBBBBr)rsumr9rr)r rs @rr zstddev.finalizesc 6Q;;14;$&(yBBBBdkBBBBBdfqjQRRRrNrr=rrr7r7sFSSSSSrr7c.eZdZddgZdgZdZddZdZdS) RegexSearchregex search_stringmatch regex_searchNc:tj|||_dSr )refinditer_iter)r rDrEs r initializezRegexSearch.initializesUM::DJJJrcRt|jdfSr)nextrKr:)r idxs riteratezRegexSearch.iterates#$$**1--/ /rNNr/r0r1paramscolumnsrKrLrPr=rrrCrCsM?+) ; ; ; ; 0 0 0 0 0rrCc.eZdZgdZdgZdZddZdZdS) DateSeries)rstop step_secondsr date_seriesryct||_t||_t|}t j||_|jjdkr/|jjdkr|jj dkr|dkr d|_ dS|jj dkr_|jj dkrO|jj dkr?|jj dkr/|jj dkr|jj dkr|dkr d|_ dSd|_ dS) N)rrryrilr~rr)rrrWrdatetime timedeltarXr{r}rformatrvrwrz)r rrWrXs rrLzDateSeries.initializes077DJ/55DI|,,L ( 2< H H HD  1$$ !Q&& !Q&&%%( */T))*"a''*.A%%).D(()/Q&&)-1$$&&( 1 rc|j|jkrt|j}|xj|jz c_||jfSr )rrW StopIterationrXstrftimer])r rOcurrents rrPzDateSeries.iteratesJzDI%%##jG JJ$+ +JJ$$T[113 3rN)ryrRr=rrrVrVsN222( 2 2 2 2, 4 4 4 4 4rrVr )rt)rrQ)Sr[hashlibrrrrrIsysrr collectionsr ImportErrorr urllib.parseplayhouse._sqlite_extrr peeweer r playhouser cython_udf CONTROL_FLOWDATEFILEHELPERMATHSTRINGr5rCrGdictrrrr?rDrIrTrXr]r_rarfrsrrr version_inforrrrrrrrrrrrrrobjectrrrrr rrr)r.r1r7damerau_levenshtein_distlevenshtein_diststr_distmedianrCrVr=rrrxs+   #######GGG&!!!!!!!&&&%%%%%%%%&3333333MMM $#####AAA3333333JJJ       CCCCCCCC$     333222555%%% ---\ T T (T T    A!S[[888[8 S[[%%[%% S[[000[0  S[[%%[%V## # V  V V V T... .T  T  T V"" " V!! !    7 7 7 7 7 7 7 7........BQB6!!.... 21M 4+++++!+++  4!!!!!!!!!. 4v" 4BBBBB6BBB, 4y* 4!!!!!y!!!. 4V" 4 S S S S SV S S S*ss6{{:+NOO"ss6{{:#>??ss6{{:.//H YYt__Z. / /F^F 0 0 0 0 0m 0 0 0^D 4 4 4 4 4] 4 4 4 4 4sA1;;A AAAA)(A)8A??B B