o `f(@sddlmZddlmZddlmZddlmZddlm Z m Z dZ dZ d Z d Zd Zd Zd ZdZGdddZGdddeZdS)) DataError) ResponseError)VersionMismatchException) ExecutionPlan)AsyncQueryResult QueryResultz GRAPH.PROFILEzGRAPH.RO_QUERYz GRAPH.QUERYz GRAPH.DELETEz GRAPH.SLOWLOGz GRAPH.CONFIGz GRAPH.LISTz GRAPH.EXPLAINc@sxeZdZdZddZdddZdd Zd d Zd d ZddZ ddZ ddZ dddZ ddZ dddZdddZdS) GraphCommandszRedisGraph CommandscCst|jdkrt|jdkrdSd}|jD] \}}|t|d7}q|ddd|jD7}|ddkr=|dd}||S)z& Create entire graph. rNzCREATE ,cSsg|]}t|qS)str).0edger r V/home/ubuntu/webapp/venv/lib/python3.10/site-packages/redis/commands/graph/commands.py z(GraphCommands.commit..)lennodesedgesitemsr joinquery)selfr_noder r rcommits   zGraphCommands.commitNFc Cs|}|||}|rt}n|rtnt}||j|dg}t|tr(|d|gn|dur0tdz |j |} t || |WSt ya} zdt | vr[|r[|j |||ddWYd} ~ S| d} ~ wty} z| j|_|| ||||WYd} ~ Sd} ~ ww)a Executes a query against the graph. For more information see `GRAPH.QUERY `_. # noqa Args: q : str The query. params : dict Query parameters. timeout : int Maximum runtime for read queries in milliseconds. read_only : bool Executes a readonly query if set to True. profile : bool Return details on results produced by and time spent in each operation. --compacttimeoutN+Timeout argument must be a positive integerunknown commandF read_only)_build_params_header PROFILE_CMD RO_QUERY_CMD QUERY_CMDname isinstanceintextend Exceptionexecute_commandrrr rrversion_refresh_schema rqparamsrr"profilercmdcommandresponseer r rr(s0   zGraphCommands.querycCsd}|t|7}||S)z Merge pattern. zMERGE )r r)rpatternrr r rmergebs  zGraphCommands.mergecCs||t|jS)z| Deletes graph. For more information see `DELETE `_. # noqa ) _clear_schemar, DELETE_CMDr'rr r rdeletekszGraphCommands.deletecCs|i|_g|_dSzT Commit the graph and reset the edges and the nodes to zero length. Nrrrr;r r rflushts zGraphCommands.flushcKstd)zInternal only. Not supported.zVGRAPH.BULK is internal only. Use https://github.com/redisgraph/redisgraph-bulk-loader.)NotImplementedError)rkwargsr r rbulk|szGraphCommands.bulkcCs|j|ddS)ar Execute a query and produce an execution plan augmented with metrics for each operation's execution. Return a string representation of a query execution plan, with details on results produced by and time spent in each operation. For more information see `GRAPH.PROFILE `_. # noqa T)r2)r)rrr r rr2szGraphCommands.profilecCs|t|jS)a Get a list containing up to 10 of the slowest queries issued against the given graph ID. For more information see `GRAPH.SLOWLOG `_. # noqa 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. )r, SLOWLOG_CMDr'r;r r rslowlogs zGraphCommands.slowlogcCsB|rdnd|g}|dur|r||ntd|jtg|RS)a Retrieve or update a RedisGraph configuration. For more information see `https://redis.io/commands/graph.config-get/>`_. # noqa Args: name : str The name of the configuration value : The value we want to set (can be used only when `set` is on) set : bool Turn on to set a configuration. Default behavior is get. SETGETNz3``value`` can be provided only when ``set`` is True)appendrr, CONFIG_CMD)rr'valuesetr1r r rconfigs zGraphCommands.configcCs |tS)z Lists all graph keys in the keyspace. For more information see `GRAPH.LIST `_. # noqa )r,LIST_CMDr;r r r list_keyss zGraphCommands.list_keyscCsD|||}|t|j|}t|dtrdd|D}d|S) Get the execution plan for given query, GRAPH.EXPLAIN returns an array of operations. Args: query: the query that will be executed params: query parameters rcSg|]}|qSr decoder br r rrrz0GraphCommands.execution_plan.. r#r, EXPLAIN_CMDr'r(bytesrrrr1planr r rexecution_plans   zGraphCommands.execution_plancCs&|||}|t|j|}t|S)a8 Get the execution plan for given query, GRAPH.EXPLAIN returns ExecutionPlan object. For more information see `GRAPH.EXPLAIN `_. # noqa Args: query: the query that will be executed params: query parameters r#r,rVr'rrXr r rexplains zGraphCommands.explainNNFF)NFN)__name__ __module__ __qualname____doc__rrr8r<r?rBr2rDrKrMrZr\r r r rr s :    r c@s2eZdZd ddZd ddZd ddZd d ZdS) AsyncGraphCommandsNFc s$|}|||}|rt}n|rtnt}||j|dg}t|tr)|d|gn|dur1tdz|j |IdH} t || |IdHWSt ym} zdt | vrg|rg|j|||ddIdHWYd} ~ S| d} ~ wty} z| j|_||||||IdHWYd} ~ Sd} ~ ww)a1 Executes a query against the graph. For more information see `GRAPH.QUERY `_. # noqa Args: q : str The query. params : dict Query parameters. timeout : int Maximum runtime for read queries in milliseconds. read_only : bool Executes a readonly query if set to True. profile : bool Return details on results produced by and time spent in each operation. rrNrr Fr!)r#r$r%r&r'r(r)r*r+r,r initializerr rrr-r.r/r r rrs2  $"zAsyncGraphCommands.querycsL|||}|t|j|IdH}t|dtr!dd|D}d|S)rNNrcSrOr rPrRr r rr"rz5AsyncGraphCommands.execution_plan..rTrUrXr r rrZs   z!AsyncGraphCommands.execution_plancs.|||}|t|j|IdH}t|S)z Get the execution plan for given query, GRAPH.EXPLAIN returns ExecutionPlan object. Args: query: the query that will be executed params: query parameters Nr[rXr r rr\%s zAsyncGraphCommands.explaincs |IdHi|_g|_dSr=r>r;r r rr?3s zAsyncGraphCommands.flushr]r^)r_r`rarrZr\r?r r r rrcs   :  rcN)redisrredis.exceptionsr exceptionsrrZr query_resultrrr$r%r&r:rCrHrLrVr rcr r r rs   I