mBNj(XddlZddlmZmZGddZGddZGddZy) N)ListOptionalc 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 h/Users/ahmed/devFolder/Ultron/claude-voice/.venv/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 rceZdZdZ ddedeedeefdZede fdZ ede fd Z d Z de fd Zd edefd ZdefdZy) 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. Nnameargs 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)rrchildrenr)r rrrs r r zOperation.__init__&s!  )+ *r returncJ|jJ|jjS)z: returns operation's execution time in ms )rrr s r rzOperation.execution_time:s( !!---!!000r cJ|jJ|jjS)zB returns number of records produced by operation. )rrrs r rzOperation.records_producedBs( !!---!!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)r childs r append_childzOperation.append_childJs1%+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_countZs4==!!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)r!rrr)r r)s r __eq__zOperation.__eq__cs6!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)r args_strs r __str__zOperation.__str__rs0*2 0A))XJ''r )NN)rrrrstrrrr propertyrrrrr%r(objectboolr+r/rr r rrs#04 ++sm+ - +(1113#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) r!listr"bytesdecodeplan operations_operation_treestructured_planreverse)r r:bkeys r r zExecutionPlan.__init__s$%34 4 d1gu %(,-1AHHJD- #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 )r;rr#)r op_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)r root_aroot_bis r __compare_operationsz"ExecutionPlan.__compare_operationsss 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_strsO99&2%1 ) 4 4 6TM 6"%1 s!: c|d|S)NrOr)xys r combine_strz*ExecutionPlan.__str__..combine_strsS1#; r )_operation_traverser=r0)r rVrZs 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)r!r5r=rH)r r)rIrJs r r+zExecutionPlan.__eq__s>!]+%%""((88r c,t|jS)N)iterr;rs 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'rr[)r rDop_f aggregate_f combine_fop_resr$rs r r[z!ExecutionPlan._operation_traversessb 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_operations M7==?D HHQK4y1}!3tBx!?#&II8$r(CII!L$ "'II>RIOOPQR"!--=~ N  c$i1nd= 26q'--/= r rP|rhzcorrupted plan) r'r:countsplitrr;r#rjr%rGr") r rKlevelstackcurrentrn current_opop_levelr$ levels_back_s r r<zExecutionPlan._operation_tree s  $#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)rrrrr rEr3rHr0r/r2r+r_r[r<rr r r5r5}sL+&,d,  09949$%82Fr r5)rktypingrrrrr5rr r rzs. !11,_(_(DTTr