mBNj=ddlZddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZd Zd ZdZdZdZdZdZdZdZdZ dZ!dZ"eeeeeeeeee!e e"g Z#GddeZ$dZ%d0dZ&de'fdZ(de)fdZ*de+fdZ,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.d/Z=y)1N) 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) r 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_DURATIONn/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/falkordb/asyncio/query_result.pyrr-s^.MJLMMLKJJJIKONJJNr8rcJKtjjdyw)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%&s!#returnc Kyw)z Parse a null value. Args: value: The null value. graph: The graph instance. Returns: None: Always returns None. Nr7r>s r9 __parse_nullrDfs  scKt|tr|jSt|ts t|S|Sw)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_stringrJts7%||~ eS !5z Ls>Ac Kt|Sw)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: cXKt|tr|jn|}|dk(Sw)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_booleanrQs()6ELLNEE F?s(*c Kt|Sw)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_doublerUs <rNcKtt|Dcgc]}t|||d{}}|S7 cc}ww)z Parse an array of values. Args: value: The array value to parse. graph: The graph instance. Returns: list: The parsed list of values. N)rangelen parse_scalar)r?r@iscalars r9 __parse_arrayr\sFs r9__parse_datetimerfs  ! !%HLL 99s&(cpKtj|tjj Swra)rrdrrerr>s r9 __parse_daterh&  ! !%HLL 9 > > @@46cpKtj|tjj Swra)rrdrrerr>s r9 __parse_timerlrirjcKtj|tj}tdddtj}t ||Sw)Nrbir)tzinfo)rrdrrer )r?r@ timestampepochs r9__parse_durationrqs=&&u>I T1a 5E E **sAAcKi}|D]D}|jj|dd{}t|dd|d{}|||<F|S7$7w)z Parse node/edge properties. Args: props (List): List of properties. graph: The graph instance. Returns: dict: Dictionary containing parsed properties. rNr)schema get_propertyrY)propsr@ propertiesprop prop_name prop_values r9__parse_entity_propertiesrzsbJ,,33DG<< 'QR%88 * 9  =8s!)AAAAAAcKt|d}d}t|ddkDr3|dDcgc]%}|jj|d{'}}t |d|d{}t |d||S7.cc}w7w)z Parse the value to a node. Args: value: The value to parse. graph: The graph instance. Returns: Node: The parsed Node instance. rNrr)node_idaliaslabelsrv)rLrXrs get_labelrzr)r?r@r}r inner_labelrvs r9 __parse_noders%(mG F 58}qOTUVxXx  ..{;;;xX0q5AAJ r&Z PP<XAs4)B#A?A= A?B)B*B=A??BcKt|d}|jj|dd{}t|d}t|d}t|d|d{}t |||||S7G7w)z Parse the value to an edge. Args: value: The value to parse. graph: The graph instance. Returns: Edge: The parsed Edge instance. rrNrrr)edge_idrv)rLrs get_relationrzr )r?r@rrelation src_node_id dest_node_idrvs r9 __parse_edger s%(mG\\..uQx88HeAh-KuQx=L0q5AAJ X|W  9Bs!0A>A:2A>%A<&A><A>cKt|d|d{}t|d|d{}t||S7'7w)z Parse the value to a path. Args: value: The value to parse. graph: The graph instance. Returns: Path: The parsed Path instance. rNr)rYr)r?r@nodesedgess r9 __parse_pathrsDuQx/ /EuQx/ /E u  0 /sA=A?AAcKt}t|}td|dD]6}t|||d{}t ||dz|d{||<8|S7%7 w)z Parse the value as a map. Args: value: The value to parse. graph: The graph instance. Returns: OrderedDict: The parsed OrderedDict. rrNr)rrXrWrJrY)r?r@m n_entriesrZkeys r9 __parse_mapr/si  AE I 1i #"58U33#E!a%L%88#$ H48s!8A$A A$A" A$"A$cHKt|dt|dd}|Sw)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 longituderS)r?r@ps r9 __parse_pointrDs'58_5q?CA Hs "cfKt|d}|d}t|||d{}|S7w)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. rrN)rLPARSE_SCALAR_TYPES)r?r@ scalar_typer[s r9rYrYSs=eAh-K !HE%k25%@ @F MAs &1/1cBeZdZdZdZdZdZdZdZdZ dZ e d e fd Z e d e fd Ze d e fd Ze d e fd Ze d e fdZe d e fdZe d e fdZe d e fdZe d e fdZe d e fdZe d efdZe d efdZy) QueryResultz@ Represents the result of a query operation on a graph. c<||_g|_g|_g|_y)z Initializes a QueryResult instance. Args: graph: The graph on which the query was executed. N)r@header result_set _raw_stats)selfr@s r9__init__zQueryResult.__init__s   r8cK|j|t|dk(r |d|_y|d|_|j|d{y7w)zx Parse the response from the server. Args: response: The response from the server. rrN)_QueryResult__check_for_errorsrXr_QueryResult__parse_results)rresponses r9parsezQueryResult.parsesN ) x=A &qkDO'rlDO&&x0 0 0sA A A Act|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 mismatchrrN)rFr rIr )rrerrorversions r9__check_for_errorszQueryResult.__check_for_errorss_ hqk= 1QKE5z//"1+6w?K hrlM 22,  3r8cK|j||_t|jdk(ry|j|d{|_y7 w)z Parse the query execution result returned from the server. Args: raw_result_set: The raw result set from the server. rN)_QueryResult__parse_headerrrX_QueryResult__parse_recordsr)rraw_result_sets r9__parse_resultszQueryResult.__parse_resultssH)).9  t{{ q  $ 4 4^ DDDsAAA Ac|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)rrTsplit)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)rrrs r9__parse_headerzQueryResult.__parse_headers " r8cKg}|dD]I}g}|D]/}|jt||jd{1|j|K|S7w)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. rN)appendrYr@)rrrecordsrowrecordcells r9__parse_recordszQueryResult.__parse_recordss^!!$CF Ltzz$BBC NN6 " % Cs6AA ArBc,|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_setrr8c,|jtS)z Get the number of properties removed in the query. Returns: int: The number of properties removed. )rPROPERTIES_REMOVEDrs r9properties_removedzQueryResult.properties_removed(s$$%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_created2$$%:;;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_deleted<rr8c,|jtS)z Get the number of indices created in the query. Returns: int: The number of indices created. )rINDICES_CREATEDrs r9indices_createdzQueryResult.indices_createdF$$_55r8c,|jtS)z Get the number of indices deleted in the query. Returns: int: The number of indices deleted. )rINDICES_DELETEDrs r9indices_deletedzQueryResult.indices_deletedPrr8c2|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_executionZs$$%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_msds$$%<==r8N)r"r#r$r%rrrrrrrpropertyrLrrrrrrrrrrboolrrTrr7r8r9rrzso 1$, E" &3c335554s444s445559C99U>>r8r)rBN)>r; collectionsrrrrrenumrtypingr dateutil.relativedeltar redisr falkordb.edger falkordb.exceptionsr falkordb.noder falkordb.pathrrrrrrrrrrrrrSTATSrrArDrIrJrLrMrrQrTrUr\r_rfrhrlrqrzrrrrrrYrrr7r8r9rs #330> !  !##%)//3  (4(V ' #( 3  4  %    %T %:H:AAAA+M+ (QQ&*    { *  &#*r>r>r8