Adiciona o parâmetro pretty_print=False para tostring() em XMLWithPre…

… e PidProviderXMLAdapter (#539)
scieloorg · Jan 25, 2024 · 14a18eb · 14a18eb
1 parent 6edce03
commit 14a18eb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/packtools/sps/pid_provider/xml_sps_adapter.py b/packtools/sps/pid_provider/xml_sps_adapter.py
@@ -13,8 +13,8 @@ def __init__(self, xml_with_pre, pkg_name=None):
         self.xml_with_pre = xml_with_pre
         self.pkg_name = pkg_name
 
-    def tostring(self):
-        return self.xml_with_pre.tostring()
+    def tostring(self, pretty_print=False):
+        return self.xml_with_pre.tostring(pretty_print=pretty_print)
 
     @property
     def sps_pkg_name(self):

diff --git a/packtools/sps/pid_provider/xml_sps_lib.py b/packtools/sps/pid_provider/xml_sps_lib.py
@@ -338,8 +338,12 @@ def article_id_parent(self):
             front.append(parent)
             return parent
 
-    def tostring(self):
-        return self.xmlpre + etree.tostring(self.xmltree, encoding="utf-8").decode(
+    def tostring(self, pretty_print=False):
+        return self.xmlpre + etree.tostring(
+            self.xmltree,
+            encoding="utf-8",
+            pretty_print=pretty_print,
+        ).decode(
             "utf-8"
         )