2i (HddlZGddZGddZGddZy)Nc eZdZdZdedefdZy) ProfileStatsz ProfileStats Class for representing runtime execution statistics of an operation. Attributes: records_produced (int): The number of records produced. execution_time (float): The execution time in milliseconds. records_producedexecution_timec ||_||_y)a Initializes a new ProfileStats instance with the given records_produced and execution_time. Args: records_produced (int): The number of records produced. execution_time (float): The execution time in milliseconds. N)rr)selfrrs a/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/falkordb/execution_plan.py__init__zProfileStats.__init__ s!/ 0N)__name__ __module__ __qualname____doc__intfloatr r r rrs 1 1e 1r rc|eZdZdZddedefdZedefdZ edefdZ d Z defd Z d e defd Zdefd Zy) Operationa? Operation Class for representing a single operation within an execution plan. Attributes: name (str): The name of the operation. args (str): Operation arguments. children (list): List of child operations. profile_stats (ProfileStats): Profile statistics for the operation. Nname profile_statsc<||_||_g|_||_y)a Creates a new Operation instance. Args: name (str): The name of the operation. args (str, optional): Operation arguments. profile_stats (ProfileStats, optional): Profile statistics for the operation. N)rargschildrenr)rrrrs r r zOperation.__init__$s!" !  *r returnc.|jjS)z: returns operation's execution time in ms )rrrs r rzOperation.execution_time2s !!000r c.|jjS)zB returns number of records produced by operation. )rrrs r rzOperation.records_produced9s !!222r crt|ts td|jj ||S)z Appends a child operation to the current operation. Args: child (Operation): The child operation to append. Returns: Operation: The updated operation instance. zchild must be Operation) isinstancer Exceptionrappend)rchilds r append_childzOperation.append_child@s1%+56 6 U# r c,t|jS)zx Returns the number of child operations. Returns: int: Number of child operations. )lenrrs r child_countzOperation.child_countPs4==!!r oct|tsy|j|jk(xr|j|jk(S)a  Compares two Operation instances for equality based on their name and arguments. Args: o (object): Another Operation instance for comparison. Returns: bool: True if the operations are equal, False otherwise. F)rrrr)rr's r __eq__zOperation.__eq__Ys6!Y'yyAFF":tyyAFF'::r c\|jdnd|jz}|j|S)z Returns a string representation of the operation. Returns: str: String representation of the operation. z | )rr)rargs_strs r __str__zOperation.__str__hs0*2 0A))XJ''r )NN)r r rrstrboolr propertyrrrr#r&objectr)r-rr r rrs +S +D +111 3#33  "S" ; ;4 ;((r rcVeZdZdZdZdZdefdZdefdZ de defdZ d Z d Z d Zy ) ExecutionPlanz ExecutionPlan Class for representing a collection of operations. Attributes: plan (list): List of strings representing the collection of operations. structured_plan (Operation): Root of the structured operation tree. cFt|ts tdt|dtr|Dcgc]}|j }}||_i|_|j|_|j D]}|j |j!ycc}w)z Creates a new ExecutionPlan instance. Args: plan (list): List of strings representing the collection of operations. zplan must be an arrayrN) rlistr bytesdecodeplan operations_operation_treestructured_planreverse)rr8bkeys r r zExecutionPlan.__init__|s$%34 4 d1gu %(,-1AHHJ-D- #335?? +C OOC ( ( * + .sBc@||jvr|j|SgS)z Collects all operations with specified name from plan Args: op_name (string): Name of operation to collect Returns: List[Operation]: All operations with the specified name )r9rr!)rop_nameopsops r collect_operationsz ExecutionPlan.collect_operationss$ doo %??7+ + r rc||k7ry|j|jk7ryt|jD]0}|j|j||j|r0yy)z Compares execution plan operation trees. Returns: bool: True if operation trees are equal, False otherwise. FT)r&range"_ExecutionPlan__compare_operationsr)rroot_aroot_bis r __compare_operationsz"ExecutionPlan.__compare_operationssx V     6#5#5#7 7v))+, A,,V__Q-?QRAST r cRd}d}|j|jt||S)z Returns a string representation of the execution plan. Returns: str: String representation of the execution plan. c dj|Dcgc]}|jD]}d|z c}}Scc}}w)N  )join splitlines) str_children str_childlines r aggregate_strz,ExecutionPlan.__str__..aggregate_strsR99&2! ) 4 4 6TM! s!: c|d|S)NrMr)xys r combine_strz*ExecutionPlan.__str__..combine_strsS1#; r )_operation_traverser;r.)rrTrXs r r-zExecutionPlan.__str__s0  ''  #}k  r r'cxt|tsy|j}|j}|j||S)z Compares two execution plans. Returns: bool: True if the two plans are equal, False otherwise. F)rr3r;rF)rr'rGrHs r r)zExecutionPlan.__eq__s>!]+%%""((88r c,t|jS)N)iterr9rs r __iter__zExecutionPlan.__iter__sDOO$$r c ||}t|jdk(r|S|jDcgc]}|j||||}}||||Scc}w)a{ Traverses the operation tree recursively applying functions. Args: op: Operation to traverse. op_f: Function applied for each operation. aggregate_f: Aggregation function applied for all children of a single operation. combine_f: Combine function applied for the operation result and the children result. r)r%rrY)rrBop_f aggregate_f combine_fop_resr"rs r rYz!ExecutionPlan._operation_traversespb r{{ q M    $ $UD+y I  X!677  sAcd}d}g}d}d}|t|jkr|j|}|jd}||k(r||jd}|j|j vrg|j |j<|j |jj ||r!|j}|j||}|dz }|j |n||dzk(r||jd}|j|j vrg|j |j<|j |jj ||j||j ||}|dz }|dz }n>||kr.||z dz} t| D]} |j}|| z}n td|t|jkr|dS)z Builds the operation tree from the string representation. Returns: Operation: Root of the structured operation tree. rNcd}|dj}|jdt|dkDrd|dvrtt j d|dj d}tt j d|dj d}t||}|jdt|t|dk(rd|S|dj|S)NrzRecords producedzRecords produced: (\d+)zExecution time: (\d+.\d+) ms) strippopr%rresearchgrouprrr)rrrrrs r create_operationz7ExecutionPlan._operation_tree..create_operation s M7==?D HHQK4y1}!3tBx!?#&II8$r(CII!L$ "'II>RIOOPQR"!--=~ N  c$i1nd= 26q'--/= r rN|rfzcorrupted plan) r%r8countsplitrr9r!rhr#rEr ) rrIlevelstackcurrentrl current_opop_levelr" levels_back_s r r:zExecutionPlan._operation_trees $#dii. 1J!''/H5 ))9)9#)>?::T__424DOOEJJ/ +2259#iikG((/Q U#UQY&))9)9#)>?::T__424DOOEJJ/ +2259$$U+ W% QE!$h.2 {+*A#iikG*$ 011M#dii. NQxr N)r r rrr rCr/rFr.r-r1r)r]rYr:rr r r3r3ssL+&,d,  .9949$%8.Fr r3)rirrr3rr r rws+ 11*W(W(tQQr