Skip to content

Commit 2d7770a

Browse files
committed
Apply more suggestions by @kvid
1 parent 426c11b commit 2d7770a

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/wireviz/svgembed.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import re
55
import base64
66
from pathlib import Path
7+
from typing import Union
78

89
mime_subtype_replacements = {'jpg': 'jpeg', 'tif': 'tiff'}
910

@@ -25,15 +26,15 @@ def embed_svg_images(svg_in: str, base_path: Path):
2526
return svg_out
2627

2728

28-
def get_mime_subtype(filename: Path):
29-
mime_subtype = filename.suffix.lstrip('.').lower()
29+
def get_mime_subtype(filename: Union[str, Path]):
30+
mime_subtype = Path(filename).suffix.lstrip('.').lower()
3031
if mime_subtype in mime_subtype_replacements:
3132
mime_subtype = mime_subtype_replacements[mime_subtype]
3233
return mime_subtype
3334

3435

35-
def embed_svg_images_file(filename_in: Path, overwrite: bool = True):
36-
filename_in = filename_in.resolve()
36+
def embed_svg_images_file(filename_in: Union[str, Path], overwrite: bool = True):
37+
filename_in = Path(filename_in).resolve()
3738
filename_out = filename_in.with_suffix('.b64.svg')
3839
filename_out.write_text(embed_svg_images(filename_in.read_text(), filename_in.parent))
3940
if overwrite:

0 commit comments

Comments
 (0)