save and write fixed, __repr__() and pretty() methods added

pull/114/head
Andy Port 2020-06-23 22:42:23 -07:00
parent 4b7f17c7bd
commit ab44fcd564
1 changed files with 7 additions and 3 deletions

View File

@ -253,7 +253,11 @@ class Document:
def paths(self, group_filter=lambda x: True, def paths(self, group_filter=lambda x: True,
path_filter=lambda x: True, path_conversions=CONVERSIONS): path_filter=lambda x: True, path_conversions=CONVERSIONS):
"""Returns a list of all paths in the document.""" """Returns a list of all paths in the document.
Note that any transform attributes are applied before returning
the paths.
"""
return flattened_paths(self.tree.getroot(), group_filter, return flattened_paths(self.tree.getroot(), group_filter,
path_filter, path_conversions) path_filter, path_conversions)
@ -418,7 +422,7 @@ class Document:
return parseString(repr(self)).toprettyxml(**kwargs) return parseString(repr(self)).toprettyxml(**kwargs)
def save(self, filepath, prettify=False, **kwargs): def save(self, filepath, prettify=False, **kwargs):
with open(filepath, 'w') as output_svg: with open(filepath, 'w+') as output_svg:
if prettify: if prettify:
output_svg.write(self.pretty(**kwargs)) output_svg.write(self.pretty(**kwargs))
else: else:
@ -438,6 +442,6 @@ class Document:
# write to a (by default temporary) file # write to a (by default temporary) file
with open(filepath, 'w') as output_svg: with open(filepath, 'w') as output_svg:
output_svg.write(self.as_string()) output_svg.write(repr(self))
open_in_browser(filepath) open_in_browser(filepath)