2i=ddlZddlmZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZd ZdZdZdZdZdZdZdZdZdZdZ dZ!eeeeeeeeee ee!g Z"GddeZ#dZ$d1dZ%de&fdZ'de(fdZ)de*fd Z+de,fd!Z-defd"Z.defd#Z/defd$Z0defd%Z1de fd&Z2de fd'Z3d(Z4defd)Z5defd*Z6defd+Z7defd,Z8d-Z9d.Z:e$e%e'e)e+e-e.e6e5e7e8e9e/e0e1e2e3gZ;Gd/d0Zr?iscalars r8 __parse_arrayrZs66;3u:5F Gl58U+ GF G MHs1c>|Dcgc] }t|c}Scc}w)z Parse a vector32f. Args: value: The vector to parse. graph: The graph instance. Returns: list: The parsed vector. rQ)r>r?vs r8__parse_vectorf32r]s$ $E!H $$ $sc,tj|SN)rutcfromtimestampr=s r8__parse_datetimeras  $ $U ++r7cHtj|jSr_)rr`rr=s r8 __parse_daterc  $ $U + 0 0 22r7cHtj|jSr_)rr`rr=s r8 __parse_timerfrdr7c^tj|}tddd}t||S)Nir )rr`r )r>r? timestampepochs r8__parse_durationrjs-))%0I T1a E E **r7c|i}|D]4}|jj|d}t|dd|}|||<6|S)z Parse node/edge properties. Args: props (List): List of properties. graph: The graph instance. Returns: dict: Dictionary containing parsed properties. rr N)schema get_propertyrW)propsr? propertiesprop prop_name prop_values r8__parse_entity_propertiesrssTJ+LL--d1g6 !$qr(E2 * 9+ r7ct|d}d}t|ddkDr+|dDcgc]}|jj|}}t |d|}t |d||Scc}w)z Parse the value to a node. Args: value: The value to parse. graph: The graph instance. Returns: Node: The parsed Node instance. rNr r)node_idaliaslabelsro)rKrVrl get_labelrsr )r>r?rvrx inner_labelros r8 __parse_noder{ss%(mG F 58}qINqR+%,,((5RR*58U;J r&Z PPSs"A+ct|d}|jj|d}t|d}t|d}t|d|}t |||||S)z Parse the value to an edge. Args: value: The value to parse. graph: The graph instance. Returns: Edge: The parsed Edge instance. rr rrr)edge_idro)rKrl get_relationrsr )r>r?r}relation src_node_id dest_node_idros r8 __parse_edgersg%(mG||((q2HeAh-KuQx=L*58U;J  X|WQ[ \\r7cVt|d|}t|d|}t||S)z Parse the value to a path. Args: value: The value to parse. graph: The graph instance. Returns: Path: The parsed Path instance. rr )rWr)r>r?nodesedgess r8 __parse_pathr s1 q5 )E q5 )E u r7ct}t|}td|dD]&}t|||}t ||dz|||<(|S)z Parse the value as a map. Args: value: The value to parse. graph: The graph instance. Returns: OrderedDict: The parsed OrderedDict. rrr )rrVrUrIrW)r>r?m n_entriesrXkeys r8 __parse_maprs[  AE I 1i #3U1Xu-eAElE2#3 Hr7c@t|dt|dd}|S)z Parse the value to point. Args: value: The value to parse. graph: The graph instance. Returns: dict: The parsed dictionary representing a point. rr )latitude longituderQ)r>r?ps r8 __parse_pointr/s#58_5q?CA Hr7cJt|d}|d}t|||}|S)z Parse a scalar value from a value in the result set. Args: value: The value to parse. graph: The graph instance. Returns: Any: The parsed scalar value. rr )rKPARSE_SCALAR_TYPES)r>r? scalar_typerYs r8rWrW=s0eAh-K !HE  ,UE :F Mr7cheZdZdZdZdZdZdZdZdZ e de fd Z e de fd Z e defd Ze defd Ze defd Ze defdZe defdZe defdZe defdZe defdZe defdZe defdZe defdZe defdZy) QueryResultzD Represents the result of a query operation on a graph. c||_g|_g|_g|_|j |t |dk(r |d|_y|d|_|j |y)z Initializes a QueryResult instance. Args: graph: The graph on which the query was executed. response: The response from the server. r rN)_graph_header _result_set _raw_stats_QueryResult__check_for_errorsrV_QueryResult__parse_results)selfr?responses r8__init__zQueryResult.__init__gsc!   ) x=A &qkDO'rlDO   *r7ct|dtr%|d}t|dk(r|d}t|}|t|dtr|dy)z Checks if the response contains an error. Args: response: The response from the server. Raises: ResponseError: If an error is encountered. rzversion mismatchr rN)rEr rHr)rrerrorversions r8__check_for_errorszQueryResult.__check_for_errors~s_ hqk= 1QKE5z//"1+6w?K hrlM 22,  3r7c|j||_t|jdk(ry|j||_y)z Parse the query execution result returned from the server. Args: raw_result_set: The raw result set from the server. rN)_QueryResult__parse_headerrrV_QueryResult__parse_recordsr)rraw_result_sets r8__parse_resultszQueryResult.__parse_resultss?**>:  t||  ! //?r7c|jD]7}||vst|jddjddcSy)a Get the value of a specific statistical metric. Args: s (str): The statistical metric to retrieve. Returns: float: The value of the specified statistical metric. Returns 0 if the metric is not found. z: r  r)rrRsplit)rsstats r8__get_statisticszQueryResult.__get_statisticssNOO @DDyTZZ-a066s;A>?? @r7c|d}|S)z Parse the header of the result. Args: raw_result_set: The raw result set from the server. Returns: list: An array of column name/column type pairs. rr6)rrheaders r8__parse_headerzQueryResult.__parse_headers " r7c |dDcgc]&}|Dcgc]}t||jc}(}}}|Scc}wcc}}w)z Parses the result set and returns a list of records. Args: raw_result_set: The raw result set from the server. Returns: list: A list of records. r )rWr)rrrowcellrecordss r8__parse_recordszQueryResult.__parse_recordssM&a( := =\$ , =   > s >9 >>rAc|jS)z~ Get the header of the result. Returns: list: An array of column name/column type pairs. )rrs r8rzQueryResult.headers||r7c|jS)z Get a list of the results from a query. Returns: list: A list of each row returned from a query. )rrs r8 result_setzQueryResult.result_setsr7c,|jtS)zy Get the number of labels added in the query. Returns: int: The number of labels added. )_QueryResult__get_statistics LABELS_ADDEDrs r8 labels_addedzQueryResult.labels_addeds$$\22r7c,|jtS)z Get the number of labels removed in the query. Returns: int: The number of labels removed. )rLABELS_REMOVEDrs r8labels_removedzQueryResult.labels_removed$$^44r7c,|jtS)z Get the number of nodes created in the query. Returns: int: The number of nodes created. )r NODES_CREATEDrs r8 nodes_createdzQueryResult.nodes_created$$]33r7c,|jtS)z Get the number of nodes deleted in the query. Returns: int: The number of nodes deleted. )r NODES_DELETEDrs r8 nodes_deletedzQueryResult.nodes_deletedrr7c,|jtS)z Get the number of properties set in the query. Returns: int: The number of properties set. )rPROPERTIES_SETrs r8properties_setzQueryResult.properties_setrr7c,|jtS)z Get the number of properties removed in the query. Returns: int: The number of properties removed. )rPROPERTIES_REMOVEDrs r8properties_removedzQueryResult.properties_removeds$$%788r7c,|jtS)z Get the number of relationships created in the query. Returns: int: The number of relationships created. )rRELATIONSHIPS_CREATEDrs r8relationships_createdz!QueryResult.relationships_created#$$%:;;r7c,|jtS)z Get the number of relationships deleted in the query. Returns: int: The number of relationships deleted. )rRELATIONSHIPS_DELETEDrs r8relationships_deletedz!QueryResult.relationships_deleted-rr7c,|jtS)z Get the number of indices created in the query. Returns: int: The number of indices created. )rINDICES_CREATEDrs r8indices_createdzQueryResult.indices_created7$$_55r7c,|jtS)z Get the number of indices deleted in the query. Returns: int: The number of indices deleted. )rINDICES_DELETEDrs r8indices_deletedzQueryResult.indices_deletedArr7c2|jtdk(S)z Check if the query execution plan was cached. Returns: bool: True if the query execution plan was cached, False otherwise. r )rCACHED_EXECUTIONrs r8cached_executionzQueryResult.cached_executionKs$$%56!;;r7c,|jtS)z Get the server execution time of the query. Returns: float: The server execution time of the query in milliseconds. )rINTERNAL_EXECUTION_TIMErs r8 run_time_mszQueryResult.run_time_msUs$$%<==r7N)r!r"r#r$rrrrrrpropertylistrrrKrrrrrrrrrrboolrrRrr6r7r8rrcs+., @ " D  3c335554s444s445559C99U>>r7r)rAN)=r:enumrtypingr collectionsrrrrdateutil.relativedeltar redisr edger noder pathr exceptionsrrrrrrrrrrrrrSTATSrr@rCrHrIrKrLrrOrRrSrZr]rarcrfrjrsr{rrrrrWrrr6r7r8rs #))06) *) ) *++,.113 (4(T ' C& S  T  E  4  %t %,h,3$33$3+m+ &Q$Q$]$]$ $   (  &#(z>z>r7