3 d+@s<ddlmZddlZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z ddlmZmZmZmZmZmZddlmZddlmZddlZGd d d eZGd d d eZGd ddeZGdddeZGddde ZGddde Z Gddde Z!Gddde"Z#Gddde"Z$Gddde"Z%dS))absolute_importN) text_type iteritems)MappingMutableMapping)NamespaceManager)VariableBNodeGraphConjunctiveGraphURIRefLiteral)Node) CompValuec@seZdZdddZdS) SPARQLErrorNcCstj||dS)N) Exception__init__)selfmsgr>/tmp/pip-build-7vycvbft/rdflib/rdflib/plugins/sparql/sparql.pyrszSPARQLError.__init__)N)__name__ __module__ __qualname__rrrrrrsrc@seZdZdddZdS) NotBoundErrorNcCstj||dS)N)rr)rrrrrrszNotBoundError.__init__)N)rrrrrrrrrsrc@seZdZdZddZdS) AlreadyBoundz\sz#Bindings.__str__..z}))join)rr)rr__str__[szBindings.__str__cCst|S)N)r)rrrr__repr__^szBindings.__repr__) rrrrrr%r&r(r)r-r.r3r4rrrrr*s rc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS) FrozenDictz^ An immutable hashable dict Taken from http://stackoverflow.com/a/2704866/81121 cOst|||_d|_dS)N)rr _hash)rargskwargsrrrrjs zFrozenDict.__init__cCs t|jS)N)iterr )rrrrr.nszFrozenDict.__iter__cCs t|jS)N)lenr )rrrrr-qszFrozenDict.__len__cCs |j|S)N)r )rr$rrrr%tszFrozenDict.__getitem__cCsP|jdkrJd|_x8t|D],\}}|jt|N_|jt|N_qW|jS)Nr)r6rhash)rr$r'rrr__hash__ws  zFrozenDict.__hash__cstfddt|DS)Nc3s|]}|dkr|VqdS)rNr)r/r,)varsrrr1sz%FrozenDict.project..)r5r)rr=r)r=rprojectszFrozenDict.projectcCstt|j| S)N)boolset intersection)rotherrrrdisjointDomainszFrozenDict.disjointDomainc Cs@x:|D]2}y||||kr dSWqtk r6YqXqWdS)NFT)r#)rrBr0rrr compatibles  zFrozenDict.compatiblecCsttjt|t|}|S)N)r5 itertoolschainr)rrBresrrrmergeszFrozenDict.mergecCs t|jS)N)strr )rrrrr3szFrozenDict.__str__cCs t|jS)N)reprr )rrrrr4szFrozenDict.__repr__N)rrrrrr.r-r%r<r>rCrDrHr3r4rrrrr5bs  r5c@sneZdZddZddZddZddZd d Zd d Zd dZ e e Z e eZ e eZ dddZddZdS)FrozenBindingscOstj|f||||_dS)N)r5rctx)rrLr7r8rrrrszFrozenBindings.__init__cCsJt|tst|}t|ttfkr&|S||jkr<|jj|S|j|SdS)N) isinstancerrtyper r rL initBindings)rr$rrrr%s   zFrozenBindings.__getitem__cst|jfddt|DS)Nc3s|]}|dkr|VqdS)rNr)r/r,)r=rrr1sz)FrozenBindings.project..)rKrLr)rr=r)r=rr>szFrozenBindings.projectcCs t|jtjt|t|}|S)N)rKrLrErFr)rrBrGrrrrHszFrozenBindings.mergecCs|jjS)N)rLnow)rrrr_nowszFrozenBindings._nowcCs|jjS)N)rLbnodes)rrrr_bnodesszFrozenBindings._bnodescCs|jjS)N)rLprologue)rrrr _prologueszFrozenBindings._prologueNcs*sgtjfddtDS)zY return a frozen dict only of bindings made in self since before c3s>|]6}|dks2|djjks2|ddkr|VqdS)rN)rLrO)r/r,)_exceptbeforerrrr1s z(FrozenBindings.forget..)rKrLr)rrWrVr)rVrWrrforgets zFrozenBindings.forgetcst|jfddt|DS)z@ return a frozen dict only of bindings in these c3s|]}|dkr|VqdS)rNr)r/r,)theserrr1sz*FrozenBindings.remember..)rKrLr)rrYr)rYrrememberszFrozenBindings.remember)N)rrrrr%r>rHrQrSrUpropertyrTrRrPrXrZrrrrrKs  rKc@seZdZdZdddZdddZddZeed d Zd d d Z ddZ d!ddZ d"ddZ ddZ ddZddZddZddZdS)# QueryContextz@ Query context - passed along when evaluating the query NcCs||_t|pgd|_|r&|jj|t|trV||_tjj j rJ|j |_ qb|j j |_ n d|_||_ d|_tjjtjj|_tjt|_dS)N)r")rOrbindingsupdaterMr _datasetrdflibpluginssparqlZSPARQL_DEFAULT_GRAPH_UNIONdatasetgraphZdefault_contextrTdatetimerPisodatetzinfoUTC collections defaultdictr rR)rrdr]rOrrrrs     zQueryContext.__init__cCsDt|jdk r|jn|j|p|j|jd}|j|_|j|_|j|_|S)N)rO)r\r_rdr]rOrTrR)rr]rrrrclones &zQueryContext.clonecCs|jdkrtd|jS)Nz*You performed a query operation requiring z'a dataset (i.e. ConjunctiveGraph), but z&operating currently on a single graph.zQYou performed a query operation requiring a dataset (i.e. ConjunctiveGraph), but zwYou performed a query operation requiring a dataset (i.e. ConjunctiveGraph), but operating currently on a single graph.)r_r)rrrr _get_datasets  zQueryContext._get_datasetzcurrent dataset)docFc sTfdd}tjjjs2|rP|j|jj|7_n|rD||j|n ||j|dS)Ncsty|j|fSYnXy|j|fddiSYnXy|j|fddiStd|YnXdS)NformatZn3ntz3Could not load %s as either RDF/XML, N3 or NTriples)loadr)rdsource)r8rr_loadsz QueryContext.load.._load)r`rarbZSPARQL_LOAD_GRAPHSrdrcZ get_context)rrrdefaultr8rsr)r8rrqs  zQueryContext.loadc Cs8t|ttfkr|Sy |j|Stk r2dSXdS)N)rNr rr]r#)rr$rrrr%/s  zQueryContext.__getitem__c Cs"y||Stk r|SXdS)N)r#)rr$rtrrrget8szQueryContext.getcs6r"t|fddt|jDSt|t|jSdS)zO Return a static copy of the current variable bindings as dict c3s"|]\}}|kr||fVqdS)Nr)r/r0v)r=rrr1Dsz(QueryContext.solution..N)rKrr])rr=r)r=rsolution>s  zQueryContext.solutioncCs,||jkr|j||krt||j|<dS)N)r]r)rr$r'rrrr(JszQueryContext.__setitem__cCs|j}||_|S)N)rlrd)rrdrkrrr pushGraphPszQueryContext.pushGraphcCs|jt|j}|S)N)rlrr])rrkrrrpushUszQueryContext.pushcCs |jgS)N)rl)rrrrcleanYszQueryContext.cleancCs|j|}|S)zO Create a new read/write query context from the given solution )rl)rZfrozenbindingscrrrthawas zQueryContext.thaw)NNN)N)F)N)N)rrrrrrlrmr[rcrqr%rurwr(rxryrzr|rrrrr\s      r\c@s0eZdZdZddZddZddZdd Zd S) ProloguezI A class for holding prefixing bindings and base URI information cCsd|_tt|_dS)N)baserr namespace_manager)rrrrrpszPrologue.__init__cCs6|jjj|pd}|dkr&td|t||p0dS)NzUnknown namespace prefix : %s)rstore namespacerr )rprefix localnamensrrr resolvePNameus zPrologue.resolvePNamecCs|jj||dddS)NT)replace)rbind)rrurirrrr{sz Prologue.bindcCsnt|trJ|jdkr$|j|j|jS|jdkrjt|j|j|j |j dSn t|t rjd|krjt ||j dS|S)z Apply BASE / PREFIXes to URIs (and to datatypes in Literals) TODO: Move resolving URIs to pre-processing Zpnameliteral)langdatatype:)r~) rMrnamerrrr stringr absolutizerr r~)rZirirrrr~s   zPrologue.absolutizeN)rrrrrrrrrrrrr}js r}c@seZdZdZddZdS)Queryz' A parsed and translated query cCs||_||_dS)N)rTalgebra)rrTrrrrrszQuery.__init__N)rrrrrrrrrrsr)& __future__rrirErerfsixrrZ rdflib.compatrrZrdflib.namespacerr`rr r r r r Z rdflib.termrZ!rdflib.plugins.sparql.parserutilsrZrdflib.plugins.sparqlrrrrrrr5rKobjectr\r}rrrrrs,     8@B)