o minu@sUddlmZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZerYddlmZmZmZmZmZmZmZddlmZmZddlmZddl m!Z!m"Z"d d gZ#d Z$d e%d <Gdd d eZ&Gdd d eZ'd S)) annotations) TYPE_CHECKINGAny CollectionDict GeneratorIteratorMappingOptionalSetTupleUnionoverload)Store) _coalesce)Graph _ContextType _ObjectType_PredicateType _SubjectType_TriplePatternType _TripleType)QueryUpdate)Result) IdentifierURIRef SimpleMemoryMemoryNNoneANYcseZdZdZ  dBdCfdd Z dDdEddZ dFdGddZ dFdHddZdFdIddZdJdKd%d&Z dLd(d)Z dMd*d+Z dNd-d.Z dOd0d1Z dPfd        zSimpleMemory.addtriple_patternrOptional[_ContextType]cCsRt||D]\\}}}}|j|||=|j|||=|j|||=qdSr%)listtriplesr(r)r*)r.rGr4r:r;r<cr1r1r2removezs zSimpleMemory.remove>Iterator[Tuple[_TripleType, Iterator[Optional[_ContextType]]]]ccsV|\}}}|tkr|j}||vr||}|tkrN||vrK|tkr6|||vr3|||f|fVdS dS||D] }|||f|fVqz*SimpleMemory.__contexts..r1r1r.r1r1r2 __contextsszSimpleMemory.__contextsqueryUnion[Query, str]initNsMapping[str, Any] initBindingsMapping[str, Identifier] queryGraphkwargsrrc "tt|j||||fi|dSr%)r&rrtr.rtrvrxrzr{r/r1r2rt  zSimpleMemory.queryupdateUnion[Update, str]c r|r%)r&rrr.rrvrxrzr{r/r1r2rr~zSimpleMemory.updateNNr!r"r#r$Fr3rr4rr5r6r7rr%rGrr4rHr7r)rGrr4rHr7rMr4rHr7rTTrXrYrZrr[r6r7rrXrYr7rbrZrr7r"r7rg)r7rm) rtrurvrwrxryrzrYr{rr7r) rrrvrwrxryrzrYr{rr7r)__name__ __module__ __qualname____doc__r'rFrLrJrWrarZrXrkrOrtr __classcell__r1r1r/r2r*s&  /  @      cs.eZdZdZdZdZdZ  dYdZfdd Z d[d\ddZ d]d^ddZ d]d_ddZ d`dad"d#Z dbd%d&Z dcd'd(Z ddd*d+Z d]ded.d/Zd]dfd1d2Zdgd5d6Zdgd7d8Zdhd:d;Z d[did>d?ZdjdAdBZdkdCdDZedldEdFZedmdGdFZdndHdFZdodIdJZdpfdSdT ZdqfdWdX ZZS)rrzAn 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$cs^tt||||_i|_i|_i|_i|_i|_i|_ i|_ dt i|_ 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__defaultContextsr-r/r1r2r's  zMemory.__init__Fr3rr4rr5r6r7rcCstj||||d|dur|j||\}}}|j}z||}Wnty0i}||<Ynwz||} WntyFi} ||<Ynwz| |} d} Wnty^d| |<d} Ynw||| ||| rkdS|j} z| |} Wntyi} | |<Ynwz| |}Wntyi}| |<Ynwd||<|j}z||}Wntyi}||<Ynwz||}Wntyi}||<Ynwd||<dS)r8)r5NTr9F) rrFrr LookupErrorKeyError_Memory__add_triple_contextrr)r.r3r4r5r:r;object_r=r>r?_ triple_existsr@rArBrCrDrEr1r1r2rF9sb                 z Memory.addrGrrHc Cs:||}|j||dD]`\}}|\}}}||D]} |dur%|| kr%q||| q|j|dd} d| vrG|dusAt| dkrG||dt||dkrl|j|||=|j|||=|j|||=|j|=q |dur||j vrt|j |dkr|j |=|dkr||j vr|j s|j |dSdSdSdS)N)r4T skipQuotedr9rrU) _Memory__ctx_to_strrJ_Memory__get_context_for_triple_Memory__remove_triple_contextlenrrrrrr graph_awarerL) r.rGr4req_ctxr3rKr:r;rctxctxsr1r1r2rLus6     z Memory.removeXGenerator[Tuple[_TripleType, Generator[Optional[_ContextType], None, None]], None, None]ccs||}|\}}}|dur2|dur2|dur2||jvrdS|j|D] }|||fVq%dS|durh|durh|durh|}z|j|||}|||r[|||fVWdSWdStygYdSw|dur|j} || vr| |} |dur|| vr|dur|| |vr|||f}|||r|||fVdSdS dSt| |D]} ||| f}|||r|||fVqdS dSt| D]A} |dur|| | vr|| |f}|||r|||fVq qt| | D]} || | f}|||r|||fVqqdS dS|dur|j } || vr| |}|durU||vrRt||D]}|||f}|||rN|||fVq8dS dSt|D]%} t|| D]}||| f}|||r{|||fVqeq[dSdS|dur|j }||vr||}t|D]%}t||D]} || |f}|||r|||fVqqdSdS|j} t| D]1}| |} t| D]#} t| | D]} || | f}|||r|||fVqqאqdSrN) rrcopy_Memory__contextsr_Memory__triple_has_contextrrIrPrr)r.rGr4rr:r;rr3rr=rQr?rEr@rRrBrCrSr1r1r2rJs                         zMemory.triplesrXrYrZrr[cCr\r])rr_rrr`r1r1r2ras$    z Memory.bindrbcCrcr%)rr_rdr1r1r2rZ!rezMemory.namespacecCrcr%)rr_rfr1r1r2rX$rez Memory.prefixrgccrhr%)rrirjr1r1r2rk'rlzMemory.namespacesOptional[_TripleType]rmcCsn|dus|dkrddt|jDS|\}}}z|j|||}||WSty6dddDYSw)NrUcsrnr%r1)ror4r1r1r2rp/rqz"Memory.contexts..csrnr%r1)rorr1r1r2rp6rqr1)rIrrrr)r.r3Zsubjpredobjrr1r1r2contexts+s   zMemory.contextsrTcCs&||}||jvr dSt|j|S)Nr)rrr)r.r4rr1r1r2rW8s  zMemory.__len__graphrcCs&|js t||dS|j|dSr%)rr add_graphrrFr.rr1r1r2r>szMemory.add_graphcCsJ|js t||dS|d|z |j|WdSty$YdSw)NrU)rr remove_graphrLrrrr1r1r2rDs  zMemory.remove_graphrcCs||}t|}|r1z|j|}Wnty%|j}|j|<Ynw|||<|s0||d<n|r=||i}|j|<n ||d|i}|j|<|sR|jd|||jvr]t|j|<|j|||jdurm||_||jkrx|j|=dSdS)z;add the given context to the set of contexts for the tripleN) rr6rrrrrrFr)r.r3rr4r5rZtriple_contextr1r1r2Z__add_triple_contextOs8       zMemory.__add_triple_contextrCollection[Optional[str]]cCs.|j||j}|s|Sdd|DS)zdreturn a list of contexts (str) for the triple, skipping quoted contexts if skipQuoted==TruecSsg|]\}}|s|qSr1r1)rorr5r1r1r2 sz3Memory.__get_context_for_triple..)rr_rrPri)r.r3rrr1r1r2Z__get_context_for_tripleszMemory.__get_context_for_triplercCs||j||jvS)z5return True if the triple exists in the given context)rr_r)r.r3rr1r1r2Z__triple_has_contextszMemory.__triple_has_contextcCsL|j||j}||=||jkr|j|=n||j|<|j||dS)z"remove the context from the tripleN)rr_rrrrL)r.r3rrr1r1r2Z__remove_triple_contexts    zMemory.__remove_triple_contextcCdSr%r1r.rr1r1r2 __ctx_to_strzMemory.__ctx_to_strcCrr%r1rr1r1r2rrcCs|durdSzd|jjj|j}||j|<|WStyBt|tr>d|jj|}||jvr5|YS||j|<|YStdw)Nz{}:{}z1Cannot use that type of object as a Graph context) formatr#r0rrAttributeError isinstancerY RuntimeError)r.rctx_strr1r1r2rs     csfddj|ddDS)zgreturn a generator for all the non-quoted contexts (dereferenced) the encoded triple appears inc3s&|]}|durj||VqdSr%)rr_)rorrrr1r2rps z$Memory.__contexts..Tr)r)r.r3r1rrr2rss  zMemory.__contextsrtrurvrwrxryrzrc r|r%)r&rrtr}r/r1r2rt"z Memory.queryrUnion[Update, Any]c r|r%)r&rrrr/r1r2rrz Memory.updaterrrrr%r)rGrr4rHr7rrrrrr)r3rr7rmr)rrr7r) r3rrr6r4rHr5r6r7r)r3rrr6r7r)r3rrr"r7r6)r3r)rrr7rY)rrr7r)rrHr7r")r3rr7rm) rtrurvrwrxryrzrYr7r) rrrvrwrxryrzrYr7r)rrrrZ context_awareZ formula_awarerr'rFrLrJrarZrXrkrrWrrrrrrrrrrtrrr1r1r/r2r sH% ? ) l      6        )( __future__rtypingrrrrrrr r r r r rZ rdflib.storerZ rdflib.utilrZ rdflib.graphrrrrrrrZrdflib.plugins.sparql.sparqlrrZ rdflib.queryrZ rdflib.termrr__all__r __annotations__rrr1r1r1r2s8  $  d