[fileio/odmlparser] Add xmlparser kwargs option

To allow passing custom stylesheets for xml documents
also via the odml.save function, kwargs are now supported
in odml.save and the odmlparser has been adjusted accordingly.

Author: mpsonntag

odml/fileio.py

@@ -25,19 +25,21 @@ def load(filename, backend="xml", show_warnings=True):
     return reader.from_file(filename)
 
 
-def save(obj, filename, backend="xml"):
+def save(obj, filename, backend="xml", **kwargs):
     """
     Save an open odML document to file of a specified format.
     :param obj: odML document do be saved.
     :param filename: Filename and path where the odML document
                      should be saved.
     :param backend: Format in which the odML document is to be saved.
                     The default format is XML.
+    :param kwargs: Writer backend keyword arguments. Refer to the documentation
+                   of the available parsers to check which arguments are supported.
     """
     writer = ODMLWriter(backend)
     if "." not in filename.split(os.pathsep)[-1]:
         filename = filename + ".%s" % backend
-    return writer.write_file(obj, filename)
+    return writer.write_file(obj, filename, **kwargs)
 
 
 def display(obj, backend="xml"):

odml/tools/odmlparser.py

@@ -46,7 +46,7 @@ def __init__(self, parser='XML'):
 
         self.parser = parser
 
-    def write_file(self, odml_document, filename):
+    def write_file(self, odml_document, filename, **kwargs):
         """
         Writes an odml.Document to a file using the format
         defined in the ODMLWriter.parser property. Supported formats are
@@ -55,6 +55,8 @@ def write_file(self, odml_document, filename):
 
         :param odml_document: odml.Document.
         :param filename: path and filename of the output file.
+        :param kwargs: Writer backend keyword arguments. Refer to the documentation
+                       of the available parsers to check which arguments are supported.
         """
 
         # Write document only if it does not contain validation errors.
@@ -74,28 +76,34 @@ def write_file(self, odml_document, filename):
             msg += " Run the Documents 'validate' method to access them.\n%s" % report
             warnings.warn(msg)
 
-        with open(filename, 'w') as file:
-            # Add XML header to support odML stylesheets.
-            if self.parser == 'XML':
-                file.write(xmlparser.XMLWriter.header)
-
-            file.write(self.to_string(odml_document))
+        # Allow kwargs when writing XML documents to support individual style sheets
+        if self.parser == 'XML':
+            local_style = False
+            custom_template = None
+
+            if "local_style" in kwargs and isinstance(kwargs["local_style"], bool):
+                local_style = kwargs["local_style"]
+            if "custom_template" in kwargs and isinstance(kwargs["custom_template"], str):
+                custom_template = kwargs["custom_template"]
+            xmlparser.XMLWriter(odml_document).write_file(filename, local_style=local_style,
+                                                          custom_template=custom_template)
+        else:
+            with open(filename, 'w') as file:
+                file.write(self.to_string(odml_document))
 
     def to_string(self, odml_document):
         """
         Parses an odml.Document to a string in the file format
         defined in the ODMLWriter.parser property. Supported formats are
-        JSON, XML, YAML and RDF.
+        JSON, YAML and RDF.
 
         :param odml_document: odml.Document.
         :return: string containing the content of the odml.Document in the
                  specified format.
         """
         string_doc = ''
 
-        if self.parser == 'XML':
-            string_doc = unicode(xmlparser.XMLWriter(odml_document))
-        elif self.parser == "RDF":
+        if self.parser == "RDF":
             # Use XML as default output format for now.
             string_doc = RDFWriter(odml_document).get_rdf_str("xml")
         else: