Minor refactoring and reformatting

Author: leonarduschen

pandas/io/common.py

@@ -17,6 +17,7 @@
 )
 import mmap
 import os
+from pathlib import Path
 from typing import (
     IO,
     Any,
@@ -519,19 +520,19 @@ def infer_compression(
     raise ValueError(msg)
 
 
-def check_parent_directory(fpath: str):
+def check_parent_directory(path: Path | str) -> None:
     """
     Check if parent directory of a file exists, raise OSError if it does not
 
     Parameters
     ----------
-    fpath: str
+    path: Path | str
         File path
 
     """
-    dirname = os.path.dirname(fpath)
-    if len(dirname) != 0 and not os.path.isdir(dirname):
-        raise OSError(fr"Cannot save file into a non-existent directory: '{dirname}'")
+    parent = Path(path).parent
+    if not parent.is_dir():
+        raise OSError(fr"Cannot save file into a non-existent directory: '{parent}'")
 
 
 def get_handle(
@@ -646,9 +647,9 @@ def get_handle(
     compression_args = dict(ioargs.compression)
     compression = compression_args.pop("method")
 
-    # GH 24306
-    if mode not in ["r", "rb"] and is_path:  # Only for write methods
-        check_parent_directory(str(handle))
+    # Only for write methods
+    if "r" not in mode and is_path:
+        check_parent_directory(handle)
 
     if compression:
         # compression libraries do not like an explicit text-mode

pandas/io/formats/format.py

@@ -1150,7 +1150,7 @@ def get_buffer(buf: FilePathOrBuffer[str] | None, encoding: str | None = None):
     if hasattr(buf, "write"):
         yield buf
     elif isinstance(buf, str):
-        check_parent_directory(str(buf))
+        check_parent_directory(buf)
         with open(buf, "w", encoding=encoding, newline="") as f:
             # GH#30034 open instead of codecs.open prevents a file leak
             #  if we have an invalid encoding argument.