File: //lib/python3.9/site-packages/redis/commands/graph/__pycache__/execution_plan.cpython-39.opt-1.pyc
a
��bV � @ s6 d dl Z G dd� d�ZG dd� d�ZG dd� d�ZdS )� Nc @ s e Zd ZdZdd� ZdS )� ProfileStatszB
ProfileStats, runtime execution statistics of operation.
c C s || _ || _d S �N)�records_produced�execution_time)�selfr r � r �G/usr/lib/python3.9/site-packages/redis/commands/graph/execution_plan.py�__init__ s zProfileStats.__init__N)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ sH e Zd ZdZddd�Zdd� Zdd� Zeed �d
d �Z e
d �d
d�Z dS )� Operationz<
Operation, single operation within execution plan.
Nc C s || _ || _|| _g | _dS )z�
Create a new operation.
Args:
name: string that represents the name of the operation
args: operation arguments
profile_stats: profile statistics
N)�name�args�
profile_stats�children)r r r r r r r r s zOperation.__init__c C s* t |t�r| |u rtd��| j�|� | S )Nzchild must be Operation)�
isinstancer � Exceptionr �append)r �childr r r � append_child! s zOperation.append_childc C s
t | j�S r ��lenr )r r r r � child_count( s zOperation.child_count��o�returnc C s& t |t�sdS | j|jko$| j|jkS )NF)r r r r )r r r r r �__eq__+ s
zOperation.__eq__�r c C s& | j d u rdnd| j }| j� |� �S )N� z | )r r )r �args_strr r r �__str__1 s zOperation.__str__)NN) r
r r r
r r r �object�boolr �strr" r r r r r s
r c @ sN e Zd ZdZdd� Zdd� Zed�dd�Zee d �d
d �Z
d d
� Z dd� Z dS )�
ExecutionPlanz2
ExecutionPlan, collection of operations.
c C sB t |t�std��t |d t�r.dd� |D �}|| _| �� | _dS )z�
Create a new execution plan.
Args:
plan: array of strings that represents the collection operations
the output from GRAPH.EXPLAIN
zplan must be an arrayr c S s g | ] }|� � �qS r )�decode)�.0�br r r �
<listcomp>G � z*ExecutionPlan.__init__.<locals>.<listcomp>N)r �listr �bytes�plan�_operation_tree�structured_plan)r r. r r r r ; s
zExecutionPlan.__init__c C sT ||kr dS |� � |� � kr dS t|� � �D ]"}| �|j| |j| �s, dS q,dS )z{
Compare execution plan operation tree
Return: True if operation trees are equal, False otherwise
FT)r �range�_compare_operationsr )r �root_a�root_b�ir r r r2 L s z!ExecutionPlan._compare_operationsr c C s" dd� }dd� }| � | jt||�S )Nc S s d� dd� | D ��S )N�
c S s" g | ]}|� � D ] }d | �qqS )� )�
splitlines)r( Z str_child�liner r r r* e s �z?ExecutionPlan.__str__.<locals>.aggraget_str.<locals>.<listcomp>)�join)Z str_childrenr r r � aggraget_strc s
��z+ExecutionPlan.__str__.<locals>.aggraget_strc S s | � d|� �S )Nr6 r )�x�yr r r � combine_strl s z*ExecutionPlan.__str__.<locals>.combine_str)�_operation_traverser0 r% )r r; r> r r r r" b s
�zExecutionPlan.__str__r c C s& t |t�sdS | j}|j}| �||�S )zfCompares two execution plans
Return: True if the two plans are equal False otherwise
F)r r& r0 r2 )r r r3 r4 r r r r s s
zExecutionPlan.__eq__c sF �|�}t |j�dkr|S � ���fdd�|jD �}�|� |��S dS )aq
Traverse 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 c s g | ]}�� |�� ���qS r )r? )r( r �� aggregate_f� combine_f�op_fr r r r* � s �z5ExecutionPlan._operation_traverse.<locals>.<listcomp>Nr )r �oprC rA rB Zop_resr r r@ r r? � s �z!ExecutionPlan._operation_traversec C s� d}d}g }d}dd� }|t | j�k r�| j| }|�d�}||krt||�d��}|rf|�� }|�|� |}|d7 }q||d kr�||�d��}|�|� |�|� |}|d7 }|d7 }q||k r�|| d } t| �D ] }
|�� }q�|| 8 }qtd��q|d S ) z7Build the operation tree from the string representationr Nc S s� d }| d � � }| �d� t| �dkrzd| d v rztt�d| d ��d��}tt�d| d ��d��}t||�}| �d� t |t| �dkr�d n
| d � � |�S )Nr zRecords produced���zRecords produced: (\d+)� z Execution time: (\d+.\d+) ms)
�strip�popr �int�re�search�group�floatr r )r r r r r r r r �_create_operation� s
��
�z8ExecutionPlan._operation_tree.<locals>._create_operationr7 �|rF zcorrupted plan) r r. �count�splitrH r r r1 r ) r r5 �level�stack�currentrN Z
current_opZop_levelr Z levels_back�_r r r r/ � s:
z ExecutionPlan._operation_treeN)
r
r r r
r r2 r% r" r# r$ r r? r/ r r r r r& 6 s r& )rJ r r r& r r r r �<module> s
(