Merge pull request #2 from versae/master

Adding a filter option when dumping or saving an ontology and its individuals

driver.py

@@ -61,7 +61,7 @@ def context_2_user_context(self, context): raise NotImplementedError
 
   def parse(self, f): raise NotImplementedError
 
-  def save(self, f, format = "pretty-xml"): raise NotImplementedError
+  def save(self, f, format = "pretty-xml", filter=None): raise NotImplementedError
 
   def _abbreviate  (self, iri, create_if_missing = True): return iri
   def _unabbreviate(self, iri): return iri
@@ -238,11 +238,13 @@ def _guess_format(f):
   return "rdfxml"
 
 
-def _save(f, format, graph):
+def _save(f, format, graph, filter=None):
   if   format == "ntriples":
     _unabbreviate = lru_cache(None)(graph._unabbreviate)
 
     for s,p,o,d in graph._iter_triples():
+      if filter and callable(filter) and not filter(graph, s, p, o, d):
+        continue
       if   s < 0: s = "_:%s" % (-s)
       else:       s = "<%s>" % _unabbreviate(s)
       p = "<%s>" % _unabbreviate(p)
@@ -263,6 +265,8 @@ def _save(f, format, graph):
     c_2_iri = { c : iri for c, iri in graph._iter_ontology_iri() }
 
     for c,s,p,o,d in graph._iter_triples(True):
+      if filter and callable(filter) and not filter(graph, s, p, o, d, c):
+        continue
       if   s < 0: s = "_:%s" % (-s)
       else:       s = "<%s>" % _unabbreviate(s)
       p = "<%s>" % _unabbreviate(p)
@@ -423,6 +427,8 @@ def purge():
     s_lines   = []
     current_s = ""
     for s,p,o,d in graph._iter_triples(False, True):
+      if filter and callable(filter) and not filter(graph, s, p, o, d):
+        continue
       if s != current_s:
         if current_s: purge()
         current_s = s
@@ -518,3 +524,4 @@ def flatten(l, deep = ""):
 
 
     f.write(b"""\n\n</rdf:RDF>\n""")
+