File: //usr/lib/python3.9/site-packages/redis/commands/graph/__pycache__/query_result.cpython-39.pyc
a
��b�+ �
@ s� d dl mZ d dlmZ ddlmZ ddlmZ ddlm Z ddl
m Z dZ d Z
d
Zd Zd Zd
ZdZdZdZdZe e
eeeeeeeeg
ZG dd� d�ZG dd� d�ZG dd� d�ZdS )� )� OrderedDict)�
ResponseError� )�Edge)�VersionMismatchException)�Node)�Pathz Labels addedz
Nodes createdz
Nodes deletedzRelationships deletedzProperties setzRelationships createdzIndices createdzIndices deletedzCached executionzinternal execution timec @ s e Zd ZdZdZdZdZdS )�ResultSetColumnTypesr r � � N)�__name__�
__module__� __qualname__ZCOLUMN_UNKNOWN�
COLUMN_SCALAR� COLUMN_NODE�COLUMN_RELATION� r r �E/usr/lib/python3.9/site-packages/redis/commands/graph/query_result.pyr $ s r c @ s< e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
S )�ResultSetScalarTypesr r r
r � � � � � � �
� N)r r
r �
VALUE_UNKNOWN�
VALUE_NULL� VALUE_STRING�
VALUE_INTEGER�
VALUE_BOOLEAN� VALUE_DOUBLE� VALUE_ARRAY�
VALUE_EDGE�
VALUE_NODE�
VALUE_PATH� VALUE_MAP� VALUE_POINTr r r r r + s r c @ s e Zd Zd;dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zed"d#� �Zd$d%� Zed&d'� �Zed(d)� �Zed*d+� �Zed,d-� �Zed.d/� �Zed0d1� �Z ed2d3� �Z ed4d5� �Z ed6d7� �Z ed8d9� �Z d:S )<� QueryResultFc C sd || _ g | _g | _| �|� t|�dkr8| �|d � n(|rH| �|� n| �|d � | �|� dS )a7
A class that represents a result of the query operation.
Args:
graph:
The graph on which the query was executed.
response:
The response from the server.
profile:
A boolean indicating if the query command was "GRAPH.PROFILE"
r r ���N)�graph�header�
result_set�_check_for_errors�len�parse_statistics�
parse_profile�
parse_results)�selfr+ �responseZprofiler r r �__init__; s
zQueryResult.__init__c C sP t |d t�r6|d }t|�dkr2|d }t|�}|�t |d t�rL|d �d S )Nr zversion mismatchr r* )�
isinstancer �strr )r3 r4 �error�versionr r r r. X s z QueryResult._check_for_errorsc C s. | � |�| _t| j�dkr d S | �|�| _d S �Nr )� parse_headerr, r/ �
parse_recordsr- )r3 �raw_result_setr r r r2 e s zQueryResult.parse_resultsc C sZ i | _ t|�D ] \}}t|t�r|�� ||< qtD ]"}| �||�}|d ur2|| j |< q2d S �N)�
statistics� enumerater6 �bytes�decode�STATS�
_get_value)r3 Zraw_statistics�idx�stat�s�vr r r r0 n s
z QueryResult.parse_statisticsc C s |d }|S r: r )r3 r= r, r r r r; { s zQueryResult.parse_headerc C s� g }|d }|D ]�}g }t |�D ]�\}}| j| d tjkrN|�| �|�� q | j| d tjkrt|�| �|�� q | j| d tjkr�|�| � |�� q t
d� q |�|� q|S )Nr r zUnknown column type.
) r@ r, r r �append� parse_scalarr �
parse_noder �
parse_edge�print)r3 r= Zrecordsr- �row�recordrE �cellr r r r< � s �
zQueryResult.parse_recordsc C s< i }|D ].}| j �|d �}| �|dd � �}|||< q|S �Nr r )r+ Z get_propertyrJ )r3 Zprops�
properties�propZ prop_nameZ
prop_valuer r r �parse_entity_properties� s
z#QueryResult.parse_entity_propertiesc C s, t |t�r|�� S t |t�s$t|�S |S d S r> )r6 rA rB r7 )r3 rP r r r � parse_string� s
zQueryResult.parse_stringc C s` t |d �}d }t|d �dkrDg }|d D ]}|�| j�|�� q,| �|d �}t|||d�S )Nr r r
)�node_idZlabelrR )�intr/ rI r+ Z get_labelrT r )r3 rP rV �labelsZ inner_labelrR r r r rK � s zQueryResult.parse_nodec C sT t |d �}| j�|d �}t |d �}t |d �}| �|d �}t|||||d�S )Nr r r
r r )�edge_idrR )rW r+ Z get_relationrT r )r3 rP rY ZrelationZ src_node_idZ dest_node_idrR r r r rL � s
�zQueryResult.parse_edgec C s& | � |d �}| � |d �}t||�S rQ )rJ r )r3 rP ZnodesZedgesr r r �
parse_path� s zQueryResult.parse_pathc C sH t � }t|�}td|d�D ](}| �|| �}| �||d �||< q|S )Nr r
r )r r/ �rangerU rJ )r3 rP �mZ n_entries�i�keyr r r � parse_map� s zQueryResult.parse_mapc C s( i }t |d �|d< t |d �|d< |S )Nr Zlatituder Z longitude)�float)r3 rP �pr r r � parse_point� s zQueryResult.parse_pointc C sz t |d �}|d }d }|tjkr*d }�nL|tjkrB| �|�}�n4|tjkrXt |�}�n |tjkr�t|t�rt|� � n|}|dkr�d}n|dkr�d}nt
d� n�|tj kr�t |�}n�|tj
kr�| }}tt|��D ]}| �|| �||< q�n�|tjk�r| �|�}nt|tjk�r| �|�}n\|tjk�r2| �|�}nD|tjk�rJ| �|�}n,|tjk�rb| �|�}n|tjk�rvt
d� |S ) Nr r �trueTZfalseFzUnknown boolean type
zUnknown scalar type
) rW r r r rU r r! r6 rA rB rM r" r` r# r[ r/ rJ r% rK r$ rL r& rZ r' r_ r( rb r )r3 rP Z scalar_type�valueZscalarZarrayr] r r r rJ � sF
zQueryResult.parse_scalarc C s dd� |D �| _ d S )Nc S s" g | ]}|d |� d�� �� �qS )r �,)�index�strip)�.0�xr r r �
<listcomp> � z-QueryResult.parse_profile.<locals>.<listcomp>)r- )r3 r4 r r r r1 s zQueryResult.parse_profilec C s t | j�dkS r: )r/ r- �r3 r r r �is_empty6 s zQueryResult.is_emptyc C s6 |D ],}| |v rt |�d�d �d�d � S qd S )Nz: r � r )r` �split)rS r? rF r r r rD 9 s "zQueryResult._get_valuec C s || j v r| j | S dS r: )r? )r3 rF r r r � _get_statA s zQueryResult._get_statc C s
| � t�S r> )rp � LABELS_ADDEDrl r r r � labels_addedD s zQueryResult.labels_addedc C s
| � t�S r> )rp �
NODES_CREATEDrl r r r �
nodes_createdH s zQueryResult.nodes_createdc C s
| � t�S r> )rp �
NODES_DELETEDrl r r r �
nodes_deletedL s zQueryResult.nodes_deletedc C s
| � t�S r> )rp �PROPERTIES_SETrl r r r �properties_setP s zQueryResult.properties_setc C s
| � t�S r> )rp �RELATIONSHIPS_CREATEDrl r r r �relationships_createdT s z!QueryResult.relationships_createdc C s
| � t�S r> )rp �RELATIONSHIPS_DELETEDrl r r r �relationships_deletedX s z!QueryResult.relationships_deletedc C s
| � t�S r> )rp �INDICES_CREATEDrl r r r �indices_created\ s zQueryResult.indices_createdc C s
| � t�S r> )rp �INDICES_DELETEDrl r r r �indices_deleted` s zQueryResult.indices_deletedc C s | � t�dkS )Nr )rp �CACHED_EXECUTIONrl r r r �cached_executiond s z QueryResult.cached_executionc C s
| � t�S r> )rp �INTERNAL_EXECUTION_TIMErl r r r � run_time_msh s zQueryResult.run_time_msN)F)!r r
r r5 r. r2 r0 r; r< rT rU rK rL rZ r_ rb rJ r1 rm � staticmethodrD rp �propertyrr rt rv rx rz r| r~ r� r� r� r r r r r) : sN
4$
r) N)� collectionsr Zredisr Zedger �
exceptionsr Znoder �pathr rq rs ru r{ rw ry r} r r� r� rC r r r) r r r r �<module> s: �