U d(@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|jdkr t|jdkr dSd}|jD]\}}|t|d7}q.|ddd|jD7}|ddkrz|dd}||S)z& Create entire graph. rNzCREATE ,cSsg|] }t|qS)str).0Zedger r A/tmp/pip-unpacked-wheel-f5h5_hbx/redis/commands/graph/commands.py sz(GraphCommands.commit..)lennodesedgesitemsr joinquery)selfr_noder r rcommits  zGraphCommands.commitNFc Cs|}|||}|rt}n |r$tnt}||j|dg}t|trP|d|gn|dk r`tdz|j |} t || |WSt k r} z2dt | kr|r|j |||ddWY S| W5d} ~ XYnHtk r} z(| j|_|| ||||WYSd} ~ XYnXdS)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(s*   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 )Z _clear_schemar* DELETE_CMDr%rr r rdeletekszGraphCommands.deletecCs|i|_g|_dSzT Commit the graph and reset the edges and the nodes to zero length. Nrrrr8r r rflushtszGraphCommands.flushcKs tddS)zInternal only. Not supported.zVGRAPH.BULK is internal only. Use https://github.com/redisgraph/redisgraph-bulk-loader.N)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)r0)r)rrr r rr0szGraphCommands.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%r8r r rslowlogs zGraphCommands.slowlogcCs@|rdnd|g}|dk r0|r(||ntd|jtf|S)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. ZSETGETNz3``value`` can be provided only when ``set`` is True)appendrr* CONFIG_CMD)rr%valuesetr/r r rconfigs zGraphCommands.configcCs |tS)z Lists all graph keys in the keyspace. For more information see `GRAPH.LIST `_. # noqa )r*LIST_CMDr8r r r list_keysszGraphCommands.list_keyscCsD|||}|t|j|}t|dtr:dd|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 rcSsg|] }|qSr decoder br r rrsz0GraphCommands.execution_plan.. r!r* EXPLAIN_CMDr%r&bytesrrrr/Zplanr 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*rQr%rrSr r rexplains zGraphCommands.explain)NNFF)NF)N)N)__name__ __module__ __qualname____doc__rrr6r9r<r?r0rArGrIrTrVr r r rr s :     r c@s2eZdZd ddZd ddZd ddZd d ZdS)AsyncGraphCommandsNFc s0|}|||}|rt}n |r$tnt}||j|dg}t|trP|d|gn|dk r`tdz(|j |IdH} t || |IdHWSt k r} z8dt | kr|r|j|||ddIdHWY S| W5d} ~ XYnNtk r*} z.| j|_||||||IdHWYSd} ~ XYnXdS)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. rrNrrFr)r!r"r#r$r%r&r'r(r)r*rZ initializerr rrr+r,r-r r rrs*   zAsyncGraphCommands.querycsJ|||}|t|j|IdH}t|dtr@dd|D}d|S)rJNrcSsg|] }|qSr rKrMr r rr"sz5AsyncGraphCommands.execution_plan..rOrPrSr r rrTs 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 NrUrSr r rrV%s zAsyncGraphCommands.explaincs|IdHi|_g|_dSr:r;r8r r rr<3szAsyncGraphCommands.flush)NNFF)N)N)rWrXrYrrTrVr<r r r rr[s :  r[N)ZredisrZredis.exceptionsr exceptionsrrTrZ query_resultrrr"r#r$r7r@rDrHrQr r[r r r rs    I