GeezerVison

from pathlib import Path
import shutil
from typing import AnyStr, Generic


class DummyPathLike(Generic[AnyStr]):
    """Since pathlib.Path can't accept bytes itself."""

    def __init__(self, path: AnyStr) -> None:
        self._path: AnyStr = path

    def __fspath__(self) -> AnyStr:
        """Return the file system path representation of the object."""
        return self._path

Path("some_folder").mkdir()
Path("some_folder/test.txt").touch()

shutil.make_archive(base_name="my_archive", format="gztar", root_dir="some_folder", base_dir=".")

shutil.unpack_archive(
    filename="my_archive.tar.gz",
    extract_dir="extract_folder",
    format="gztar",
)
shutil.unpack_archive(
    filename=b"my_archive.tar.gz",
    extract_dir="extract_folder",
    format="gztar",
)
shutil.unpack_archive(
    filename=Path("my_archive.tar.gz"),
    extract_dir="extract_folder",
    format="gztar",
)
shutil.unpack_archive(
    filename=DummyPathLike(b"my_archive.tar.gz"),
    extract_dir="extract_folder",
    format="gztar",
)


Path("some_folder/test.txt").unlink()
Path("my_archive.tar.gz").unlink()
Path("extract_folder/test.txt").unlink()
Path("extract_folder").rmdir()

# N.B. these versions below fail with TypeError: `Can't mix strings and bytes in path components`
shutil.unpack_archive(
    filename="my_archive.tar.gz",
    extract_dir=b"extract_folder",
    format="gztar",
)
shutil.unpack_archive(
    filename="my_archive.tar.gz",
    extract_dir=DummyPathLike(b"extract_folder"),
    format="gztar",
)