File: //usr/lib/python3.9/site-packages/redis/commands/graph/__pycache__/__init__.cpython-39.pyc
a
��b� � @ sX d dl mZmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z G dd� de�Z d S )
� )� quote_string�
random_string�stringify_param_value� )�
GraphCommands)�Edge)�Node)�Pathc @ s� e Zd ZdZe� fdd�Zedd� �Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d �d d �Zd d!� Zd"d#� Zd$d%� Zd&S )'�Graphz/
Graph, collection of nodes and edges.
c C s< || _ || _|j| _i | _g | _g | _g | _g | _d| _dS )z%
Create a new graph.
� N) �NAME�clientZexecute_command�nodes�edges�_labels� _properties�_relationship_types�version)�selfr
�name� r �A/usr/lib/python3.9/site-packages/redis/commands/graph/__init__.py�__init__
s zGraph.__init__c C s | j S �N)r �r r r r r s z
Graph.namec C s g | _ g | _g | _d S r )r r r r r r r �
_clear_schema s zGraph._clear_schemac C s$ | � � | �� | �� | �� d S r )r �_refresh_labels�_refresh_relations�_refresh_attributesr r r r �_refresh_schema% s zGraph._refresh_schemac C s< | � � }d gt|� | _t|�D ]\}}|d | j|< q d S �Nr )�labels�lenr � enumerate)r Zlbls�i�lr r r r + s zGraph._refresh_labelsc C s< | � � }d gt|� | _t|�D ]\}}|d | j|< q d S r )�relationship_typesr" r r# )r Zrelsr$ �rr r r r 3 s zGraph._refresh_relationsc C s< | � � }d gt|� | _t|�D ]\}}|d | j|< q d S r )�
property_keysr" r r# )r Zpropsr$ �pr r r r ; s zGraph._refresh_attributesc C s8 z| j | }W n$ ty2 | �� | j | }Y n0 |S )zo
Returns a label by it's index
Args:
idx:
The index of the label
)r �
IndexErrorr )r �idxZlabelr r r � get_labelC s zGraph.get_labelc C s8 z| j | }W n$ ty2 | �� | j | }Y n0 |S )z~
Returns a relationship type by it's index
Args:
idx:
The index of the relation
)r r* r )r r+ Zrelationship_typer r r � get_relationT s zGraph.get_relationc C s8 z| j | }W n$ ty2 | �� | j | }Y n0 |S )zu
Returns a property by it's index
Args:
idx:
The index of the property
)r r* r )r r+ Z propertier r r � get_propertye s zGraph.get_propertyc C s" |j du rt� |_ || j|j < dS )z+
Adds a node to the graph.
N)�aliasr r )r �noder r r �add_nodev s
zGraph.add_nodec C s4 | j |jj r | j |jj s$td��| j�|� dS )z,
Adds an edge to the graph.
z$Both edge's end must be in the graphN)r Zsrc_noder/ Z dest_node�AssertionErrorr �append)r �edger r r �add_edge~ s zGraph.add_edgec C sH t |t�std��d}|�� D ]$\}}|t|�d t|� d 7 }q |S )Nz'params' must be a dictzCYPHER �=� )�
isinstance�dict� TypeError�items�strr )r �paramsZ
params_header�key�valuer r r �_build_params_header� s
zGraph._build_params_headerF�� read_onlyc O sX dd� |D �}d|� dd� |�� d�}|�dd �}|rJ|dd� |�� �7 }| j||d �S )
Nc S s g | ] }t |��qS r )r )�.0�argr r r �
<listcomp>� � z(Graph.call_procedure.<locals>.<listcomp>zCALL �(�,�)�yz YIELD rA )�join�get�query)r Z procedurerB �argsZkwagrs�qrJ r r r �call_procedure� s zGraph.call_procedurec C s | j ddd�jS )Nz db.labelsTrA �rP Z
result_setr r r r r! � s z Graph.labelsc C s | j ddd�jS )Nzdb.relationshipTypesTrA rQ r r r r r& � s zGraph.relationship_typesc C s | j ddd�jS )Nzdb.propertyKeysTrA rQ r r r r r( � s zGraph.property_keysN)�__name__�
__module__� __qualname__�__doc__r r �propertyr r r r r r r, r- r. r1 r5 r@ rP r! r& r( r r r r r
s&
r
N)
Zhelpersr r r Zcommandsr r4 r r0 r �pathr r
r r r r �<module> s