[tools] Set origin_file_name on file load

Author: mpsonntag

odml/tools/odmlparser.py

@@ -10,6 +10,8 @@
 import sys
 import yaml
 
+from os.path import basename
+
 from . import xmlparser
 from .dict_parser import DictWriter, DictReader
 from ..info import FORMAT_VERSION
@@ -131,6 +133,8 @@ def from_file(self, file, doc_format=None):
                     return
 
             self.doc = DictReader().to_odml(self.parsed_doc)
+            # Provide original file name via the in memory document
+            self.doc._origin_file_name = basename(file)
             return self.doc
 
         elif self.parser == 'JSON':
@@ -142,6 +146,8 @@ def from_file(self, file, doc_format=None):
                     return
 
             self.doc = DictReader().to_odml(self.parsed_doc)
+            # Provide original file name via the in memory document
+            self.doc._origin_file_name = basename(file)
             return self.doc
 
         elif self.parser == 'RDF':

odml/tools/rdf_converter.py

@@ -222,7 +222,11 @@ def to_odml(self):
 
     def from_file(self, filename, doc_format):
         self.g = Graph().parse(source=filename, format=doc_format)
-        return self.to_odml()
+        docs = self.to_odml()
+        for d in docs:
+            # Provide original file name via the document
+            d._origin_file_name = os.path.basename(filename)
+        return docs
 
     def from_string(self, file, doc_format):
         self.g = Graph().parse(source=StringIO(file), format=doc_format)

odml/tools/xmlparser.py

@@ -10,6 +10,7 @@
 from lxml.builder import E
 # this is needed for py2exe to include lxml completely
 from lxml import _elementpath as _dummy
+from os.path import basename
 
 try:
     from StringIO import StringIO
@@ -187,7 +188,12 @@ def from_file(self, xml_file):
             raise ParserException(e.msg)
 
         self._handle_version(root)
-        return self.parse_element(root)
+        doc = self.parse_element(root)
+
+        # Provide original file name via the in memory document
+        if isinstance(xml_file, unicode):
+            doc._origin_file_name = basename(xml_file)
+        return doc
 
     def from_string(self, string):
         try: