allow Paths in save()

altair_saver/_utils.py

@@ -2,6 +2,7 @@
 from http import client
 import io
 import os
+import pathlib
 import socket
 import subprocess
 import sys
@@ -134,8 +135,9 @@ def temporary_filename(
 
 @contextlib.contextmanager
 def maybe_open(fp: Union[IO, str], mode: str = "w") -> Iterator[IO]:
-    """Context manager to write to a file specified by filename or file-like object"""
-    if isinstance(fp, str):
+    """Context manager to write to a file specified by filename, Path or
+    file-like object"""
+    if isinstance(fp, str) or isinstance(fp, pathlib.PurePath):
         with open(fp, mode) as f:
             yield f
     elif isinstance(fp, io.TextIOBase) and "b" in mode:
@@ -151,10 +153,12 @@ def maybe_open(fp: Union[IO, str], mode: str = "w") -> Iterator[IO]:
 
 
 def extract_format(fp: Union[IO, str]) -> str:
-    """Extract the altair_saver output format from a file or filename."""
+    """Extract the altair_saver output format from a file, filename or Path."""
     filename: Optional[str]
     if isinstance(fp, str):
         filename = fp
+    elif isinstance(fp, pathlib.PurePath):
+        filename = str(fp)
     else:
         filename = getattr(fp, "name", None)
     if filename is None: