3
d0 @ s( d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl T d dl
m Z m Z m
Z
mZmZmZmZmZ d dlZejeZdZdZG d d
d
eZG d d d Zed
e diZ G dd deZ G dd dej Z G dd dej!Z"dd Z#dd Z$dd Z%dd Z&d d Z'd d Z(d!d" Z)d#d$ Z*dS )% ) defaultdictN)Error)
Serializer)*)Literal
Identifier
QualifiedName Namespace ProvDocument
ProvBundlefirstparse_xsd_datetimezTrung Dong Huynhztrungdong@donggiang.comc @ s e Zd ZdS )ProvJSONExceptionN)__name__
__module__ __qualname__ r r 9/tmp/pip-build-7vycvbft/prov/prov/serializers/provjson.pyr s r c @ s e Zd Zdd ZdddZdS )AnonymousIDGeneratorc C s i | _ d| _d S )Nr )_cache_count)selfr r r __init__! s z AnonymousIDGenerator.__init__idc C s: || j kr0| jd7 _td|| jf | j |< | j | S )N z_:%s%d)r r r )r objZ local_prefixr r r get_anon_id% s
z AnonymousIDGenerator.get_anon_idN)r )r r r r r r r r r r s r z
xsd:doublezxsd:intc @ s e Zd ZdZdd Zdd ZdS )ProvJSONSerializerzD
PROV-JSON serializer for :class:`~prov.model.ProvDocument`
c
K sr t j }zZtj| j|fdti| |jdd t|t jrL|j |j
n|j |j
j d W d|j X dS )z
Serializes a :class:`~prov.model.ProvDocument` instance to
`PROV-JSON `_.
:param stream: Where to save the output.
clsr zutf-8N)
ioStringIOjsondumpdocumentProvJSONEncoderseek
isinstance
TextIOBasewritereadencodeclose)r streamkwargsbufr r r serialize: s z ProvJSONSerializer.serializec K s: t |tjs$tj|j jd}|}tj|fdti|S )z
Deserialize from the `PROV JSON
`_ representation to a
:class:`~prov.model.ProvDocument` instance.
:param stream: Input data.
zutf-8r ) r& r r' r r) decoder! loadProvJSONDecoder)r r, r- r. r r r deserializeO s z ProvJSONSerializer.deserializeN)r r r __doc__r/ r3 r r r r r 5 s r c s e Zd Z fddZ ZS )r$ c s& t |trt|S tt| j|S d S )N)r& r
encode_json_documentsuperr$ r* )r o) __class__r r default^ s
zProvJSONEncoder.default)r r r r9
__classcell__r r )r8 r r$ ] s r$ c s e Zd Z fddZ ZS )r2 c s, t t| j|f||}t }t|| |S )N)r6 r2 r0 r
decode_json_document)r sargsr- containerr# )r8 r r r0 f s
zProvJSONDecoder.decode)r r r r0 r: r r )r8 r r2 e s r2 c C s |d kr d S | j |}|S )N)valid_qualified_name)bundlevalueZqualified_namer r r r? n s
r? c C s6 t | }x(| jD ] }t |}||d t|j< qW |S )Nr@ )encode_json_containerbundlesstr
identifier)r# r> r@ Z bundle_jsonr r r r5 u s
r5 c s t t}i }x | jj D ]}|j||j< qW | jjrB| jjj|d< |rN||d< t fdd}x | jD ]}|j }t
| }t ||}i } |j r(x|j j
D ]\}
} | sqt |
} |
tkrt t| | | < q|
tkrt| j | | < qt| dkrtt| | | < qtdd | D | | < qW ||| krD| || |< qj|| | }
t|
drj|
g|| |< || | j| qjW |S ) Nr9 prefixc s | j r | j S j| S )N)Z _identifierr )r) id_generatorr r real_or_anon_id s z.encode_json_container..real_or_anon_idr c s s | ]}t |V qd S )N)encode_json_representation).0rA r r r s z(encode_json_container..items)r dict _namespacesZget_registered_namespacesurirF _defaultr Z_recordsget_typeZ
PROV_N_MAPrD _attributesrM PROV_ATTRIBUTE_QNAMESr ZPROV_ATTRIBUTE_LITERALS isoformatlenrJ listhasattrappend)r@ r> prefixes namespacerI recordrec_typeZ rec_labelrE Z record_jsonattrvalues attr_nameZcurrent_contentr )rH r rB ~ sH rB c C sf t }d| kr | d }| d= t| | x:|j D ].\}}t|d}t|| |j||j| q0W d S )Nr@ )r# )rN decode_json_containerrM r Z
add_bundler? )contentr# rC Z bundle_idZbundle_contentr@ r r r r; s
r; c s d| krP| d }x8|j D ],\}}|dkr<jt|| qj| qW | d= x| D ]}t| }x| | j D ]\}}t|dr|g} n|} xb| D ]X}
t } g } d }
x|
j D ]\}}|tkrt| nt| t krzt
|t rPt |dkrF|t
kr. tkr.|dd }
|d }nd}tj| t|n|d }n|} tkrht|nt|}|| < qt
|t r| j fdd|D q| j t|f qW j||| | |
r| t }x |
D ]}j|t| qW qW qvW qXW d S ) NrF r9 rM r r zIThe prov package does not support PROV attributes having multiple values.c 3 s | ]} t |fV qd S )N)decode_json_representation)rK rA )r^ r@ r r rL
s z(decode_json_container..)rM Z
add_namespacer Zset_default_namespaceZPROV_RECORD_IDS_MAPrX rN ZPROV_ATTRIBUTES_ID_MAPr? ZPROV_ATTRIBUTESr& rW rV ZPROV_MEMBERSHIPZPROV_ATTR_ENTITYloggererrorr rT r
extendrY rc Z
new_recordZPROV_ATTR_COLLECTIONZ
membership)Zjcr@ rZ rF rP Z rec_type_strr] Zrec_idrb elementselement
attributesZother_attributesZmembership_extra_membersr` r_ rA Z error_msgZ
collectionmemberr )r^ r@ r ra s^
ra c C s t | trt| S t | tjr,| j ddS t | trFt| tjdS t | t r\| j
ddS t | t krz| t t | dS | S d S )Nz xsd:dateTime)$typez
xsd:anyURI)
r& r literal_json_representationdatetimerU r rD PROV_QUALIFIEDNAME_strr rP rl LITERAL_XSDTYPE_MAP)rA r r r rJ s
rJ c C s| t | trt| d }d| kr"| d nd }t||}d| kr@| d nd }|tkrTt|S |tkrft||S t|||S n| S d S )Nrk rl lang)r& rN r? Z
XSD_ANYURIr ro r )literalr@ rA datatypelangtagr r r rc 0 s
rc c C s6 | j | j| j }}}|r$||dS |t|dS d S )N)rk rr )rk rl )rA rt ru rD )rs rA rt ru r r r rm E s
rm )+ collectionsr rn r r! Zprovr Zprov.serializersr Zprov.constantsZ
prov.modelr r r r r
r r r
logging getLoggerr rd
__author__ __email__r r floatintrq r JSONEncoderr$ JSONDecoderr2 r? r5 rB r; ra rJ rc rm r r r r s4 (
( ;X