3 d<@sddlZddlZddlmZddlZddlZddlZddlZddlm Z m Z ddl Tddl m Z dZdZejeZeeZejeedd eDZd ZGd d d ejZGd dde ZddZddZddZddZ ddZ!ddZ"dS)N)etree)DEFAULT_NAMESPACESsorted_attributes)*) Serializerz Lion Krischerz"krischer@geophysik.uni-muenchen.deccs|]}t||fVqdS)N)FULL_NAMES_MAP).0Z rec_type_idr 8/tmp/pip-build-7vycvbft/prov/prov/serializers/provxml.py sr z http://www.w3.org/2001/XMLSchemac@s eZdZdS)ProvXMLExceptionN)__name__ __module__ __qualname__r r r r r sr c@s<eZdZdZdddZdddZdd Zd d Zd d ZdS)ProvXMLSerializerz9PROV-XML serializer for :class:`~prov.model.ProvDocument`FcKs||j|j|d}x |jjD]}|j|||dqWtj|}t|tjrf|jtj |dddj dn|j|dddddS) a Serializes a :class:`~prov.model.ProvDocument` instance to `PROV-XML `_. :param stream: Where to save the output. :type force_types: boolean, optional :param force_types: Will force xsd:types to be written for most attributes mainly PROV-"attributes", e.g. tags not in the PROV namespace. Off by default meaning xsd:type attributes will only be set for prov:type, prov:location, and prov:value as is done in the official PROV-XML specification. Furthermore the types will always be set if the Python type requires it. False is a good default and it should rarely require changing. )bundle force_types)relementrT)xml_declaration pretty_printzutf-8zUTF-8)rrencodingN) serialize_bundledocumentZbundlesr ElementTree isinstanceio TextIOBasewritetostringdecode)selfstreamrkwargsZxml_rootretr r r serialize%s   zProvXMLSerializer.serializeNcCsdd|jjjD}|jjjr0|jjjj|d<x"|jD]}||kr8|j||j<q8Wx8tjD],\}}|j}|jdkr|j d}|||j<q^W|dk rt j |t d|d} nt j t d|d} |jrt|j| jt d <x|jD]} | j} | jrt| jnd} | rt d | i} nd} t| j}|j| |}t j | t || }xJt| |D]:\}}t j |t|jj|j}t|tjjr|jdt d gkrd |jjj|jjf|jt!d <|j"dk r|j"|jt#d <|j$}nRt|tjj%r|t&krd|jt!d <t|}n t|t'j'r(|j(}nt|}t)t'j't*t+tjj,g}t-|}|spt.||ksp|t/t0t1gkrht!d |jkrht|j2d rh|t&ko| rh|t3t4gkrhd}t|t)rt5}|j6}n|t|trt7}njt|t*rt8}nXt|t+rt9}nFt|t'j'r8|jjdks2d|jj6krLt:}nt|tjj,rLt;}|dk rht||jt!d <|t&kr|r||jt d<n||_<qTWqW| S)a Serializes a bundle or document to PROV XML. :param bundle: The bundle or document. :param element: The XML element to write to. Will be created if None. :type force_types: boolean, optional :param force_types: Will force xsd:types to be written for most attributes mainly PROV-"attributes", e.g. tags not in the PROV namespace. Off by default meaning xsd:type attributes will only be set for prov:type, prov:location, and prov:value as is done in the official PROV-XML specification. Furthermore the types will always be set if the Python type requires it. False is a good default and it should rarely require changing. cSsi|]}|j|jqSr )uriprefix)rnsr r r Wsz6ProvXMLSerializer.serialize_bundle..NZxsd# bundleContent)nsmapridZInternationalizedStringz%s:%stypelangz xsd:QNamezprov:provtimeref)=r _namespacesZget_registered_namespaces_defaultr% namespacesr&ritemsrstripr SubElement_ns_provElement identifierstrattribZ_recordsget_typeZ _identifierlist attributes_derive_record_labelr_ns namespace localpartrr/modelLiteraldatatypePROV_ns_xsilangtag_ns_xmlvalueZ QualifiedNameZPROV_ATTRIBUTE_QNAMESdatetime isoformatboolfloatintZ Identifiertupler- PROV_TYPEZ PROV_LOCATIONZ PROV_VALUE startswithZPROV_ATTR_TIMEZ PROV_LABELZ XSD_BOOLEANlowerZ XSD_STRINGZ XSD_DOUBLEZXSD_INTZ XSD_DATETIMEZ XSD_ANYURItext)r rrrr+rBkeyrKr%Zxml_bundle_rootrecordrec_typer:attrsr? rec_labelelemattrZsubelemvZ ALWAYS_CHECKZxsd_typer r r rFs                 z"ProvXMLSerializer.serialize_bundlec Kst|tjrPtj4}|j|jjd|jddtj |j }WdQRXntj |j }x$|j dD]}|j }|j |qjWtjj}|j|||S)z Deserialize from `PROV-XML `_ representation to a :class:`~prov.model.ProvDocument` instance. :param stream: Input data. zutf-8rNz //comment())rrrBytesIOrreadencodeseekrparsegetrootZxpathZ getparentremover/rDZ ProvDocumentdeserialize_subtree)r r!r"bufxml_doccprr r r deserializes     zProvXMLSerializer.deserializec Csx |D]}tj|}|jtdjkr0td|jdkrHtjdt qt d}||j krd|j |nd}|dk rzt ||}|jdkr|j |d}|j||qt|}t|j} t| } td |j krt ||j td } |jtd | f|j| ||} | | kr| j| qW|S) a Deserialize an etree element containing a PROV document or a bundle and write it to the provided internal object. :param xml_doc: An etree element containing the information to read. :param bundle: The bundle object to write to. r/z2Non PROV element discovered in document or bundle.otherz[Document contains non-PROV information in . It will be ignored in this package.r,Nr*)r:r-)rQNamerBrr%r localnamewarningswarn UserWarningr8r<xml_qname_to_QualifiedNamerre_extract_attributesFULL_PROV_RECORD_IDS_MAP PROV_BASE_CLSrHappendrGZ new_recordZadd_asserted_type) r rgrrqnameZid_tagZrec_idbr?Z q_prov_namerXrKZrecr r r res:        z%ProvXMLSerializer.deserialize_subtreecCslt|}x^t|D]R\}}|tkr$qt|tjjr8|j}|tkrt||kr|j ||ft|}PqW|S)aD Helper function trying to derive the record label taking care of subtypes and what not. It will also remove the type declaration for the attributes if it was used to specialize the type. :param rec_type: The type of records. :param attributes: The attributes of the record. ) rr>rRrr/rDrErKrtrd)r rXr?rZrVrKr r r r@*s z&ProvXMLSerializer._derive_record_label)F)NF) r rr__doc__r$rrjrer@r r r r r"s ! 8rc Csg}x|D]}tj|}t|d|j|jf}x|jjD]\}}|tdkrt||}|tkrnt||j }qt j j |j |}q:|t dkrt||}q:|tdkrt j j |j |d}q:tjd|t|t|ftq:W|js|j }|j||fq W|S)zr Extract the PROV attributes from an etree element. :param element: The lxml.etree.Element instance. z%s:%sr-r1r.)rIzThe element '%s' contains an attribute %s='%s' which is not representable in the prov module's internal data model and will thus be ignored.)rrlrqr&rmr<r5rHZ XSD_QNAMErUr/rDrEr8rJrnror;rpru) rr?ZsubelZsqnameZ_trVrKrFZ_vr r r rrAs.        rrcCsd|kr\|jdd\}}||jkr\|j|}|tkr:t}n|tjkrJt}n t||}||Sd|jkr|jd}td|}||Std|dS)N:z/Could not create a valid QualifiedName for "%s")splitr+ XML_XSD_URIZXSDrGr%Z Namespacer )rZ qname_strr&rCZns_urir'r r r rqjs        rqcCs d||fS)Nz{%s}%sr )r'tagr r r rAsrAcCsttdj|S)Nr/)rArr%)r~r r r r8sr8cCsttdj|S)Nxsi)rArr%)r~r r r rHsrHcCsd}t||S)Nz$http://www.w3.org/XML/1998/namespace)rA)r~ZNS_XMLr r r rJsrJ)#rLlogginglxmlrrrnr/Zprov.identifierZ prov.modelrrZprov.constantsZprov.serializersr __author__ __email__ getLoggerr loggerdictZ PROV_N_MAPrupdateZADDITIONAL_N_MAPrsr}Errorr rrrrqrAr8rHrJr r r r s6    !)