warehack.ing/WireViz
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use Pathlib to overwrite file

Browse Source
This commit is contained in:
Daniel Rojas 2020-11-15 12:30:23 +01:00
parent cb240af1f7
commit 973125cd75
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/wireviz/svgembed.py
View File

@ -4,13 +4,12 @@
import re import re
import base64 import base64
from pathlib import Path from pathlib import Path
import os
mime_subtype_replacements = {'jpg': 'jpeg', 'tif': 'tiff'} mime_subtype_replacements = {'jpg': 'jpeg', 'tif': 'tiff'}
def embed_svg_images(filename_in: Path, overwrite: bool = True): def embed_svg_images(filename_in: Path, overwrite: bool = True):
filename_in = Path(filename_in).resolve() 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 images_b64 = {} # cache base-64 encoded images
re_xlink=re.compile(r"xlink:href=\"(?P<URL>.*?)\"", re.IGNORECASE) re_xlink=re.compile(r"xlink:href=\"(?P<URL>.*?)\"", re.IGNORECASE)
with open(filename_in,'r') as file_in, open(filename_out,'w') as file_out: 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) file_out.write(line)
if overwrite: if overwrite:
os.remove(filename_in) filename_out.replace(filename_in)
os.rename(filename_out, filename_in)