ext2rdf/src/RDFConverter/TripleStructureConverter.py at master · Weissger/ext2rdf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
__author__ = 'tmy'

from rdflib import Graph, Literal, Namespace, RDFS, URIRef
import urllib.parse as urllib
import logging
import pandas as pd
from ext2rdf.src.Utilities.Constants import LOG_LEVEL, NAMESPACE
from ext2rdf.src.RDFConverter.AbstractConverter import AbstractConverter

log = logging.getLogger()
log.setLevel(LOG_LEVEL)


class Converter(AbstractConverter):

    def __init__(self):
        self.namespace = Namespace(NAMESPACE)

    def convert(self, data_frame):
        log.info("converting data_frame...")
        graph = Graph()
        for _, row in data_frame.iterrows():

            # Subject
            subject_uri = urllib.quote(row['Subject'], '')
            subject_node = URIRef(self.namespace[subject_uri])
            graph.add((subject_node, RDFS.label, Literal(row['Subject'])))

            # Predicate
            predicate_uri = urllib.quote(row['Predicate'], '')
            predicate_node = URIRef(self.namespace[predicate_uri])
            graph.add((predicate_node, RDFS.label, Literal(row['Predicate'])))

            # Object
            object_uri = urllib.quote(row['Object'], '')
            object_node = URIRef(self.namespace[object_uri])
            graph.add((object_node, RDFS.label, Literal(row['Object'])))

            graph.add((subject_node, predicate_node, object_node))

            # Predicate Lemmatized
            if not pd.isnull(row['PredicateLemma']):
                predicate_lemma_uri = urllib.quote(row['PredicateLemma'], '')
                predicate_lemma_node = URIRef(self.namespace[predicate_lemma_uri])
                graph.add((subject_node, predicate_lemma_node, object_node))

        return graph