jy-LddlmZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZmZmZejd Zejee fZ!Gdde"eZ#GddeZ$Gdde$Z%Gdde$Z&dS)) annotationsN)ABCMetaabstractmethod)MappingSequenceSet)reduce)xor)BrokenHydrationObject) iter_items) deprecated)BrokenRecordError)NodePath Relationship_TceZdZUdZded<d*dZdZfdZd+fd ZeZ d,dZ d,dZ dZ d-fd Z d.fd ZeddZd/d0dZd1dZ d2d3d!Zd4d#Zd5d&Zd'Zd6d)ZxZS)7Recordz A :class:`.Record` is an immutable ordered collection of key-value pairs. It is generally closer to a :py:class:`namedtuple` than to a :py:class:`OrderedDict` in as much as iteration of the collection will yield values rather than keys. z t.Tuple[str] _Record__keyscg}g}t|D]/\}}||||0t||}t||_|SN)r appendtuple__new__r)clsiterablekeysvalueskeyvalueinsts U/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/neo4j/_data.pyrzRecord.__new__8ss$X.. ! !JC KK    MM% }}S&))Dkk  cDtd|d|j|dS)NzRecord contains broken data at z ('z'))rr)selfindexs r$_broken_record_errorzRecord._broken_record_errorBs1 Ne N N E8J N N N   r%ct|}t|tr|||j|Sr)super __getitem__ isinstancer r)error)r'r(r" __class__s r$_super_getitem_singlezRecord._super_getitem_singleGsK##E** e2 3 3 D++E22 C r%returnstrc d|jjdddt|jt DdS)N< c3*K|]\}}|d|VdS)=Nr).0fieldr"s r$ z"Record.__repr__..PsMOO%!&uu-OOOOOOr%>)r/__name__joinziprr+__iter__)r'r/s r$__repr__zRecord.__repr__Msy N # # # HHOO),T[%'':J:J:L:L)M)MOOO O O O O  r%otherobjectboolc$t|t}t|t}|ra|r_tjtj|}t |t |kot |t |kS|r?tjtj|}t |t |kS|r?tjtj|}t |t |kSdS)z In order to be flexible regarding comparison, the equality rules for a record permit comparison with any other Sequence or Mapping. :param other: :returns: F)r-rrtcastlistdict)r'rAcompare_as_sequencecompare_as_mappings r$__eq__z Record.__eq__Vs)99'w77  #5 F19e,,E::e,JdtE{{1J J  F1:u--E::e, ,  F19e,,E::e, ,5r%c.|| Sr)rK)r'rAs r$__ne__z Record.__ne__ks;;u%%%%r%cvtttt|Sr)r xor_operatormaphashitemsr's r$__hash__zRecord.__hash__ns$lCdjjll$;$;<<r(r0 IndexError)r'r!rr r(r/s r$r,zRecord.__getitem__ws c5 ! ! 5;s#DWW((--F>>#dF"3"344 4 5JJsOOE--e44 4   44 s)B B! B!z)r'startstopr!rr s r$ __getslice__zRecord.__getslice__sIE4  {3tS!~~c$//000r%Ndefaultt.Optional[object]c |jt|}n#t$r|cYSwxYwd|cxkrt |krnn||S|S)z Obtain a value from the record by key, returning a default value if the key does not exist. :param key: a key :param default: default value :returns: a value r)rr(r2 ValueErrorlenr0r'r!rcr(s r$getz Record.gets K%%c#hh//EE   NNN   ! ! ! !D ! ! ! ! !--e44 4Ns '* 99_KintcTt|tr3d|cxkrt|jkrnn|St |t|t r= |j|S#t$r}t||d}~wwxYwt|)za Return the index of the given item. :param key: a key :returns: index rN) r-rkrgrr^r2r(rfKeyError TypeError)r'r!excs r$r(z Record.indexs c3   !C****#dk******* S// ! S ! ! ! -{((--- - - -smm, -C.. sA99 BBBrcp ||}||S#ttf$r|cYSwxYw)a@ Obtain a single value from the record by index or key. If no index or key is specified, the first value is returned. If the specified item does not exist, the default value is returned. :param key: an index or key :param default: default value :returns: a single value )r(r^rmrhs r$r"z Record.valuesM JJsOOE; H%   NNN s 55 t.List[str]c*t|jS)zM Return the keys of the record. :returns: list of key names )rGrrSs r$rz Record.keyss DK   r%r t.List[t.Any]c|r_g}|D]X} ||}|||4#t$r|dYUwxYw|St|S)a Return the values of the record, optionally filtering to include only certain values by index or key. :param keys: indexes or keys of the items to include; if none are provided, all values will be included :returns: list of values N)r(rrmrGr'rdr!rXs r$r z Record.valuess  !A & && 3AHHT!W%%%% ###HHTNNNNN#HDzzs;AAcN|rng}|D]g} |}|j||fA#t$r||dfYdwxYw|St fdt t DS)zu Return the fields of the record as a list of key and value tuples :returns: a list of value tuples Nc3\K|]&}j||fV'dSr)rr0)r8rXr's r$r:zRecord.items..sP//[^T%?%?%B%BC//////r%)r(rrrmrGrangergrus` r$rRz Record.itemss  A 8 88 3AHHdk!nd1g67777 ***HHc4[)))))*H////"3t99--///// /sA  !A-,A-t.Dict[str, t.Any]cntt|j|S)a] Return the keys and values of this record as a dictionary, optionally including only certain values by index or key. Keys provided in the items that are not in the record will be inserted with a value of :data:`None`; indexes provided that are out of bounds will trigger an :exc:`IndexError`. :param keys: indexes or keys of the items to include; if none are provided, all values will be included :returns: dictionary of values, keyed by field name :raises: :exc:`IndexError` if an out-of-bounds index is specified )RecordExporter transformrHrR)r'rs r$dataz Record.datas/))$ztz4/@*A*ABBBr%)r)r1r2)rArBr1rC)r1rU)r!rZr1r[r)r!r2rcrdr1r[)r!rjr1rk)rN)r!rjrcrdr1r[)r1rq)rrjr1rs)rrjr1rz)r< __module__ __qualname____doc____annotations__rr)r0r@__str__rKrMrTr?r,rrbrir(r"rr rRr~ __classcell__)r/s@r$rr/s           G*&&&&=== 5 5 5 5 5 5ZNOO11PO1 $!!!!(:>&!!!!*///$CCCCCCCCr%rc(eZdZdZedZdS)DataTransformerzO Abstract base class for transforming data from one form into another. cdS)zs Transform a value, or collection of values. :param x: input value :returns: output value Nr)r'xs r$r}zDataTransformer.transform sr%N)r<rrrrr}rr%r$rrs9  ^   r%r) metaclassceZdZdZdZdS)r|z@ Transformer class used by the :meth:`.Record.data` method. ct|tr"t|St|trYt|j|jjt|jfSt|tr|jg}t|j D]m\}}| |jj| |j |dzn|St|tr|St|tr-t!|}|t#j|St|t$r-t!|}|t#j|St|t&r8t!|}|fd|DS|S)Nr c3LK|]\}}||fVdSr)r})r8krYr's r$r:z+RecordExporter.transform..-s8DD$!Q4>>!,,-DDDDDDr%)r-rr}rHr start_noder/r<end_noderrW relationshipsrnodesr2rtyperPrrrR)r'rpathrX relationshiptyps` r$r}zRecordExporter.transforms a   >>$q''** * < ( ( NN4 #5#566K(NN4 #3#3446 64  NN1<001D#,Q_#=#= < << DNN<+A+JKKLLL DNN171q5>::;;;;K 3   H 8 $ $ q''C3s4>1--.. . 3   q''C3s4>1--.. . 7 # # q''C3DDDD!''))DDDDD DHr%N)r<rrrr}rr%r$r|r|s-r%r|ceZdZdZdZdZdS)RecordTableRowExporterz;Transformer class used by the :meth:`.Result.to_df` method.ct|tsJt|}|fd|DS)Nc3K|]]\}}||ddddD]}|V^dS)\\\.\.prefixN _transformreplacerR)r8rrYitemr's r$r:z3RecordTableRowExporter.transform..8sq!#4!8!8!@!@e!L!L /  UWW r%)r-rrrR)r'rrs` r$r}z RecordTableRowExporter.transform5sf!W%%%%%1ggs wwyy r%cHt|trKdz|jdz|ji}|fd|D|St|t rndz|jdz|jjdz|jjdz|j j i}|fd|D|St|tst|tr|iSt|tr)tfd t|DSt|t r9t#|}|fd |DS|iS) Nz%s().element_idz %s().labelsc30K|]\}}d||fVdS)z().prop.Nrr8rrYrs r$r:z4RecordTableRowExporter._transform..D7OOTQ&&&!!4a8OOOOOOr%z%s->.element_idz%s->.start.element_idz%s->.end.element_idz %s->.typec30K|]\}}d||fVdS)z->.prop.Nrrs r$r:z4RecordTableRowExporter._transform..Mrr%c3K|]:\}}|d|fzD]}|V;dS)z%s[].%irN)rrR)r8rXrYrrr's r$r:z4RecordTableRowExporter._transform..Rs|Aq OOi61+5,%'' r%c 3K|]b\}}|d|ddddD]}|VcdS)z{}.rrrrrNr)r8rrYrrr's r$r:z4RecordTableRowExporter._transform..[sAq OO666199T63J3J5s a  % !F*AL&C JJOOOOQWWYYOOO O O OJ < ( ( !F*AL'&0!,2I%. 0Ef$ak&: C JJOOOOQWWYYOOO O O OJ 4  Jq#$6$6 A;  8 $ $ %aLL 7 # # q''C3GGII A; r%N)r<rrrr}rrr%r$rr2s8EE&&&&&r%r)' __future__rtypingrEabcrrcollections.abcrrr functoolsr operatorr rO_codec.hydrationr _confr _metar exceptionsrgraphrrrTypeVarrUnionrkr2rjrrrr|rrr%r$rs "#""""" ((((((333333))))))QYt__WS#XRCRCRCRCRCUGRCRCRCj           _@22222_22222r%