o i6v@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  dCdDfdd Z dEdFddZ dGdHddZ dGdIddZdGdJddZdKdLd%d&Z dMd(d)Z dNd*d+Z dOd-d.Z dPd0d1Z dQfd=d> ZdRfdAdB ZZS)Sra) 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']cs8tt||||_i|_i|_i|_i|_i|_dSN) superr__init__r#_SimpleMemory__spo_SimpleMemory__pos_SimpleMemory__osp_SimpleMemory__namespace_SimpleMemory__prefixselfr!r# __class__W/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/rdflib/plugins/stores/memory.pyr'5s zSimpleMemory.__init__Ftriple '_TripleType'context'_ContextType'quotedboolreturnrcCs|\}}}|j}z||}Wn i}||<Yz||} Wn i} ||<Yd| |<|j} z| |} Wn i} | |<Yz| |} Wn i} | |<Yd| |<|j} z| |}Wn i}| |<Yz||}Wn i}||<Yd||<dS)6 Add a triple to the store of triples. N)r(r)r*)r.r3r5r7subject predicateobjectspopooposossospsppr1r1r2addOs>        zSimpleMemory.addtriple_pattern'_TriplePatternType'Optional['_ContextType']cCsRt||D]\\}}}}|j|||=|j|||=|j|||=qdSr%)listtriplesr(r)r*)r.rIr5r<r=r>cr1r1r2remove}s zSimpleMemory.removeBIterator[Tuple['_TripleType', Iterator[Optional['_ContextType']]]]ccsV|\}}}|tkr|j}||vr||}|tkrN||vrK|tkr6|||vr3|||f|fVdS dS||D] }|||f|fVqr?subjectDictionaryrArGrBpredicateDictionaryrDrEobjectDictionaryr1r1r2rMsz          zSimpleMemory.triplesintcCs d}|dD]}|d7}q|S)NrNNNr;)rM)r.r5ir3r1r1r2__len__s zSimpleMemory.__len__Tprefixstr namespace'URIRef'overridecC|j|}t|j||j|}|r1|dur|j|=|dur%|j|=||j|<||j|<dSt||d|jt||<t||d|jt||<dSN)default)r+getrr,r.r[r]r_bound_namespace bound_prefixr1r1r2binds$    zSimpleMemory.bindOptional['URIRef']cC|j|dSr%)r+rcr.r[r1r1r2r]zSimpleMemory.namespacecCrir%)r,rcr.r]r1r1r2r[rkzSimpleMemory.prefixIterator[Tuple[str, 'URIRef']]cc$|jD] \}}||fVqdSr%)r+itemsr.r[r]r1r1r2 namespaces zSimpleMemory.namespaces%Generator['_ContextType', None, None]cCsdddDS)Ncs|]}|VqdSr%r1).0rNr1r1r2 z*SimpleMemory.__contexts..r1r1r.r1r1r2 __contextsszSimpleMemory.__contextsqueryUnion['Query', str]initNsMapping[str, Any] initBindingsMapping['str', 'Identifier'] queryGraph'str'kwargsr'Result'c "tt|j||||fi|dSr%)r&rrzr.rzr|r~rrr/r1r2rz  zSimpleMemory.queryupdateUnion['Update', str]c rr%)r&rrr.rr|r~rrr/r1r2rrzSimpleMemory.updateNNr!r"r#r$Fr3r4r5r6r7r8r9rr%rIrJr5rKr9r)rIrJr5rKr9rPr5rKr9rWTr[r\r]r^r_r8r9rr[r\r9rhr]r^r9r"r9rm)r9rs) rzr{r|r}r~rrrrrr9r) rrr|r}r~rrrrrr9r)__name__ __module__ __qualname____doc__r'rHrOrMrZrgr]r[rqrRrzr __classcell__r1r1r/r2r*s&  1  @      cs.eZdZdZdZdZdZ  dZd[fdd Z d\d]ddZ d^d_ddZ d^d`ddZ dadbd"d#Z dcd%d&Z ddd'd(Z ded*d+Z d^dfd.d/Zd^dgd1d2Zdhd5d6Zdhd7d8Zdid:d;Z d\djd>d?ZdkdAdBZdldCdDZedmdEdFZedndGdFZdodHdFZdpdIdJZdqfdTdU ZdrfdXdY ZZS)srz An 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__Fr3r4r5r6r7r8r9rcCstj||||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)r:)r7NTr;F) rrHrr LookupErrorKeyError_Memory__add_triple_contextrr)r.r3r5r7r<r=object_r?r@rA_ triple_existsrBrCrDrErFrGr1r1r2rH=sb                 z Memory.addrIrJrKc 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)r5T skipQuotedr;rrX) _Memory__ctx_to_strrM_Memory__get_context_for_triple_Memory__remove_triple_contextlenrrrrrr graph_awarerO) r.rIr5req_ctxr3rNr<r=rctxctxsr1r1r2rO{s6     z Memory.remove\Generator[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אqdSrQ) rrcopy_Memory__contextsr_Memory__triple_has_contextrrLrSrr)r.rIr5rr<r=rr3rr?rTrArGrBrUrDrErVr1r1r2rMs                         zMemory.triplesr[r\r]r^r_cCr`ra)rrcrrrdr1r1r2rg s$    z Memory.bindrhcCrir%)rrcrjr1r1r2r]'rkzMemory.namespacecCrir%)rrcrlr1r1r2r[*rkz Memory.prefixrmccrnr%)rrorpr1r1r2rq-rrzMemory.namespacesOptional['_TripleType']rscCsj|dus|dkrdd|jDS|\}}}z|j|||}||WSty4dddDYSw)NrXcsrtr%r1)rur5r1r1r2rv5rwz"Memory.contexts..csrtr%r1)rurr1r1r2rv<rwr1)rrrr)r.r3subjpredobjrr1r1r2contexts1s   zMemory.contextsrWcCs&||}||jvr dSt|j|S)Nr)rrr)r.r5rr1r1r2rZ>s  zMemory.__len__graph'Graph'cCs&|js t||dS|j|dSr%)rr add_graphrrHr.rr1r1r2rDszMemory.add_graphcCsJ|js t||dS|d|z |j|WdSty$YdSw)NrX)rr remove_graphrOrrrr1r1r2rJs  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) rr8rrrrrrHr)r.r3rr5r7rtriple_contextr1r1r2__add_triple_contextUs<       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)rurr7r1r1r2 sz3Memory.__get_context_for_triple..)rrcrrSro)r.r3rrr1r1r2__get_context_for_tripleszMemory.__get_context_for_triplercCs||j||jvS)z5return True if the triple exists in the given context)rrcr)r.r3rr1r1r2__triple_has_contextszMemory.__triple_has_contextcCsL|j||j}||=||jkr|j|=n||j|<|j||dS)z"remove the context from the tripleN)rrcrrrrO)r.r3rrr1r1r2__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 isinstancer\ RuntimeError)r.rctx_strr1r1r2rs     csfddj|ddDS)zgreturn a generator for all the non-quoted contexts (dereferenced) the encoded triple appears inc3s&|]}|durj||VqdSr%)rrc)rurrxr1r2rvs z$Memory.__contexts..Tr)r)r.r3r1rxr2rys  zMemory.__contextsrzr{r|r}r~rrrrc rr%)r&rrzrr/r1r2rz"z Memory.queryrUnion['Update', Any]c rr%)r&rrrr/r1r2rrz Memory.updaterrrrr%r)rIrJr5rKr9rrrrrr)r3rr9rsr)rrr9r) r3r4rr8r5rKr7r8r9r)r3r4rr8r9r)r3r4rr"r9r8)r3r4)rr6r9r\)rrr9r)rrKr9r")r3r4r9rs) rzr{r|r}r~rrrr9r) rrr|r}r~rrrr9r)rrrr context_aware formula_awarerr'rHrOrMrgr]r[rqrrZrrrrrrrrrrzrrr1r1r/r2rsH% A ) l      6        )( __future__rtypingrrrrrrr r r r r r rdflib.storer rdflib.utilr rdflib.graphrrrrrrrrdflib.plugins.sparql.sparqlrr rdflib.queryr rdflib.termrr__all__r __annotations__rrr1r1r1r2s8  $  g