ipD ddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z d Z d Zd Zd Zd ZdZdZdZdZdZdZdZe eeeeeeeeeeeg ZdedefdZGddZGddZGddZGddeZ y) N) OrderedDict) ResponseError)Edge)VersionMismatchException)Node)Pathz Labels addedzLabels removedz Nodes createdz Nodes deletedzRelationships deletedzProperties setzProperties removedzRelationships createdzIndices createdzIndices deletedzCached executionzinternal execution timevaluereturncR|j}|dvry|dvrytd|)z Convert a string representation of truth to True or False. True values are 'y', 'yes', 't', 'true', 'on', and '1'; false values are 'n', 'no', 'f', 'false', 'off', and '0'. Raises ValueError if 'value' is anything else. )yyesttrueon1T)nnoffalseoff0Fzinvalid truth value )lower ValueError)r s U/root/searxng-venv/lib/python3.12/site-packages/valkey/commands/graph/query_result.py strtoboolr)s: KKME 44 7 7/w788ceZdZdZdZdZdZy)ResultSetColumnTypesrrN)__name__ __module__ __qualname__COLUMN_UNKNOWN COLUMN_SCALAR COLUMN_NODECOLUMN_RELATIONrrrr9sNMKOrrc<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy )ResultSetScalarTypesrrr r! N)r"r#r$ VALUE_UNKNOWN VALUE_NULL VALUE_STRING VALUE_INTEGER VALUE_BOOLEAN VALUE_DOUBLE VALUE_ARRAY VALUE_EDGE VALUE_NODE VALUE_PATH VALUE_MAP VALUE_POINTr)rrr+r+@s>MJLMMLKJJJIKrr+ceZdZd(dZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZedZdZedZedZedZedZ edZ!edZ"edZ#ed Z$ed!Z%ed"Z&ed#Z'ed$Z(ed%Z)ed&Z*y')) QueryResultc||_g|_g|_|j|t |dk(r|j |dy|r|j |y|j |d|j|y)a7 A class that represents a result of the query operation. Args: graph: The graph on which the query was executed. response: The response from the server. profile: A boolean indicating if the query command was "GRAPH.PROFILE" rrNgraphheader result_set_check_for_errorslenparse_statistics parse_profile parse_resultsselfrEresponseprofiles r__init__zQueryResult.__init__Psz   x( x=A   ! !(1+ .    x (  ! !(2, /   x (rct|dtr%|d}t|dk(r|d}t|}|t|dtr|dy)z: Check if the response contains an error. rzversion mismatchrrCN) isinstancerstrr)rNrOerrorversions rrHzQueryResult._check_for_errorsms_ hqk= 1QKE5z//"1+09K hrlM 22,  3rc|j||_t|jdk(ry|j||_yzL Parse the query execution result returned from the server. rN parse_headerrFrI parse_recordsrGrNraw_result_sets rrLzQueryResult.parse_results}s>''7  t{{ q ,,^<rci|_t|D])\}}t|ts|j ||<+t D]&}|j ||}|||j|<(y)z@ Parse the statistics returned in the response. N) statistics enumeraterSbytesdecodeSTATS _get_value)rNraw_statisticsidxstatsvs rrJzQueryResult.parse_statisticsss#>2 4IC$&&*kkms# 4 'A>2A}%&" 'rc|d}|S)z1 Parse the header of the result. rr))rNr]rFs rrZzQueryResult.parse_headers  " rc |dDcgc]C}t|Dcgc]*\}}|j|j|d|,c}}E}}}}|Scc}}wcc}}}w)F Parses the result set and returns a list of records. rr)r`parse_record_typesrF)rNr]rowrfcellrecordss rr[zQueryResult.parse_recordssy&a(   "+3 C='' C(8(;?? @rc@||jvr|j|SdSr)r_)rNrgs r _get_statzQueryResult._get_statRs!(,(?tt$FQFrc,|jtS)z/Returns the number of labels added in the query)r LABELS_ADDEDrs r labels_addedzQueryResult.labels_addedUs~~l++rc,|jtS)z1Returns the number of labels removed in the query)rLABELS_REMOVEDrs rlabels_removedzQueryResult.labels_removedZ~~n--rc,|jtS)z0Returns the number of nodes created in the query)r NODES_CREATEDrs r nodes_createdzQueryResult.nodes_created_~~m,,rc,|jtS)z0Returns the number of nodes deleted in the query)r NODES_DELETEDrs r nodes_deletedzQueryResult.nodes_deleteddrrc,|jtS)z1Returns the number of properties set in the query)rPROPERTIES_SETrs rproperties_setzQueryResult.properties_setirrc,|jtS)z5Returns the number of properties removed in the query)rPROPERTIES_REMOVEDrs rproperties_removedzQueryResult.properties_removedns~~011rc,|jtS)z8Returns the number of relationships created in the query)rRELATIONSHIPS_CREATEDrs rrelationships_createdz!QueryResult.relationships_createds~~344rc,|jtS)z8Returns the number of relationships deleted in the query)rRELATIONSHIPS_DELETEDrs rrelationships_deletedz!QueryResult.relationships_deletedxrrc,|jtS)z2Returns the number of indices created in the query)rINDICES_CREATEDrs rindices_createdzQueryResult.indices_created}~~o..rc,|jtS)z2Returns the number of indices deleted in the query)rINDICES_DELETEDrs rindices_deletedzQueryResult.indices_deletedrrc2|jtdk(S)z:Returns whether or not the query execution plan was cachedr)rCACHED_EXECUTIONrs rcached_executionzQueryResult.cached_executions~~./144rc,|jtS)z.Returns the server execution time of the query)rINTERNAL_EXECUTION_TIMErs r run_time_mszQueryResult.run_time_mss~~566rcvtj|jtj|jtj |j tj|jtj|jtj|jtj|jtj|j tj"|j$tj&|j(tj*|j,tj.|j0i SN)r+r5rr6rr7rr8rr9rr:rr<rr;rr=rr>rr?rr4rrs rrzQueryResult.parse_scalar_typess ! + +T__ - -t/@/@ . .0B0B . .0B0B - -t/@/@ , ,d.>.> + +T__ + +T__ + +T__ * *DNN , ,d.>.> . .0B0B  rctj|jtj|jtj |j tj|jiSr) rr&rur'rr(rr%rrs rrmzQueryResult.parse_record_typessN ! . .0A0A , ,doo 0 0$// / /1C1C   rNF)+r"r#r$rQrHrLrJrZr[r|rrrrrrrrrrrrrurKr staticmethodrdrpropertyrrrrrrrrrrrrrrmr)rrrArAOs): ='    J" &"    J)G,,..----..225555////5577    rrAcTeZdZdZdZddZdZdZdZdZ dZ d Z d Z d Z d Zy )AsyncQueryResultzp Async version for the QueryResult class - a class that represents a result of the query operation. cy)zC To init the class you must call self.initialize() Nr)rs rrQzAsyncQueryResult.__init__s rc*K||_g|_g|_|j|t |dk(r|j |d|S|r|j ||S|j |d|j|d{|S7w)a Initializes the class. Args: graph: The graph on which the query was executed. response: The response from the server. profile: A boolean indicating if the query command was "GRAPH.PROFILE" rrrCNrDrMs r initializezAsyncQueryResult.initializes   x( x=A   ! !(1+ .    x (    ! !(2, /$$X. . .  /sBB B Bc$Kd}t|ddkDr>g}|dD]4}|j|jj|d{6|j |dd{}t |d}t |||S7B7"w)z. Parses a node from the cell. Nrrr r)rIrrErr|rr)rNrorrrxrs rrzAsyncQueryResult.parse_nodes tAwq.AB4HHHF NN6 "  # Is(A20A, A* A,A2*A,,A2cK|j||_t|jdk(ry|j|d{|_y7 wrXrYr\s rrLzAsyncQueryResult.parse_resultssH''7  t{{ q  $ 2 2> BBBsAAA AcKi}|D]I}|jj|dd{}|j|ddd{}|||<K|S7)7w)rrrNrrsrvs rr|z(AsyncQueryResult.parse_entity_propertiessk   /D"jj55d1g>>I#00ab::J$.Jy ! /  ?:s!)AAAAAAc Kt|d}|jj|dd{}t|d}t|d}|j|dd{}t |||||S7L7w)rrrNr r!r,rrrs rrzAsyncQueryResult.parse_edgesd1g,00a99$q'l 47| 77Q@@  <Z  :As!0BA?7B*B+BBcK|j|dd{}|j|dd{}t||S7,7w)rrNrrrs rrzAsyncQueryResult.parse_path/sL''Q00''Q00E5!!10sA AA A A  A cKt}t|}td|dD]8}|j||}|j ||dzd{||<:|S7 w)z* Parse the cell to a map. rr rNrrs rrzAsyncQueryResult.parse_map7sq MI q)Q' :A##DG,C,,T!a%[99AcF ::sAA$A" A$cKtt|Dcgc]}|j||d{ }}|S7 cc}ww)z$ Parse array value. Nrrs rrzAsyncQueryResult.parse_arrayGsG>C3u:=NO))%(333OO 4Os%AAA AAAANr)r"r#r$__doc__rQrrrur[rLr|rrrrr)rrrrs@  <J"   C  &" rr)!r collectionsrvalkeyredger exceptionsrnoderpathr rrrrrrrrrrrrrcrTboolrrr+rArr)rrrs #!0 !  /!)/##%3  9S 9T 9   Z Z z `{`r