U cc@sNddlmZddlmZddlmZddlmZddlm Z GdddZ d S) ) DataError) ResponseError)VersionMismatchException) ExecutionPlan) QueryResultc@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).0edger r W/var/www/html/project/venv/lib/python3.8/site-packages/redis/commands/graph/commands.py sz(GraphCommands.commit..)lennodesedgesitemsr joinquery)selfr_noder r rcommit s  zGraphCommands.commitNFc Cs(|}|dk r|||}|r$d}n |r,dnd}||j|dg}|r`t|tsTtd|d|g7}z|j|} t|| |WStk r} zFdt| krt d d t| kr|r|j |||d d WY S| W5d} ~ XYnHt k 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. Nz GRAPH.PROFILEzGRAPH.RO_QUERYz GRAPH.QUERYz --compactz+Timeout argument must be a positive integertimeoutzwrong number of argumentsz>Note: RedisGraph Python requires server version 2.2.8 or abovezunknown commandF) read_only)_build_params_headername isinstanceint Exceptionexecute_commandrrr printrrversionZ_refresh_schema) rqparamsrrprofilercmdcommandresponseer r rrs4     zGraphCommands.querycCsd}|t|7}||S)z Merge pattern. zMERGE )r r)rpatternrr r rmerge^s zGraphCommands.mergecCs||d|jS)z| Deletes graph. For more information see `DELETE `_. # noqa z GRAPH.DELETE)Z _clear_schemar"rrr r rdeletegszGraphCommands.deletecCs|i|_g|_dS)zT Commit the graph and reset the edges and the nodes to zero length. N)rrrr.r r rflushpszGraphCommands.flushcKs tddS)zInternal only. Not supported.zVGRAPH.BULK is internal only. Use https://github.com/redisgraph/redisgraph-bulk-loader.N)NotImplementedError)rkwargsr r rbulkxszGraphCommands.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)r')r)rrr r rr'szGraphCommands.profilecCs|d|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. z GRAPH.SLOWLOG)r"rr.r r rslowlogs zGraphCommands.slowlogcCs>|rdnd|g}|dk r0|r(||ntd|jd|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 GRAPH.CONFIG)r6)appendrr")rrvaluesetr&r r rconfigs zGraphCommands.configcCs |dS)z Lists all graph keys in the keyspace. For more information see `GRAPH.LIST `_. # noqa z GRAPH.LIST)r"r.r r r list_keysszGraphCommands.list_keyscCsL|dk r|||}|d|j|}t|dtrBdd|D}d|S)z 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 N GRAPH.EXPLAINrcSsg|] }|qSr )decode)r br r rrsz0GraphCommands.execution_plan.. )rr"rrbytesrrrr&Zplanr r rexecution_plans zGraphCommands.execution_plancCs.|dk r|||}|d|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 Nr<)rr"rrrAr r rexplains zGraphCommands.explain)NNFF)NF)N)N)__name__ __module__ __qualname____doc__rrr-r/r0r3r'r4r:r;rBrCr r r rr s ?     rN) ZredisrZredis.exceptionsr exceptionsrrBrZ query_resultrrr r r rs