From 973125cd751903fbba2a4834c393de0af20b5478 Mon Sep 17 00:00:00 2001 From: Daniel Rojas Date: Sun, 15 Nov 2020 12:30:23 +0100 Subject: [PATCH] Use Pathlib to overwrite file --- src/wireviz/svgembed.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/wireviz/svgembed.py b/src/wireviz/svgembed.py index 17984ab..313f932 100644 --- a/src/wireviz/svgembed.py +++ b/src/wireviz/svgembed.py @@ -4,13 +4,12 @@ import re import base64 from pathlib import Path -import os mime_subtype_replacements = {'jpg': 'jpeg', 'tif': 'tiff'} def embed_svg_images(filename_in: Path, overwrite: bool = True): filename_in = Path(filename_in).resolve() - filename_out = f'{filename_in.with_suffix("")}.b64.svg' + filename_out = Path(f'{filename_in.with_suffix("")}.b64.svg') images_b64 = {} # cache base-64 encoded images re_xlink=re.compile(r"xlink:href=\"(?P.*?)\"", re.IGNORECASE) with open(filename_in,'r') as file_in, open(filename_out,'w') as file_out: @@ -37,5 +36,4 @@ def embed_svg_images(filename_in: Path, overwrite: bool = True): file_out.write(line) if overwrite: - os.remove(filename_in) - os.rename(filename_out, filename_in) + filename_out.replace(filename_in)