2ibddlmZmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZdZd Zd Zd Zd Zd ZdZdZdZGddZy))ListDictOptional) GraphSchema) QueryResult) ExecutionPlan)SchemaVersionMismatchException) quote_stringstringify_param_valuez DB.INDEXESzDB.CONSTRAINTSz GRAPH.COPYz GRAPH.QUERYz GRAPH.DELETEz GRAPH.EXPLAINz GRAPH.SLOWLOGz GRAPH.PROFILEzGRAPH.RO_QUERYc DeZdZdZdefdZedefdZ dBdedee ee fd ee d e de f d Z dCdedee ee fd ee de fd Z dCdedee ee fd ee de fd ZdefdZdDdZdZdZdEdedefdZdEdedefdZdedefdZ dFded e deedeeede f dZdedededede f dZdedede fd Zdedede fd!Zdedede fd"Zdedede fd#Z dedede fd$Z!dedede fd%Z"de fd&Z#dd'dededed(eede f d)Z$dede fd*Z%dede fd+Z&d,d-d.ded/e d0ede fd1Z'd2ede fd3Z(d2ede fd4Z)d,d-d.d2ed/e d0ede fd5Z*d6ededefd7Z+defd8Z,d2efd9Z-defd:Z.d2efd;Z/d6ededefd<Z0defd=Z1d2efd>Z2defd?Z3d2efd@Z4de ee fgfdAZ5y)GGraphz/ Graph, collection of nodes and edges. namecb||_||_t||_|j|_y)z} Create a new graph. Args: client: The client object. name (str): Graph ID N)_nameclientrschemaexecute_command)selfrrs X/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/falkordb/graph.py__init__zGraph.__init__s, $ % *40 %55returnc|jS)zY Get the graph name. Returns: str: The graph name. )rrs rrz Graph.name*szzrNqparamstimeout read_onlyct|}|j||z}|rtnt}||j|dg}t |t r|j d|gn | td |j|}t||S#t$r,} |jj| j| d} ~ wwxYw)a Executes a query against the graph. See: https://docs.falkordb.com/commands/graph.query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. read_only (bool): Whether the query is read-only. Returns: QueryResult: query result set. z --compactrNz+Timeout argument must be a positive integer)_build_params_header RO_QUERY_CMD QUERY_CMDr isinstanceintextend Exceptionrrr rrefreshversion) rrrrrquerycmdcommandresponsees r_queryz Graph._query6s$))&1E9 (lY 5+6 gs # NNIw/ 0  IJ J +t++W5HtX. .-  KK   *G  s'B B7 'B22B7c,|j|||dS)aR Executes a query against the graph. See: https://docs.falkordb.com/commands/graph.query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. Returns: QueryResult: query result set. Frrrr/rrrrs rr*z Graph.querycs {{1VW{NNrc,|j|||dS)a_ Executes a read-only query against the graph. See: https://docs.falkordb.com/commands/graph.ro_query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. Returns: QueryResult: query result set. Tr1r2r3s rro_queryzGraph.ro_queryus {{1VW{MMrclonecp|jt|j|t|j|S)z Creates a copy of graph Args: clone (str): Name of cloned graph Returns: Graph: the cloned graph )rCOPY_CMDrrr)rr6s rcopyz Graph.copys+ Xtyy%8T[[%((rcv|jj|jt|jS)z Deletes the graph. See: https://docs.falkordb.com/commands/graph.delete.html Returns: None )rclearr DELETE_CMDrrs rdeletez Graph.deletes+ ##J ;;rcB|jt|jS)a Get a list containing up to 10 of the slowest queries issued against the graph. Each item in the list has the following structure: 1. a unix timestamp at which the log entry was processed 2. the issued command 3. the issued query 4. the amount of time needed for its execution, in milliseconds. See: https://docs.falkordb.com/commands/graph.slowlog.html Returns: List: List of slow log entries. r SLOWLOG_CMDrrs rslowlogz Graph.slowlogs$##K<rr*c|j||z}|jt|j|}t |S)a Execute a query and produce an execution plan augmented with metrics for each operation's execution. Return an execution plan, with details on results produced by and time spent in each operation. See: https://docs.falkordb.com/commands/graph.profile.html Args: query (str): The query to profile. params (dict): Query parameters. Returns: ExecutionPlan: The profile information. )r!r PROFILE_CMDrr rr*rplans rprofilez Graph.profiles< ))&1E9##KUCT""rc|j||z}|jt|j|}t |S)a{ Get the execution plan for a given query. GRAPH.EXPLAIN returns an ExecutionPlan object. See: https://docs.falkordb.com/commands/graph.explain.html Args: query (str): The query for which to get the execution plan. params (dict): Query parameters. Returns: ExecutionPlan: The execution plan. )r!r EXPLAIN_CMDrr rGs rexplainz Graph.explains<))&1E9##KUCT""rc|yt|ts tdd}|jD]%\}}|t |dzt |zdzz }'|S)z Build parameters header. Args: params (dict): The parameters. Returns: str: The parameters header. z'params' must be a dictzCYPHER = )r$dict TypeErroritemsstrr )rr params_headerkeyvalues rr!zGraph._build_params_headersj >&$'56 6!  ,,. QJC SX^.CE.JJSP PM Qr procedureargsemitc"|xsg}d}t|dkDr'i}t|D]\}}d|}|||<d|z||<d|ddj|d} |%t|dkDr| d dj|z } |j| || S) a@ Call a procedure. Args: procedure (str): The procedure to call. read_only (bool): Whether the procedure is read-only. args: Procedure arguments. emit: Procedure yield. Returns: QueryResult: The result of the procedure call. Nrparam$zCALL (,)zYIELD )rr)len enumeratejoinr/) rrXrrYrZriarg param_namers rcall_procedurezGraph.call_procedures$zr t9q=F$D/ +3$QC[ %(z" *Q + I;a/q 1  D A 6#((4.)* *A{{1Vy{AAridx_type entity_typelabel attributec|dk(rd|d}n|dk(rd|d}n td|dk(r d |d |d}n)|d k(r d |d |d}n|d k(r d|d |d}n td|j|S)aDrop a graph index. Args: idx_type (str): The type of index ("RANGE", "FULLTEXT", "VECTOR"). entity_type (str): The type of entity ("NODE" or "EDGE"). label (str): The label of the node or edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. NODE(e:r`EDGE()-[e:]->()Invalid entity typeRANGEzDROP INDEX FOR z ON (e.VECTORzDROP VECTOR INDEX FOR FULLTEXTzDROP FULLTEXT INDEX FOR zInvalid index type) ValueErrorr*)rrhrirjrkpatternrs r _drop_indexzGraph._drop_index(s & E7!nG F "ugU+G23 3 w !'')A>A  !(  1EA  #*7)79+QGA12 2zz!}rc*|jdd||S)aVDrop a range index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rsrmrxrrjrks rdrop_node_range_indexzGraph.drop_node_range_indexI BBrc*|jdd||S)aZDrop a full-text index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rurmrzr{s rdrop_node_fulltext_indexzGraph.drop_node_fulltext_indexV FE9EErc*|jdd||S)aWDrop a vector index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rtrmrzr{s rdrop_node_vector_indexzGraph.drop_node_vector_indexc&%CCrc*|jdd||S)a^Drop a range index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rsrorzr{s rdrop_edge_range_indexzGraph.drop_edge_range_indexpr}rc*|jdd||S)abDrop a full-text index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rurorzr{s rdrop_edge_fulltext_indexzGraph.drop_edge_fulltext_index}rrc*|jdd||S)a_Drop a vector index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rtrorzr{s rdrop_edge_vector_indexzGraph.drop_edge_vector_indexrrc,|jtS)zRetrieve a list of graph indices. See: https://docs.falkordb.com/commands/graph.query.html#procedures Returns: list: List of graph indices. )rg GRAPH_INDEXESrs r list_indiceszGraph.list_indicess""=11r)options propertiesc|dk(rd|d}n|dk(rd|d}n td|dk(rd }d |d |d }|d jtdj|z }|dz }|`d}|j D];\} } t | t r|| dz| zdzz }%|| dzt | zd zz }=|dddz}|d|z }|j|S)aCreate a typed index for nodes or edges. Args: idx_type (str): The type of index ("RANGE", "FULLTEXT", "VECTOR"). entity_type (str): The type of entity ("NODE" or "EDGE"). label (str): The label of the node or edge. properties: Variable number of property names to be indexed. options (dict, optional): Additional options for the index. Returns: Any: The result of the index creation query. rmrnr`rorprqrrrsrNzCREATE z INDEX FOR z ON (r_ze.{0}N{z:'z',:}z OPTIONS )rvrcmapformatrSr$rTr*) rrhrirjrrrwr options_maprVrWs r_create_typed_indexzGraph._create_typed_indexs & E7!nG F "ugU+G23 3 w HhZ{7)5 9 SXXc'..*5 66 S  K%mmo @ UeS)3:#5#<$?O}O0;O$FJ'+N#NxS&[0A'BN}N0;N$ )# ) <=( ?#S#-#(#S#-#(4C.@D.237$B$B$B%d^$B%d3i0$BrsX''%%)68& )      w w r