o mi+@sdZddlmZddlZddlZddlmZmZm Z ddl m Z ddl m Z mZmZejr7ddlmZmZmZ d,d-ddZ   d.d/ddZ d0d1ddZd2d d!Zd3d%d&Zd4d(d)Zd5d*d+ZdS)6zRDF4J utility functions.) annotationsN)DATASET_DEFAULT_GRAPH_IDDatasetGraph) prepareQuery)BNodeIdentifiedNodeURIRef) ObjectType PredicateType SubjectTypeparamsdict[str, str] graph_name8IdentifiedNode | t.Iterable[IdentifiedNode] | str | NonereturnNonecCs|durt|tr|tkrd|d<dS||d<dS|dur.t|tr.t||d<dS|durHt|tjrJddd|D}||d<dSdSdS)aJBuild the RDF4J http context query parameters dictionary. !!! Note This mutates the params dictionary key `context`. Parameters: params: The `httpx.Request` parameter dictionary. graph_name: The graph name or iterable of graph names. This is the `context` query parameter value. Nnullcontext,cSsg|]}|qSn3.0xrrS/home/kim/smarthome/.venv/lib/python3.10/site-packages/rdflib/contrib/rdf4j/util.py )sz'build_context_param..) isinstancerrrstrr tIterablejoin)r rZ graph_namesrrrbuild_context_params  r#subjr predr objr cCsD|dur ||d<|dur||d<|dur ||d<dSdS)aBuild the RDF4J http subj, predicate, and object query parameters dictionary. !!! Note This mutates the params dictionary key `subj`, `pred`, and `obj`. Parameters: params: The `httpx.Request` parameter dictionary. subj: The `subj` query parameter value. pred: The `pred` query parameter value. obj: The `obj` query parameter value. Nr$r%r&r)r r$r%r&rrrbuild_spo_param-s  r'TinferboolcCs|sd|d<dSdS)zBuild the RDF4J http infer query parameters dictionary. !!! Note This mutates the params dictionary key `infer`. Parameters: params: The `httpx.Request` parameter dictionary. infer: The `infer` query parameter value. falser(Nr)r r(rrrbuild_infer_paramFs r+data*str | bytes | t.BinaryIO | Graph | Datasettuple[t.BinaryIO, bool]c Cst|trBd|vr4t|dkr4z t|d}d}W||fSttfy3t|d}d}Y||fSwt|d}d}||fSt|t rRt|}d}||fSt|t t frw|j rgt|j ddd}n t|j d dd}d}||fS|}d}||fS) aConvert an RDF payload into a file-like object. Parameters: data: The RDF payload. This can be a python `str`, `bytes`, `BinaryIO`, or a [`Graph`][rdflib.graph.Graph] or [`Dataset`][rdflib.graph.Dataset]. Returns: A tuple containing the file-like object and a boolean indicating whether the immediate caller should close the stream.  irbTzutf-8Fzapplication/n-quads)formatencodingapplication/n-triples)rrlenopenFileNotFoundErrorOSErrorioBytesIOencodebytesrrZ context_aware serialize)r,streamZ should_closerrrrdf_payload_to_streamWs>     r>queryrheaderscCsJt|}|jjdvrd|d<dS|jjdvrd|d<dStd|jj)zBuild the SPARQL query accept header. !!! Note This mutates the headers dictionary key `Accept`. Parameters: query: The SPARQL query. )Z SelectQueryZAskQueryzapplication/sparql-results+jsonAccept)ZConstructQueryZ DescribeQueryr3zUnsupported query type: N)rZalgebraname ValueError)r?r@Zprepared_queryrrr build_sparql_query_accept_headers     rD(URIRef | t.Iterable[URIRef] | str | NonecCs8t|tst|tjrtdd|DrtddSdS)Ncss|]}t|tVqdSN)rrrrrr sz&validate_graph_name..zGraph name must not be a BNode.)rrr r!anyrC)rrrrvalidate_graph_names rIcCsNt|tst|tst|tst|tr!td|d|d|t|dS)z@Validate that the subject, predicate, and object are not BNodes.z4Subject, predicate, and object must not be a BNode: z, N)rrrCrI)r$r%r&rrrrvalidate_no_bnodess$ rJrF)r rrrrr)NNN) r rr$r r%r r&r rr)T)r rr(r)rr)r,r-rr.)r?rr@r)rrE) r$r r%r r&r rrErr)__doc__ __future__rr8typingr Z rdflib.graphrrrZrdflib.plugins.sparql.processorrZ rdflib.termrrr TYPE_CHECKINGZrdflib.contrib.rdf4j.clientr r r r#r'r+r>rDrIrJrrrrs*      3