+hB ddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZd Zd Zd Zd ZdZdZdZdZdZdZdZdZeeeeeeeeeeeeg ZGddZGddZGddZGddeZy)N) OrderedDict) strtobool) ResponseError)Edge)VersionMismatchException)Node)Pathz Labels addedzLabels removedz Nodes createdz Nodes deletedzRelationships deletedzProperties setzProperties removedzRelationships createdzIndices createdzIndices deletedzCached executionzinternal execution timeceZdZdZdZdZdZy)ResultSetColumnTypesrrN)__name__ __module__ __qualname__COLUMN_UNKNOWN COLUMN_SCALAR COLUMN_NODECOLUMN_RELATIONT/var/www/html/venv/lib/python3.12/site-packages/redis/commands/graph/query_result.pyr r *sNMKOrr c<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy )ResultSetScalarTypesrrr r N)rrr VALUE_UNKNOWN VALUE_NULL VALUE_STRING VALUE_INTEGER VALUE_BOOLEAN VALUE_DOUBLE VALUE_ARRAY VALUE_EDGE VALUE_NODE VALUE_PATH VALUE_MAP VALUE_POINTrrrrr1s>MJLMMLKJJJIKrrceZdZd(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_resultsselfr4responseprofiles r__init__zQueryResult.__init__Asz   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 mismatchrr2N) isinstancerstrr)r=r>errorversions rr7zQueryResult._check_for_errors^s_ hqk= 1QKE5z//"1+09K hrlM 22,  3rc|j||_t|jdk(ry|j||_yzL Parse the query execution result returned from the server. rN parse_headerr5r8 parse_recordsr6r=raw_result_sets rr;zQueryResult.parse_resultsns>''7  t{{ q ,,^<rci|_t|D])\}}t|ts|j ||<+t D]&}|j ||}|||j|<(y)z@ Parse the statistics returned in the response. N) statistics enumeraterBbytesdecodeSTATS _get_value)r=raw_statisticsidxstatsvs rr9zQueryResult.parse_statisticszss#>2 4IC$&&*kkms# 4 'A>2A}%&" 'rc|d}|S)z1 Parse the header of the result. rr)r=rLr5s rrIzQueryResult.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)rOparse_record_typesr5)r=rLrowrUcellrecordss rrJzQueryResult.parse_recordssy&a(   "+3 C='' C(8(;xs rr:zQueryResult.parse_profile5s/@HI11Q.446IIs)7c2t|jdk(SNr)r8r6r=s ris_emptyzQueryResult.is_empty8s4??#q((rc||D]7}||vst|jddjddcSy)Nz: r r)rsplit)rhrNrVs rrSzQueryResult._get_value;sH @Dt|TZZ-a066s;A>?? @rc@||jvr|j|SdSr)rN)r=rVs r _get_statzQueryResult._get_statCs!(,(?tt$FQFrc,|jtS)z/Returns the number of labels added in the query)r LABELS_ADDEDrs r labels_addedzQueryResult.labels_addedFs~~l++rc,|jtS)z1Returns the number of labels removed in the query)rLABELS_REMOVEDrs rlabels_removedzQueryResult.labels_removedK~~n--rc,|jtS)z0Returns the number of nodes created in the query)r NODES_CREATEDrs r nodes_createdzQueryResult.nodes_createdP~~m,,rc,|jtS)z0Returns the number of nodes deleted in the query)r NODES_DELETEDrs r nodes_deletedzQueryResult.nodes_deletedUrrc,|jtS)z1Returns the number of properties set in the query)rPROPERTIES_SETrs rproperties_setzQueryResult.properties_setZrrc,|jtS)z5Returns the number of properties removed in the query)rPROPERTIES_REMOVEDrs rproperties_removedzQueryResult.properties_removed_s~~011rc,|jtS)z8Returns the number of relationships created in the query)rRELATIONSHIPS_CREATEDrs rrelationships_createdz!QueryResult.relationships_createdd~~344rc,|jtS)z8Returns the number of relationships deleted in the query)rRELATIONSHIPS_DELETEDrs rrelationships_deletedz!QueryResult.relationships_deletedirrc,|jtS)z2Returns the number of indices created in the query)rINDICES_CREATEDrs rindices_createdzQueryResult.indices_createdn~~o..rc,|jtS)z2Returns the number of indices deleted in the query)rINDICES_DELETEDrs rindices_deletedzQueryResult.indices_deletedsrrc2|jtdk(S)z:Returns whether or not the query execution plan was cachedr)rCACHED_EXECUTIONrs rcached_executionzQueryResult.cached_executionxs~~./144rc,|jtS)z.Returns the server execution time of the query)rINTERNAL_EXECUTION_TIMErs r run_time_mszQueryResult.run_time_ms}s~~566rcvtj|jtj|jtj |j tj|jtj|jtj|jtj|jtj|j tj"|j$tj&|j(tj*|j,tj.|j0i SN)rr$rr%rnr&rr'rr(rr)rr+rxr*rr,rr-rr.rr#rrs rrzQueryResult.parse_scalar_typess ! + +T__ - -t/@/@ . .0B0B . .0B0B - -t/@/@ , ,d.>.> + +T__ + +T__ + +T__ * *DNN , ,d.>.> . .0B0B  rctj|jtj|jtj |j tj|jiSr) r rrdrrxrrrrrs rr\zQueryResult.parse_record_typessN ! . .0A0A , ,doo 0 0$// / /1C1C   rNF)+rrrr@r7r;r9rIrJrkrnrxrrrrrrrrrrrdr:r staticmethodrSrpropertyrrrrrrrrrrrrrr\rrrr0r0@s): ='    J" &"    J)G,,..----..225555////5577    rr0cTeZdZdZdZddZdZdZdZdZ 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() Nrrs rr@zAsyncQueryResult.__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" rrr2Nr3r<s 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 rp)r8rtr4rurkrsr )r=r^rvrwrgrqs rrxzAsyncQueryResult.parse_nodes tAwq.AB4HHHF NN6 "  # Is(A20A, A* A,A2*A,,A2cK|j||_t|jdk(ry|j|d{|_y7 wrGrHrKs rr;zAsyncQueryResult.parse_resultssH''7  t{{ q  $ 2 2> BBBsAAA AcKi}|D]I}|jj|dd{}|j|ddd{}|||<K|S7)7w)rarNrrbres rrkz(AsyncQueryResult.parse_entity_propertiessk   /D"jj55d1g>>I#00ab::J$.Jy ! /  ?:s!)AAAAAAc Kt|d}|jj|dd{}t|d}t|d}|j|dd{}t |||||S7L7w)rzrrNr rrr{r}rs rrzAsyncQueryResult.parse_edge sd1g,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_map(sq 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_array8sG>C3u:=NO))%(333OO 4Os%AAA AAAANr)rrr__doc__r@rrxrdrJr;rkrrrrrrrrrs@  <J"   C  &" rr) r collectionsrdistutils.utilrredisredger exceptionsrnoder pathr rrrrrrrrrrrrrRr rr0rrrrrs #$ 0 !  /!)/##%3    Z Z z `{`r