mBNj=ddlZddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZd ZdZdZdZdZdZdZdZdZdZ dZ!dZ"eeeeeeeeee!e e"g Z#GddeZ$dZ%d1dZ&de'fdZ(de)fdZ*de+fd Z,de-fd!Z.de fd"Z/de fd#Z0defd$Z1defd%Z2defd&Z3de fd'Z4d(Z5defd)Z6defd*Z7defd+Z8defd,Z9d-Z:d.Z;e%e&e(e*e,e.e/e7e6e8e9e:e0e1e2e3e4gZ<Gd/d0Z=y)2N) OrderedDict)datedatetimetimetimezone)Enum)List) relativedelta) ResponseError)Edge)SchemaVersionMismatchException)Node)Pathz Labels addedzLabels removedz Nodes createdz Nodes deletedzProperties setzIndices createdzIndices deletedzCached executionzProperties removedzRelationships deletedzRelationships createdzinternal execution timecTeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZy)ResultSetScalarTypesa" Enumeration representing different scalar types in the query result set. Attributes: VALUE_UNKNOWN (int): Unknown scalar type (0) VALUE_NULL (int): Null scalar type (1) VALUE_STRING (int): String scalar type (2) VALUE_INTEGER (int): Integer scalar type (3) VALUE_BOOLEAN (int): Boolean scalar type (4) VALUE_DOUBLE (int): Double scalar type (5) VALUE_ARRAY (int): Array scalar type (6) VALUE_EDGE (int): Edge scalar type (7) VALUE_NODE (int): Node scalar type (8) VALUE_PATH (int): Path scalar type (9) VALUE_MAP (int): Map scalar type (10) VALUE_POINT (int): Point scalar type (11) VALUE_VECTORF32 (int): Vector scalar type (12) VALUE_DATETIME (int): DateTime scalar type (13) VALUE_DATE (int): Date scalar type (14) VALUE_TIME (int): Time scalar type (15) VALUE_DURATION (int): Duration scalar type (16) rr  N)__name__ __module__ __qualname____doc__ VALUE_UNKNOWN VALUE_NULL VALUE_STRING VALUE_INTEGER VALUE_BOOLEAN VALUE_DOUBLE VALUE_ARRAY VALUE_EDGE VALUE_NODE VALUE_PATH VALUE_MAP VALUE_POINTVALUE_VECTORF32VALUE_DATETIME VALUE_DATE VALUE_TIMEVALUE_DURATIONf/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/falkordb/query_result.pyrr-s^.MJLMMLKJJJIKONJJNr8rcBtjjdy)z Parse a value of unknown type. Args: value: The value to parse. graph: The graph instance. Returns: None z Unknown type N)sysstderrwritevaluegraphs r9__parse_unknownrAXsJJ%&r8returncy)z Parse a null value. Args: value: The null value. graph: The graph instance. Returns: None: Always returns None. Nr7r>s r9 __parse_nullrDfs r8c|t|tr|jSt|ts t|S|S)z Parse the value as a string. Args: value: The value to parse. graph: The graph instance. Returns: str: The parsed string value. ) isinstancebytesdecodestrr>s r9__parse_stringrJts3%||~ eS !5z Lr8ct|S)z Parse the integer value from the value. Args: value: The value to parse. graph: The graph instance. Returns: int: The parsed integer value. )intr>s r9__parse_integerrMs u:r8cPt|tr|jn|}|dk(S)z Parse the value as a boolean. Args: value: The value to parse. graph: The graph instance. Returns: bool: The parsed boolean value. true)rFrGrHr>s r9__parse_booleanrPs$)6ELLNEE F?r8ct|S)z Parse the value as a double. Args: value: The value to parse. graph: The graph instance. Returns: float: The parsed double value. floatr>s r9__parse_doublerTs <r8cltt|Dcgc]}t|||}}|Scc}w)z Parse an array of values. Args: value: The array value to parse. graph: The graph instance. Returns: list: The parsed list of values. )rangelen parse_scalar)r?r@iscalars r9 __parse_arrayr[s96;3u:5F G5Fl58U+5FF 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. rR)r?r@vs r9__parse_vectorf32r^s $ $eE!He $$ $scLtj|tjSNtz)r fromtimestamprutcr>s r9__parse_datetimeres  ! !%HLL 99r8chtj|tjj Sr`)rrcrrdrr>s r9 __parse_daterg"  ! !%HLL 9 > > @@r8chtj|tjj Sr`)rrcrrdrr>s r9 __parse_timerjrhr8ctj|tj}tdddtj}t ||S)Nrair )tzinfo)rrcrrdr )r?r@ timestampepochs r9__parse_durationros9&&u>I T1a 5E E **r8c|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_propertyrX)propsr@ propertiesprop prop_name prop_values r9__parse_entity_propertiesrxsOJLL--d1g6 !$qr(E2 * 9 r8ct|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_idaliaslabelsrt)rLrWrq get_labelrxr)r?r@r{r} inner_labelrts r9 __parse_noderst%(mG F 58}qINqR+%,,((5R*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_idrt)rLrq get_relationrxr )r?r@rrelation src_node_id dest_node_idrts r9 __parse_edger sk%(mG||((q2HeAh-KuQx=L*58U;J X|W r8cVt|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 )rXr)r?r@nodesedgess r9 __parse_pathrs1 q5 )E q5 )E u r8ct}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 )rrWrVrJrX)r?r@m n_entriesrYkeys r9 __parse_mapr/sV  AE I 1i #U1Xu-eAElE2#$ Hr8c@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 longituderR)r?r@ps r9 __parse_pointrDs#58_5q?CA Hr8cJt|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 )rLPARSE_SCALAR_TYPES)r?r@ scalar_typerZs r9rXrXSs0eAh-K !HE  ,UE :F Mr8cheZdZdZdZdZdZdZdZdZ e de fd Z e de fd Z e defd Ze defd Ze defd Ze defdZe defdZe defdZe defdZe defdZe defdZe defdZe defdZe defdZy) QueryResultz@ 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_errorsrW_QueryResult__parse_results)selfr@responses r9__init__zQueryResult.__init__sc   ) x=A &qkDO'rlDO   *r8ct|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)rFr rIr)rrerrorversions r9__check_for_errorszQueryResult.__check_for_errorss_ hqk= 1QKE5z//"1+6w?K hrlM 22,  3r8c|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_headerrrW_QueryResult__parse_recordsr)rraw_result_sets r9__parse_resultszQueryResult.__parse_resultss?**>:  t||  ! //?r8c|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)rrSsplit)rsstats r9__get_statisticszQueryResult.__get_statisticssGOODDyTZZ-a066s;A>??$r8c|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. rr7)rrheaders r9__parse_headerzQueryResult.__parse_headers " r8c |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 )rXr)rrrowcellrecordss r9__parse_recordszQueryResult.__parse_recordssT&a( (:= =\$ , =(   > s >9 >>rBc|jS)z~ Get the header of the result. Returns: list: An array of column name/column type pairs. )rrs r9rzQueryResult.headers||r8c|jS)z Get a list of the results from a query. Returns: list: A list of each row returned from a query. )rrs r9 result_setzQueryResult.result_setsr8c,|jtS)zy Get the number of labels added in the query. Returns: int: The number of labels added. )_QueryResult__get_statistics LABELS_ADDEDrs r9 labels_addedzQueryResult.labels_addeds$$\22r8c,|jtS)z Get the number of labels removed in the query. Returns: int: The number of labels removed. )rLABELS_REMOVEDrs r9labels_removedzQueryResult.labels_removed $$^44r8c,|jtS)z Get the number of nodes created in the query. Returns: int: The number of nodes created. )r NODES_CREATEDrs r9 nodes_createdzQueryResult.nodes_created$$]33r8c,|jtS)z Get the number of nodes deleted in the query. Returns: int: The number of nodes deleted. )r NODES_DELETEDrs r9 nodes_deletedzQueryResult.nodes_deletedrr8c,|jtS)z Get the number of properties set in the query. Returns: int: The number of properties set. )rPROPERTIES_SETrs r9properties_setzQueryResult.properties_set(rr8c,|jtS)z Get the number of properties removed in the query. Returns: int: The number of properties removed. )rPROPERTIES_REMOVEDrs r9properties_removedzQueryResult.properties_removed2s$$%788r8c,|jtS)z Get the number of relationships created in the query. Returns: int: The number of relationships created. )rRELATIONSHIPS_CREATEDrs r9relationships_createdz!QueryResult.relationships_created<$$%:;;r8c,|jtS)z Get the number of relationships deleted in the query. Returns: int: The number of relationships deleted. )rRELATIONSHIPS_DELETEDrs r9relationships_deletedz!QueryResult.relationships_deletedFrr8c,|jtS)z Get the number of indices created in the query. Returns: int: The number of indices created. )rINDICES_CREATEDrs r9indices_createdzQueryResult.indices_createdP$$_55r8c,|jtS)z Get the number of indices deleted in the query. Returns: int: The number of indices deleted. )rINDICES_DELETEDrs r9indices_deletedzQueryResult.indices_deletedZrr8c2|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 r9cached_executionzQueryResult.cached_executionds$$%56!;;r8c,|jtS)z Get the server execution time of the query. Returns: float: The server execution time of the query in milliseconds. )rINTERNAL_EXECUTION_TIMErs r9 run_time_mszQueryResult.run_time_msns$$%<==r8N)r"r#r$r%rrrrrrpropertylistrrrLrrrrrrrrrrboolrrSrr7r8r9rrzs+., @" " D  3c335554s444s445559C99U>>r8r)rBN)>r; collectionsrrrrrenumrtypingr dateutil.relativedeltar redisr edger exceptionsrnoderpathrrrrrrrrrrrrrSTATSrrArDrIrJrLrMrrPrSrTr[r^rergrjrorxrrrrrrXrrr7r8r9rs #3306 !  !##%)//3  (4(V ' C( S  T  E  4  %t %:h:A$AA$A+m+ (Q$Q&$* $    *  &#*|>|>r8