From fe6e608c15f61850453331a447aa83ef29edeaca Mon Sep 17 00:00:00 2001 From: Friederike Biermann Date: Mon, 22 May 2023 10:25:09 +0200 Subject: [PATCH] Update write_svg to return svg string if not called with out_file --- pikachu/drawing/drawing.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pikachu/drawing/drawing.py b/pikachu/drawing/drawing.py index d8d214e..0db98dc 100644 --- a/pikachu/drawing/drawing.py +++ b/pikachu/drawing/drawing.py @@ -1182,7 +1182,7 @@ def draw_svg(self, annotation: Union[None, str] = None, numbered_atoms: List = N svg = self.assemble_svg() return svg - def write_svg(self, out_file: str, annotation: Union[str, None] = None, numbered_atoms: List = None) -> None: + def write_svg(self, out_file: Union[str, None] = None, annotation: Union[str, None] = None, numbered_atoms: List = None) -> Union[str, None]: self.flip_y_axis() self.move_to_positive_coords() @@ -1216,7 +1216,9 @@ def write_svg(self, out_file: str, annotation: Union[str, None] = None, numbered svg_string += self.svg_style svg_string += self.draw_svg(annotation=annotation, numbered_atoms=numbered_atoms) svg_string += "" - + + if out_file is None: + return svg_string with open(out_file, 'w') as out: out.write(svg_string)