3 dK&@sddlZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZdZdZe ee ee ee eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeiZ ddZ!ddZ"dS) N) ProvDocument ProvRecord ProvElement ProvEntity ProvActivity ProvAgent ProvRelationPROV_ATTR_ENTITYPROV_ATTR_ACTIVITYPROV_ATTR_AGENTPROV_ATTR_TRIGGERPROV_ATTR_GENERATED_ENTITYPROV_ATTR_USED_ENTITYPROV_ATTR_DELEGATEPROV_ATTR_RESPONSIBLEPROV_ATTR_SPECIFIC_ENTITYPROV_ATTR_GENERAL_ENTITYPROV_ATTR_ALTERNATE1PROV_ATTR_ALTERNATE2PROV_ATTR_COLLECTIONPROV_ATTR_INFORMEDPROV_ATTR_INFORMANTPROV_ATTR_BUNDLEPROV_ATTR_PLANPROV_ATTR_ENDERPROV_ATTR_STARTER ProvBundlezTrung Dong Huynhztrungdong@donggiang.comc Cstj}|j}t}x&|jtD]}|j||||j<q"Wx|jtD]}|j dd\}}|d|d}} |rJ| rJy@||krt |dd|||<| |krt |dd| || <Wnt k rwJYnX|j |||| |dqJW|S)aA Convert a :class:`~prov.model.ProvDocument` to a `MultiDiGraph `_ instance of the `NetworkX `_ library. :param prov_document: The :class:`~prov.model.ProvDocument` instance to convert. Nr)relation) nxZ MultiDiGraphunifieddictZ get_recordsradd_node identifierrZformal_attributesINFERRED_ELEMENT_CLASSKeyErrorZadd_edge) Z prov_documentgr!Znode_mapelementrZ attr_pair_1Z attr_pair_2Zqn1Zqn2r)*/tmp/pip-build-7vycvbft/prov/prov/graph.py prov_to_graph;s& r+c Cst}x.|jD]"}t|tr|jdk r|j|qWxP|jddD]@\}}}y |d}t|trl|j|WqDtk rYqDXqDW|S)a Convert a `MultiDiGraph `_ that was previously produced by :func:`prov_to_graph` back to a :class:`~prov.model.ProvDocument`. :param g: The graph instance to convert. NT)datar)rZnodes isinstancerZbundleZ add_recordedgesr&)r'Zprov_docn_Z edge_datarr)r)r* graph_to_prov\s   r1)#Znetworkxr Z prov.modelrrrrrrrr r r r r rrrrrrrrrrrrrrr __author__ __email__r%r+r1r)r)r)r*s0t!