j dUdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddlm Z ddlmZddlmZmZdd lmZdd lmZmZmZmZmZmZdd lmZdd lmZmZer"dd l m!Z!m"Z"m#Z#ddl m$Z%ddlm&Z&ddlm'Z'ddlm(Z(GddeZ)edZ*de+d<e,e)zZ-de+d<ej.dkZ/de+d<e/o ej.dkZ0de+d<e0o ej.dkZ1de+d<edZ2d e+d!<Gd"d#Z3e4d$Z5d%e+d&<e4d'Z6d%e+d(<e4d)Z7d%e+d*<d+d,d-d.d/d0d1d2d3Z8d4e+d5<d6d7d8d9Z9d4e+d:<e4e9d;hzZ:d%e+d<<id=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcddZ;d4e+de<dfdghgge3j<gge6gge7gdhdfdghgge3j<ggdihgge5gdhdfhgge3j<ggdjhggdkhgdhdfhgdghge3j<ggdlhggdmhgdndodfhgdghgdphge3j<gdldqhgdlhgdmhgdndogZ=dre+ds<dte=DZ=dudvdwdxdydzd{d|Z>d4e+d}<ej?d~Z@de+d<ej?dZAde+d<ej?dZBde+d<ddZCddZDGddZEGddZFGddZGddZHddZIddgZJdS)zNUtilities related to user defined functions (such as those passed to `apply`).) annotationsN) bisect_left) defaultdict)get_instructions) signature)count zip_longest)Path) TYPE_CHECKINGAnyClassVarFinalLiteral NamedTuple)LRUCache) NO_DEFAULT re_escape)CallableIteratorMutableMapping)Set) Instruction) TypeAlias) NoDefaultcFeZdZUded<ded<ded<ded<dZded <dS) StackValuestroperatorintoperator_arity left_operand right_operandN str | None from_module)__name__ __module__ __qualname____annotations__r$W/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_utils/udfs.pyrr&sPMMM"K""""""r*r)exprframeseriesr MapTarget StackEntry) r _MIN_PY311)r1 _MIN_PY312)r1 _MIN_PY314 z@MutableMapping[tuple[Callable[[Any], Any], str], BytecodeParser]_BYTECODE_PARSER_CACHE_c eZdZUddddddddd d d d d Zded<eerdhnddhZer esdddddndddddZ ded<edZ eddhZ e e zZ dddZ ded<ddiZded <d dd!d"Zded#<ehd$eezee zeeze zZee ezeeze zezZeeZd%S)&OpNames+&z//<<>>%*|z**-/^) BINARY_ADD BINARY_ANDBINARY_FLOOR_DIVIDE BINARY_LSHIFT BINARY_RSHIFT BINARY_MODULOBINARY_MULTIPLY BINARY_OR BINARY_POWERBINARY_SUBTRACTBINARY_TRUE_DIVIDE BINARY_XORzClassVar[dict[str, str]]BINARYCALL CALL_FUNCTION CALL_METHOD)POP_JUMP_FORWARD_IF_FALSEPOP_JUMP_FORWARD_IF_TRUEJUMP_IF_FALSE_OR_POPJUMP_IF_TRUE_OR_POP)POP_JUMP_IF_FALSEPOP_JUMP_IF_TRUErXrY CONTROL_FLOW) LOAD_CONST LOAD_DEREF LOAD_FAST LOAD_GLOBAL LOAD_METHOD LOAD_ATTRr_r])LOAD_FAST_BORROWLOAD_SMALL_INTSIMPLIFY_SPECIALIZEDPOLARS_EXPRESSIONzClassVar[dict[str, int]] SYNTHETIC~)UNARY_NEGATIVEUNARY_POSITIVE UNARY_NOTUNARY>IS_OP BINARY_OP COMPARE_OP CONTAINS_OP BINARY_SUBSCRN)r%r&r'rRr( frozensetr3rSr5r\ LOAD_VALUESrbLOADrerhrmset PARSEABLE_OPS MATCHABLE_OPSvalues UNARY_VALUESr)r*r+r;r;:s#! ( (F     9QfXX/=1Q R RD  ) ),(+$'#&    "% #$'#&    )TUUK =+677I  "D'&66 Q+I''E ILLL #e**  #l    #i..    M  !!M1CCKK?)KdR9U\\^^,,LLLr*r;)acosacoshasinasinhatanatanhcbrtceilcoscoshdegreesexpfloorloglog10log1ppowradianssinsinhsqrttantanhzFinal[frozenset[str]]_MATH_FUNCTIONS)npnumpy_NUMPY_MODULE_ALIASES)arccosarccosharcsinarcsinharctanarctanhrrrrrrrrrrrsignrrrrr_NUMPY_FUNCTIONSz dt.date()zdt.day()z dt.hour()zdt.microsecond()z dt.minute()z dt.month()z dt.second()z dt.year())datedayhour microsecondminutemonthsecondyearzFinal[dict[str, str]]_PYTHON_ATTRS_MAPFloat64Int64String)floatrr_PYTHON_CASTS_MAPabs_PYTHON_BUILTINSendswith str.ends_withlowerzstr.to_lowercaselstripzstr.strip_chars_start removeprefixzstr.strip_prefix removesuffixzstr.strip_suffixreplace str.replacerstripzstr.strip_chars_end startswithstr.starts_withstripzstr.strip_charstitlezstr.to_titlecaseupperzstr.to_uppercasezfillz str.zfillrzdt.daterzdt.dayrzdt.hour isoweekdayz dt.weekdayrzdt.microsecondzdt.monthz dt.secondz dt.strftimezdt.timezdt.year)rrstrftimetimer_PYTHON_METHODS_MAPr_r])argument_1_opnameargument_2_opname module_opnameattribute_opname module_nameattribute_name function_namemathjsonloadsdatetimestrptimeF)rrrrrrrcheck_load_globalrbdtz'list[dict[str, list[AbstractSet[str]]]]_MODULE_FUNCTIONSc\g|])}ttjggfD] }i|d|i *S)argument_1_unary_opname)rvr;rm).0kindunarys r+ rsa w}%%&+ /t. &..r*rrrrrrstr.json_decode)z math.acosz math.acoshz math.asinz math.asinhz math.atanz math.atanhz json.loads_MODULE_FUNC_TO_EXPR_NAMEz,pl\.col\("([^"]*)"\) & pl\.col\("\1"\)\.(.+)_RE_IMPLICIT_BOOLz^(s|srs\d?|series)\._RE_SERIES_NAMESz^bool\((.+)\)$_RE_STRIP_BOOLreturndict[str, Any]c8ttjj}tj}d} |rFtj|}|t|r |j}|dz }nn|F|i}ni|j |j }~n#~wxYw|S)z7Get all local and global variables from caller's frame.rrg) r __file__parentinspect currentframegetfilerrf_backf_locals f_globals)pkg_dirr-nfname variabless r+_get_all_caller_variablesr,s8nn#*G  " "E A OE**EG --  Q   =II=5>=U_=I EE s ABBcolr expression map_targetcd|d}|dkr|S|dkrt|r|dddS||d}d D]}t jd |d |s|cS t } d t|}t jd |d |s|S.d |}t|)z@The name of the object against which the 'map' is being invoked.pl.col("")r,r..rgr)ssrsr.z\bTrzTODO: map_target = ) rmatchsplitrresearchrnextNotImplementedError)rrrcol_expr search_exprnamermsgs r+_get_target_namerIs!#!!!HV x    ! !* - - /##C++A. .!((266 *  D9^4^^^[99    GG "a??D9^4^^^[99    / . .C c " ""r*ceZdZUdZdZded<dZded<dZded<eZ d ed <d*dZ d+dZ e d,dZ d-dZed.dZd/dZd0dZed1d Zed2d!Zed3d"Zed2d#Zd4d%Zddd&d5d)ZdS)6BytecodeParserzMIntrospect UDF bytecode and determine if we can rewrite as native expression.Nr#_map_target_namez bool | None_can_attempt_rewritedict[str, Any] | None_caller_variablesz"tuple[str, str] | NoDefault | None_col_expressionfunctionCallable[[Any], Any]rr/rNonec t|}n#t$rtg}YnwxYw||_||_|||_t||j||_ dS)aI Initialize BytecodeParser instance and prepare to introspect UDFs. Parameters ---------- function : callable The function/lambda to disassemble and introspect. map_target : {'expr','series','frame'} The underlying target object type of the map operation. ) instructionscaller_variablesrN) r TypeErroriter _function _map_target_get_param_name _param_nameRewrittenInstructionsr_rewritten_instructions)selfrroriginal_instructionss r+__init__zBytecodeParser.__init__js -$4X$>$> ! ! - - -%)HH ! ! ! - "%//99'<.!3( ( ( $$$s ..r,rct|r6td|}t|6|S)zIDrop extraneous/implied bool (eg: `pl.col("d") & pl.col("d").dt.date()`).zpl.col("\1").\2)replstring)rrsub)rr,s r+_omit_implicit_boolz"BytecodeParser._omit_implicit_boolsP&&t,, O$((.@(NND &&t,, O r*c t|}n#t$rYdSwxYwt|jx}dkr.t t |ndS)z&Return single function parameter name.Nrg)r ValueErrorlen parametersrrkeys)rsigr s r+rzBytecodeParser._get_param_names~ H%%CC   44 /:00A55 joo''(( ) ) ) s   expression_blocksdict[int, str]logical_instructionslist[Instruction]list[tuple[int, str]]c|rtd|Ddkrt|}d}|D]v}|t||jdz }|t||jdz }|dkr ||dks&|||fvr d||z||<||xxdz cc<|}w|D]!}t j|j||j<"t| S)zBInject nesting boundaries into expression blocks (as parentheses).ch|] }|j Sr)opnamerinsts r+ z1BytecodeParser._inject_nesting..sAAADDKAAAr*rgr()) rlistr!roffsetargvalr;r\r+sorteditems)rr#r% block_offsetsprev_endr-startends r+_inject_nestingzBytecodeParser._inject_nestings=  SAA,@AAABBQFF+/0A0F0F0H0H+I+I 0++D)+mT[*Q*QTU*UVE' M4;(O(ORS(STC!QJJ3-2C+C+C#E3<777:=Nu=U7U-e4-c222c9222'*H, S S181Edk1R!$+..'--//000r*c|jS)z8The map target, eg: one of 'expr', 'frame', or 'series'.)rrs r+rzBytecodeParser.map_target r*boolc|jf|jduoWt|jdko?t d|jDo!t d|jDdk|_|jS)aN Determine if we may be able to offer a native polars expression instead. Note that `lambda x: x` is inefficient, but we ignore it because it is not guaranteed that using the equivalent bare constant value will return the same output. (Hopefully nobody is writing lambdas like that anyway...) Nc3<K|]}|jtjvVdSN)r+r;rwr,s r+ z5BytecodeParser.can_attempt_rewrite..s@K7#88r*c30K|]}|jdk dVdS) RETURN_VALUErgNr*r,s r+rDz5BytecodeParser.can_attempt_rewrite..s;{n444444r*rg)rrrrallsumrr=s r+can_attempt_rewritez"BytecodeParser.can_attempt_rewrites  $ , , 455:  $ <  $ :    % ((r*c8tj|jdS)z%Print disassembled function bytecode.N)disrr=s r+rKzBytecodeParser.diss r*c|jS)zThe function being parsed.)rr=s r+rzBytecodeParser.functions ~r*c4t|jjS)zDThe original bytecode instructions from the function we are parsing.)r2r_original_instructionsr=s r+rz$BytecodeParser.original_instructionssD0GHHHr*c|jS)z0The parameter name of the function being parsed.)rr=s r+ param_namezBytecodeParser.param_namer>r*c*t|jS)zEThe rewritten bytecode instructions from the function we are parsing.)r2rr=s r+rewritten_instructionsz%BytecodeParser.rewritten_instructionssD0111r*rc jturZjSj\}}|krAtjdt |ddt d|}|f_|Sd_j d_dStt}g d}tj D]^\}}|j tj vr+j |dzj} |C|||_  fd|D }n#t$$r d_YdSwxYwd d |D} d | vr d_dS| } jd kr9jx} t-| j} | dd| } | f_| S) zDTranslate postfix bytecode instructions to polars expression/string.Nz pl\.col\("z"\)rrrrgc i|]]\}}|t|jjjjt t^S))r r rr)rrPdepth)InstructionTranslatorrrr to_expressionrrr?)rr3opsrr%rs r+ z0BytecodeParser.to_expression.. s   $1%()-)?#'#3!%  $m#'#3!$';"<"<==$   r* c3 K|] \}}|V dSrCr))r_offsetr,s r+rDz/BytecodeParser.to_expression..s&LL tLLLLLLr*pl.col(r.)rrrrrrrrr2 enumeraterr+r;r\r3appendr;r6rjoinrrrr) rrcol_namer,control_flow_blocks jump_offsetidxr-expression_strings polars_expr target_namer%s `` @r+rWzBytecodeParser.to_expressionsp  z 1 1d6J6V!1NHdhv:)H"5"5:::1y~~111 ),T{$K $   ##'D 4*$//! "4#?@@ > >IC{g222":37CJ $++D1111#K077==== !%!5!5      (;'@'@'B'B   %""   #   #'D 44 hhLL9KLLLLL  K ' '#'D 422;??K8++#'#88KA"23 TEU"V"VK)112DS2D2D2DkRR $'#5D  s6EEEsuggestion_override udf_overriderirjc ddlm}m}ddlm}|p||}||jx}t|||j}|p |j j pd} | dkr d|j d} d |vrd nd } |jd krd nd\} } |rd|d| dd|d| fnd|d| dd|d| f\} }tj d| d| d| d|||dSdS)zFGenerate warning that suggests an equivalent native polars expression.r)find_stacklevel in_terminal_that_supports_colour)PolarsInefficientMapWarningNz...zzlambda z: ...z pl.col("")zHNote: in list.eval context, pl.col("") should be written as pl.element()rr,) expressionsExpr)r.Seriesz - z.map_elements(z) z + z z - z) z +  z6.map_elements is significantly slower than the native zc API. Only use if you absolutely CANNOT implement your logic otherwise. Replace this expression... zwith this one instead: ) stacklevel)polars._utils.variousrlrmpolars.exceptionsrnrWrrrrr%rwarningswarn)rrrirjrlrmrnsuggested_expressionrg func_nameaddendumapitypeclsnamebeforeafters r+rwzBytecodeParser.warn/s         BAAAAA2Md6H6H6M6M  +#44 =.-t/? %H(?H5IJ&&=d&6===  #777[Z #v--(') GW4355 S;SSiSSSL#7LL(LL E;DDiDDD=/==8== FE MWT[   ,*?,,      = , +r*)rrrr/rr )r,rrr)rrrr#)r#r$r%r&rr')rr/)rr?)rr )rr)rr&)rr#)rrrr#)rrrir#rjr#rr )r%r&r'__doc__rr(rrrrrr staticmethodrr;propertyrrIrKrrrPrRrWrwr)r*r+rrbsWW#'''''(,,,,,/33333:DODDDD    6     \  11114   X ))))6    XIIIXI   X 222X2BBBBP+/#' 99999999r*rcJeZdZdZdd ZddZeddZddZddZ dS) rVz>Translates Instruction bytecode to a polars expression string.r r&r rrrrr/rr cX||||_||_||_dSrC)_to_intermediate_stack_stackrr)rr r rrs r+rzInstructionTranslator.__init__ns.11, KK !1!r*rrrPrUrc<||j|||S)z7Convert intermediate stack to polars expression string.)_exprr)rrrPrUs r+rWz#InstructionTranslator.to_expressionyszz$+sJ>>>r*r-rc.|jx}tjvrtj|S|jr|jS|dkr |jrdndS|dkr |jrdndS|tjvrtj|S|dkrdSd |d |}t |) z@Convert bytecode instruction to suitable intermediate op string.rnis notisrqnot ininrrreplace_strictz$unexpected or unrecognised op name (z) Please report a bug to https://github.com/pola-rs/polars/issues with the content of function you were passing to the `map` expression and the following instruction object: )r+r;r\argreprr4rmAssertionError)r-r+rs r+opzInstructionTranslator.op}sk !Fg&: : :'/ / \ &<  w  #{488 4 } $ $#{488 4 w} $ $=( (  & &##NvNNFJNN  !%% %r*valuer0ct|trGtd|j}||j|||dz}|jdkr|tj vr| dr"| drdnd}|d||S|dtj vr@|dd  dr#| drdnd}d |d ||S|j tvr |tvrd }n5|j d kr(td||t"vrd}nd}||d |dS||S||j|||dz} |dvr|jdkr|dkrdnd} |d| dS|dvr#|dkrdnd} d|vr | d |d| dn | |d| dS|dkrd|jst)|_t|j|d t*sd} t-| | d|d |dS|dkr d |d| dS|d kr d |d!| dS|d|d| } |rd | dn| S||krd"|d#S|S)$z?Take stack entry value and convert to polars expression string.z\1rgr]r1r()rrNr0z).znp.rmath.)rrr rnot_z.is_znull())rrrrirZz.is_in(z))rzrequire dict mappingr>z * 2**z).cast(pl.Int64)r?z / 2**rr) isinstancerrrrrr!r r;rzrrr$rrrgetrr"rrdictr) rrrrPrUre1callpfxe2rrr,s r+rzInstructionTranslator._exprs| eZ ( (= &##E5>::BE.ZKKB#q((W111}}Y//2%'[[%5%5?rr4"$11r14111!u 444ABB9J9J99U9U4%'[[%5%5?rr432333T333 )-BBB"222#)V33599,",,KK*++& !-2------{b{{"ZZ 3S*eaiPP)))e.@F.J.J!#t22D 22d2222+++!#t22D"99 33"33R3333 $6b66666 +++1M1J1L1L.%d&<&@&@T&J&JDQQ741#666 --2------4ZZ>r======4ZZ=r====== ,,2,,,,D*/9;t;;;;T9 j %c%%% % r*cp|dvrg}|D]}||jtjvr|jn|jtjvs(tj|jdkrHt| |d| dt|ddn[t| |d| d| dt|dd|d Sd |d }t|) zITake postfix bytecode and convert to an intermediate natural-order stack.)r,r.rgN _from_module)rr r!r"r$rAr/rzTODO: z map target not yet supported.) r_r+r;rurrmrhrrrpopgetattrr)rr rstackr-rs r+rz,InstructionTranslator._to_intermediate_stacksE + + +&(E$   {gl22LL!K7=88&044T[AAQFF#%)WWT]]+,).*.(/nd(K(K (%)WWT]]+,).2*/))B--(/nd(K(K 08ODzCCC!#&&&r*N) r r&r rrrrr/rr )rrrPrrUrrr)r-rrr) rr0rrrPrrUrrr)r r&rr/rr0) r%r&r'rrrWrrrrr)r*r+rVrVksHH " " " "????&&&\&.AAAAF#'#'#'#'#'#'r*rVceZdZdZegdZd-d Zd.d Zd/dZd0dZ ddd1dZ d2dZ d3d Z d3d!Z d3d"Zd3d#Zed4d$Zed5d&Zd6d+Zd,S)7ra] Standalone class that applies Instruction rewrite/filtering rules. This significantly simplifies subsequent parsing by injecting synthetic POLARS_EXPRESSION ops into the Instruction stream for easy identification/translation, and separates the parsing logic from the identification of expression translation opportunities. ) COPYCOPY_FREE_VARS NOT_TAKENPOP_TOPPRECALL PUSH_NULLRESUMErFTO_BOOLr Iterator[Instruction]rrr rrr cf||_||_t||_g}||jD]W}|j|jvrG|jtjvr g|_ dS| |}| |X| ||_ dSrC) rrr2rN_unpack_superinstructionsr+ _ignored_opsr;rxr_update_instructionr__rewrite)rr rr normalised_instructionsr- upgraded_insts r+rzRewrittenInstructions.__init__s "!1&*<&8&8#"$2243NOO > >D{$"333;g&;;;35D0FF $ 8 8 > > '..}==='+}}5L'M'M$$$r*rc*t|jSrC)rrr=s r+__len__zRewrittenInstructions.__len__,s4/000r*c*t|jSrC)rrr=s r+__iter__zRewrittenInstructions.__iter__/sD0111r*itemr rc|j|SrC)r)rrs r+ __getitem__z!RewrittenInstructions.__getitem__2s+D11r*F)is_attrrdopnameslist[AbstractSet[str]]argvals5list[AbstractSet[Any] | dict[Any, Any] | None] | Nonerr?r&c0t||pg}}||z}|r/|j||dzx}r|djtjvrgS|j||}t||kr*t dt |||Dr|SgS)a Check if a sequence of Instructions matches the specified ops/argvals. Parameters ---------- idx The index of the first instruction to check. opnames The full opname sequence that defines a match. argvals Associated argvals that must also match (in same position as opnames). is_attr Indicate if the match represents pure attribute access (cannot be called). rgrc3JK|]\}}}|j|vo |dup|j|vVdSrC)r+r4)rr- match_opnames match_argvals r+rDz1RewrittenInstructions._matches..Us_7 7 2m\ K= ( F%D )D7 7 7 7 7 7 r*)r _instructionsr+r;rSrGr ) rrdrrrn_required_ops idx_offset trailing_instr s r+_matcheszRewrittenInstructions._matches5s,#&g,, 2>)  "&"4Z*q.5P"QQ a '7<77I)#j.9 |   . .37 7 6Agw667 7 7 4 4 .  r*cj||_gdt|jkr|jdc}|jtjvs5t fd|j|j|j|j fDs |pdz t|jkS)a# Apply rewrite rules, potentially injecting synthetic operations. Rules operate on the instruction stream and can examine/modify it as needed, pushing updates into "updated_instructions" and returning True/False to indicate if any changes were made. rrgc34K|]}|xVdSrCr))r map_rewriterd incrementupdated_instructionss r+rDz1RewrittenInstructions._rewrite..lsI : :*k#/CDDD : : : : : :r*) rrr+r;ruany_rewrite_functions_rewrite_methods_rewrite_builtins_rewrite_attrsr_)rr r-rdrrs @@@r+rzRewrittenInstructions._rewrite_s*24C*++++"05qOD){',..c : : : : : :+)*' $ : : : 7 7.%++D111 9> !CC*++++$#r*rc ||dhdhgdtgdx}rO|d}t|j}|d||}||d |gt |S) zDReplace python attribute lookup with synthetic POLARS_EXPRESSION op.r_rbNT)rrrrgrfr+r4rr)rrr4_replaceextendr)rrdrmatching_instructionsr- expr_namepxs r+rz$RewrittenInstructions._rewrite_attrszs$(MM !][M2,- %2% %  H )+D)$+6I*9iB ! ' ')>q)A2(F G G G()))r*c||dhddhtjgtgx}r|dd\}}|jx}t vrt |}d|d}|d |||j }||j } || |ft|S) zEReplace builtin function calls with a synthetic POLARS_EXPRESSION op.r`r_r]rrNrAcast(pl.r1rfr+r4rr3r3) rr;rSrr4rrr3rr) rrdrrinst1inst2r4dtyperoperands r+rz'RewrittenInstructions._rewrite_builtinss%)MM #_{L&A7<P%&%2% %   7 1!4LE5,&+<<<)&1,E,,,*|  BnnELn99G ' '" 6 6 6()))r*c pdD]}tD]}|r|ddsdS|r)ddhg|d|d|d tjn8ddhg|d |d |d|d|d tj}|d }||||r g|d ng|d |d|d x}rt |d}||d|z\} } } |r||jst|_| jx} |jvr|j| j tfd|DsMtd| | } n| jdkrd} n| jdkrQ||dzj} d| d} | | j|dj|st |ccSn;| jdkr)td| j| j} n| j} |r| n| }| d| | |j}|rdn| jpd|_| | j}||dr||d|z|fn||ft |ccSdS)z=Replace function calls with a synthetic POLARS_EXPRESSION op.)FTrTrr`r^rrrrrrrrrr1c3 K|]}|vV dSrCr))rmrs r+rDz;RewrittenInstructions._rewrite_functions..s(&R&Ra q(8&R&R&R&R&R&Rr*rrrrzstr.to_datetime(format="rrrrfrNr)rrr;rSrrrrr4r&rr_is_stdlib_datetimerr3rr)rrdr check_globals function_kindrmodule_aliasesrattribute_countrrinst3rfmt swap_instrrrs @r+rz(RewrittenInstructions._rewrite_functionss&+] 6] 6M!2\ 6\ 6  ):):;NPT)U)U111%& 5&':;''@A'':;   ' 5&7''9:'':;  ''@A  '':;  &"/}!=,0MM#%&7&}5&'78'7-: - - (D6'*-8H*I&J&JO*?'!o*==+'E5%%1#5Q5N5P5PD2).5Id>TTT$*.*@*K*VK#&&R&R&R&R>&R&R&R#R#R) ((A(E(E#. < < < $''<#=#====== >//$=$A$A2EL22EL%% %*L *7 AEI2( )(/ (B /<&Wdd%,BVRVBO'00 0EEG(// ))BC +#1!o2EF &r]45555555ID61\ 6|qr*c tr tjndh}|||dhtjgt gp)|||tjgt gx}r|d}t |j}|djdkrr|dj}t|trG|dvrCd|vrdnd \}} d d |D} d |vrd nd } d| |d| d| | d }n |d|dz }| d||} | | n/|||dhdhdhtjgt gp-|||dhdhtjgt gx}r|d}t |j}d|dt|dz D} |dkr8t| dkr| \}}}|d|d|d|dz }n5| \}}d|d|d}n&|dd d| Ddz }| d||} | | t|S)z@Replace python method calls with synthetic POLARS_EXPRESSION op.rar]rrrg)rrstarts)rEr)r$rBc34K|]}t|VdSrC)rrvs r+rDz9RewrittenInstructions._rewrite_methods..(s(!D!D1)A,,!D!D!D!D!D!Dr*'"zstr.contains(rr0r1rfrcg|] }|j Sr))r4)ris r+rz:RewrittenInstructions._rewrite_methods..Is*r*rr1,z,n=z,literal=True)zstr.replace_all(c34K|]}t|VdSrC)reprrs r+rDz9RewrittenInstructions._rewrite_methods..Us($C$CT!WW$C$C$C$C$C$Cr*)r5r;rbrrSrr4r+rtupler`rr_r)rrdrrarr-r, param_valuerendsrxqr param_valuesoldnewrs r+rz&RewrittenInstructions._rewrite_methods sT,6Jg''M?  MM$|nglC,-     MM$gl3,-    I ,)+D&t{3D$Q'.,>>3A6= k511 1d?7719D0@0@99iLFD!D!D !D!D!DDDB"k11sAGAGvGGGGTG1GGGDD0 0000D&9$PTUUB ' ' + + + + MM!N!N!NL --     MM$|n|nglS,-    "' ,*)+D&t{3D.q37L3M3MPQ3Q/QRL}$$|$$))&2OCeGGGGG%GGGGDD+HCKcKKcKKKDDFCHH$C$Cl$C$C$CCCFFFF&9$PTUUB ' ' + + +()))r*c#K|D]E}|jdvr6dD]2}|d|j||j|V3A|VFdS)z%@%@D "% q QTXXm%<%<@Q%QJ"Htxx '<'<'H Jr*N)r rrrr rrr )rr)rr)rr rr) rdrrrrrrr?rr&)r r&rr&)rdrrr&rr)r r&rr)r-rrr)rrrrrrrr?)r%r&r'rrsrrrrrrrrrrrrrrrr)r*r+rrs9   LNNNN,111122222222 ((((((T$$$$6****"****4ccccJP*P*P*P*d\$\.JJJJJJr*rrrtuple[str, str]cv |jj}|j}n#t$rYdSwxYw|dkr|tvrd|dfS|dkrv|t vrddt |dfS|t vrr'sTTT""""""  ###### (((((((()(((((777777770BBBBBBBBBB222222      //////########89 9999j( (((($/ ////>3#3w#> >>>>>3#3w#> >>>> HRLL 9-9-9-9-9-9-9-9-z*3**:09y/I/IIIII*3)++>   %   , ,      ,, +4)4E*F*F%*PPPPP.. . % . & . & .}. #.#. . . . [. I.  8!." I#.$,%.&#'.(  1...> +L9:!+,-.*+ +L9:!+,z)* +m_!+,z") +m_+n-!+,# ~%,"   +m_+n-&-$./#T*+&<.%,"  [7>7777t!#44&2:&UVVVVVV$"*%<======" #4555555:####2FFFFFFFFRP'P'P'P'P'P'P'P'fQJQJQJQJQJQJQJQJh B,,,,^ 6 7r*