PinuUddlmZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZer(ddlmZmZmZmZmZmZmZddlmZmZddlmZddl m!Z!m"Z"d d gZ#d Z$d e%d <Gdd eZ&Gdd eZ'd S)) annotations) TYPE_CHECKINGAny CollectionDict GeneratorIteratorMappingOptionalSetTupleUnionoverload)Store) _coalesce)Graph _ContextType _ObjectType_PredicateType _SubjectType_TriplePatternType _TripleType)QueryUpdate)Result) IdentifierURIRef SimpleMemoryMemoryNNoneANYceZdZdZ d6d7fd Z d8d9dZ d:d;dZ d:dd?d Z d@d"Z dAd#Z dBd%Z dCd'Z dDfd2 ZdEfd5 ZxZS)Fra)A fast naive in memory implementation of a triple store. This triple store uses nested dictionaries to store triples. Each triple is stored in two such indices as follows `spo[s][p][o]` = 1 and `pos[p][o][s]` = 1. Authors: Michel Pelletier, Daniel Krech, Stefan Niederhauser N configuration Optional[str] identifierOptional[Identifier]ctt||||_i|_i|_i|_i|_i|_dSN) superr__init__r%_SimpleMemory__spo_SimpleMemory__pos_SimpleMemory__osp_SimpleMemory__namespace_SimpleMemory__prefixselfr#r% __class__s p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/rdflib/plugins/stores/memory.pyr*zSimpleMemory.__init__4sc lD!!**=999$    /1+- Ftriplercontextrquotedboolreturnr cz|\}}}|j} ||}n#ix}||<YnxYw ||} n#ix} ||<YnxYwd| |<|j} | |} n#ix} | |<YnxYw | |} n#ix} | |<YnxYwd| |<|j} | |}n#ix}| |<YnxYw ||}n#ix}||<YnxYwd||<dS)%Add a triple to the store of triples.N)r+r,r-)r1r5r6r7subject predicateobjectspopooposossospspps r3addzSimpleMemory.addNsF&,"Fj #WBB # " "BW #9 AA # " "A9 & j %YBB %"$ $BY 6 AA  A6 ' j "VBB "! !BV !7 AA ! A7 ) s@ #0 ; A A#A,, A7B BB(( B3triple_patternrOptional[_ContextType]ct||D]E\\}}}}|j|||=|j|||=|j|||=FdSr()listtriplesr+r,r-)r1rJr6r=r>r?cs r3removezSimpleMemory.removezs| 04DLL4P4P/Q/Q 7 7 + (Wi! 7#I.v6 9%f-g6 6"7+I66 7 7r4>Iterator[Tuple[_TripleType, Iterator[Optional[_ContextType]]]]c#K|\}}}|tkr|j}||vr ||}|tkrt||vrn|tkr)|||vr|||f|fVdSdS||D]}|||f|fVdSdS|D]l} |tkr'||| vr|| |f|fV34|| D]}|| |f|fVmdSdS|tkr|j} || vr| |} |tkr@|| vr:| |D]} | ||f|fVdSdS| D]<}| |D]} | ||f|fV;dSdS|tkrf|j} || vrW| |}|D]<} || D]} | | |f|fV;dSdS|j}|D]Y} || }|D]:} || D]}| | |f|fV;ZdSz)A generator over all the triples matchingN)r!r+_SimpleMemory__contextskeysr,r-)r1rJr6r=r>r?r@subjectDictionaryrBrHrCpredicateDictionaryrErFobjectDictionarys r3rNzSimpleMemory.tripless &4"F c>>*C#~~$'L!## $555!S==%):9)EEE'. 6&BDOODUDU&U U U U U U $%6y%A%F%F%H%HQQ'. 1&=t?P?P&P P P P PQQ.3355II!S==%):1)==='.6&:DOO%>%@%@II'.1ot7H7H&H H H H HIII #  *CC&))n#S==!444!4V!!>GGA#$i"3T__5F5F"FFFFFG GGs]]*C}}#&v; )..00@@A-a05577@@ !Vndoo.?.??????@}@@*CXXZZ ; ;$'F!*//11;;A.q16688;; !Qi):)::::::;; ; ;r4intcDd}|dD]}|dz }|S)NrNNNr<)rN)r1r6ir5s r3__len__zSimpleMemory.__len__s3 ll#566  F FAAr4Tprefixstr namespaceroverridec|j|}t|j||j|}|r*||j|=||j|=||j|<||j|<dSt|||jt||<t|||jt||<dSN)default)r.getrr/r1r^r`rabound_namespace bound_prefixs r3bindzSimpleMemory.binds*..v66 M  i ( ( M  o . .   '$\2*M/2'-DM) $'0D V $ $ $DMfDDDDM)OY?? @AJAAAD Y|V<< = = =r4Optional[URIRef]c8|j|dSr()r.rer1r^s r3r`zSimpleMemory.namespace##FD111r4c8|j|dSr()r/rer1r`s r3r^zSimpleMemory.prefix}  D111r4Iterator[Tuple[str, URIRef]]c#TK|jD] \}}||fV dSr()r.itemsr1r^r`s r3 namespaceszSimpleMemory.namespacesG!%!1!7!7!9!9 $ $ FI)# # # # # $ $r4#Generator[_ContextType, None, None]cddDS)Nc3K|]}|VdSr().0rOs r3 z*SimpleMemory.__contexts..sar4rzrz)r1s r3 __contextszSimpleMemory.__contextss2r4queryUnion[Query, str]initNsMapping[str, Any] initBindingsMapping[str, Identifier] queryGraphkwargsrrc Ptt|j||||fi|dSr()r)rr~r1r~rrrrr2s r3r~zSimpleMemory.querysF (lD!!' 6<  7=     r4updateUnion[Update, str]c Ptt|j||||fi|dSr()r)rrr1rrrrrr2s r3rzSimpleMemory.updatesF )lD!!( FL*  8>     r4NNr#r$r%r&Fr5rr6rr7r8r9r r(rJrr6rKr9r )rJrr6rKr9rQr6rKr9rYTr^r_r`rrar8r9r r^r_r9rjr`rr9r$r9rq)r9rw) r~rrrrrrr_rrr9r) rrrrrrrr_rrr9r )__name__ __module__ __qualname____doc__r*rIrPrNr]rir`r^rurTr~r __classcell__r2s@r3rr*sn(,+/.......< *****^+/77777+/>;>;>;>;>;@222222222$$$$                  r4ceZdZdZdZdZdZ dEdFfd Z dGdHdZ dIdJdZ dIdKdZ dLdMdZ dNd Z dOd!Z dPd#Z dIdQd&ZdIdRd(ZdSd+ZdSd,ZdTd.Z dGdUd1ZdVd3ZdWd4ZedXd5ZedYd6ZdZd7Zd[d8Zd\fdA Zd]fdD ZxZS)^rzAn in memory implementation of a triple store. Same as SimpleMemory above, but is Context-aware, Graph-aware, and Formula-aware Authors: Ashley Sommer TNr#r$r%r&c&tt||||_i|_i|_i|_i|_i|_i|_ i|_ dti|_ t|_ d|_dSr()r)rr*r% _Memory__spo _Memory__pos _Memory__osp_Memory__namespace_Memory__prefix_Memory__context_obj_map_Memory__tripleContextsset_Memory__contextTriples_Memory__all_contexts_Memory__defaultContextsr0s r3r*zMemory.__init__s fd$$]333$    /1+- 35NPHLcee}*-%%FJr4Fr5rr6rr7r8r9r ctj||||||j||\}}}|j} ||}n#t$r ix}||<YnwxYw ||} n#t$r ix} ||<YnwxYw | |} d} n#t $r d| |<d} YnwxYw||| ||| rdS|j} | |} n#t$r ix} | |<YnwxYw | |}n#t$r ix}| |<YnwxYwd||<|j} ||}n#t$r ix}||<YnwxYw ||}n#t$r ix}||<YnwxYwd||<dS)r;)r7NTr<F) rrIrr LookupErrorKeyError_Memory__add_triple_contextrr)r1r5r6r7r=r>object_r@rArB_ triple_existsrCrDrErFrGrHs r3rIz Memory.add9sJ  $7777     # #G , , ,&,#Gj #WBB # # # " "BW # #9 AA # # # " "A9  # "' A MM " " "AgJ!MMM " !!&-&III   Fj %YBB % % %"$ $BY % !7 AA ! ! ! A7  !' j #WBB # # # " "BW # !7 AA ! ! ! A7  !) s~A A A $A--BB BB$#B$ CC('C(,C55D D D""D65D6:EEErJrrKc||}|||D]\}}|\}}}||D]!} ||| kr ||| "||d} d| vr+|t | dkr||dt ||dkrD|j|||=|j|||=|j|||=|j|=|/||j vr&t |j |dkr|j |=|dkr,||j vr%|j s |j |dSdSdSdS)N)r6T skipQuotedr<rr[) _Memory__ctx_to_strrN_Memory__get_context_for_triple_Memory__remove_triple_contextlenrrrrrr graph_awarerP) r1rJr6req_ctxr5rOr=r>rctxctxss r3rPz Memory.removeus ##G,,ngFF 2 2IFA*0 'GY44V<< : :&7c>>,,VS999900D0IIDt||CIINN,,VT:::4008899Q>>Jw' 27;Jy)'27;Jw'0;)&1  4000D)'233q88%g. 0 0 04...$/   & &w / / / / / 1 0....r4XGenerator[Tuple[_TripleType, Generator[Optional[_ContextType], None, None]], None, None]c#j K||}|\}}}|L|J|H||jvrdS|j|D]}|||fVdS|d|b|`|} |j|||}|||r|||fVdSdS#t $rYdSwxYw||j} || vr|| |} ||| vr|D|| |vr8|||f}|||r|||fVdSdSdSt| |D]6} ||| f}|||r|||fV7dSdSt| D]} |@|| | vr5|| |f}|||r|||fVCDt| | D]6} || | f}|||r|||fV7dSdS||j } || vr| |}|f||vr`t||D]6}|||f}|||r|||fV7dSdSt|D]b} t|| D]6}||| f}|||r|||fV7adSdS||j }||vr||}t|D]b}t||D]6} || |f}|||r|||fV7adSdS|j} t| D]}| |} t| D]`} t| | D]6} || | f}|||r|||fV7adSrS) rrcopy_Memory__contextsr_Memory__triple_has_contextrrMrUrr)r1rJr6rr=r>rr5rr@rVrBrHrCrWrErFrXs r3rNzMemory.triplessr##G,,&4#G ?y0W_d333/8==?? 6 6doof5555555 6 6 Y%:w?R$F Jw' 27;,,VW==: $//&"9"9999999::     *C#~~$'L!( $555".&*;I*FFF*19g)F#'#<#>>*11g)>#'#<#*C*C*E*E%F%FJJ*11a#'#<#./s"EEGEEEEEEr4c3K|]}|VdSr(rz)r{rs r3r|z"Memory.contexts..6s??!A??????r4rz)rMrrrr)r1r5subjpredobjrs r3contextszMemory.contexts+s >V'999EE40C+D+DEEE E dC # 4 &s+A??6** * # # #"?r??? " " " #s-AA54A5rYcv||}||jvrdSt|j|S)Nr)rrr)r1r6rs r3r]zMemory.__len__8s>(( d+ + +14(-...r4graphrcv|jstj||dS|j|dSr()rr add_graphrrIr1rs r3rzMemory.add_graph>sB + OD% ( ( ( ( (   # #E * * * * *r4c|jstj||dS|d| |j|dS#t $rYdSwxYw)Nr[)rr remove_graphrPrrrs r3rzMemory.remove_graphDs   tU + + + + + KK*E 2 2 2 #**511111    sA A A rc:||}t|}|rO |j|}n3#t$r&|jx}|j|<YnwxYw|||<|s||d<n!|r||ix}|j|<n||d|ix}|j|<|s |jd|||jvrt|j|<|j|||j||_||jkr |j|=dSdS)z;add the given context to the set of contexts for the tripleN) rr8rrrrrrIr)r1r5rr6r7rtriple_contexts r3__add_triple_contextzMemory.__add_triple_contextOs((f   !%!6v!>    *//11!6v!>!>!>  #)N3  .'-t$ BEvN!6v!>!>&B!6v!>  4  !$ ' + +F 3 3 3 d+ + +),D !# & c"&&v...  ! )%3D " T3 3 3%f--- 4 3s 6-A&%A&rCollection[Optional[str]]c|j||j}|s|Sd|DS)zdreturn a list of contexts (str) for the triple, skipping quoted contexts if skipQuoted==Truecg|] \}}|| Srzrz)r{rr7s r3 z3Memory.__get_context_for_triple..s!BBB V6BBBBr4)rrerrUrs)r1r5rrs r3__get_context_for_triplezMemory.__get_context_for_triplesT $((1GHH 99;; CBtzz||BBBBr4rcF||j||jvS)z5return True if the triple exists in the given context)rrer)r1r5rs r3__triple_has_contextzMemory.__triple_has_contexts%d+//8NOOOOr4c|j||j}||=||jkr |j|=n ||j|<|j||dS)z"remove the context from the tripleN)rrerrrrP)r1r5rrs r3__remove_triple_contextzMemory.__remove_triple_contexts}$((1GHHMMOO I 4) ) )%f--,0D !& ) c"))&11111r4cdSr(rzr1rs r3 __ctx_to_strzMemory.__ctx_to_strs69cr4cdSr(rzrs r3rzMemory.__ctx_to_strs/2sr4cR|dS d|jjj|j}||j|<|S#t $r`t |tr;d|jj|}||jvr|cYS||j|<|cYStdwxYw)Nz{}:{}z1Cannot use that type of object as a Graph context) formatr%r2rrAttributeError isinstancer_ RuntimeError)r1rctx_strs r3rzMemory.__ctx_to_strs ;4 TnnS^%=%FWWG.1D "7 +N T T T#s## !..)?EEd444"NNN25&w/RSS S Ts5<A B& B&B&cJfd|dDS)zgreturn a generator for all the non-quoted contexts (dereferenced) the encoded triple appears inc3RK|]!}|j||V"dSr()rre)r{rr1s r3r|z$Memory.__contexts..sH  "  " & &w 8 8""""  r4Tr)r)r1r5s` r3r}zMemory.__contextssA    88D8QQ    r4r~rrrrrrrc Ptt|j||||fi|dSr()r)rr~rs r3r~z Memory.querys6 "fd!%zTTVTTTTTr4rUnion[Update, Any]c Ptt|j||||fi|dSr()r)rrrs r3rz Memory.updates6 #fd"66<VVvVVVVVr4rrrrr(r)rJrr6rKr9rrrrrr)r5rr9rwr)rrr9r ) r5rrr8r6rKr7r8r9r )r5rrr8r9r)r5rrr$r9r8)r5r)rrr9r_)rr r9r )rrKr9r$)r5rr9rw) r~rrrrrrr_r9r) rrrrrrrr_r9r )rrrr context_aware formula_awarerr*rIrPrNrir`r^rurr]rrrrrrrrrr~rrrs@r3rr s MMK(,+/KKKKKKKJ :::::~+/$0$0$0$0$0R+/jBjBjBjBjBX422222222$$$$ /3 # # # # #///// ++++ 3.3.3.3.l7< C C C C CPPPP 2 2 2 2999X9 222X2TTTT*    UUUUUUWWWWWWWWWWr4)( __future__rtypingrrrrrr r r r r rr rdflib.storer rdflib.utilr rdflib.graphrrrrrrrrdflib.plugins.sparql.sparqlrr rdflib.queryr rdflib.termrr__all__r!__annotations__rrrzr4r3rs #""""""                            !!!!!! /;:::::::######........ 8 $ ` ` ` ` ` 5` ` ` FNWNWNWNWNWUNWNWNWNWNWr4