53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
|
from pathlib import Path
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from texteller.logger import get_logger
|
||
|
|
|
||
|
|
_logger = get_logger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def resolve_path(path: str | Path) -> str:
|
||
|
|
if isinstance(path, str):
|
||
|
|
path = Path(path)
|
||
|
|
return str(path.expanduser().resolve())
|
||
|
|
|
||
|
|
|
||
|
|
def touch(path: str | Path) -> None:
|
||
|
|
if isinstance(path, str):
|
||
|
|
path = Path(path)
|
||
|
|
path.touch(exist_ok=True)
|
||
|
|
|
||
|
|
|
||
|
|
def mkdir(path: str | Path) -> None:
|
||
|
|
if isinstance(path, str):
|
||
|
|
path = Path(path)
|
||
|
|
path.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
|
||
|
|
def rmfile(path: str | Path) -> None:
|
||
|
|
if isinstance(path, str):
|
||
|
|
path = Path(path)
|
||
|
|
path.unlink(missing_ok=False)
|
||
|
|
|
||
|
|
|
||
|
|
def rmdir(path: str | Path, mode: Literal["empty", "recursive"] = "empty") -> None:
|
||
|
|
"""Remove a directory.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
path: Path to directory to remove
|
||
|
|
mode: "empty" to only remove empty directories, "all" to recursively remove all contents
|
||
|
|
"""
|
||
|
|
if isinstance(path, str):
|
||
|
|
path = Path(path)
|
||
|
|
|
||
|
|
if mode == "empty":
|
||
|
|
path.rmdir()
|
||
|
|
_logger.info(f"Removed empty directory: {path}")
|
||
|
|
elif mode == "recursive":
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
shutil.rmtree(path)
|
||
|
|
_logger.info(f"Recursively removed directory and all contents: {path}")
|
||
|
|
else:
|
||
|
|
raise ValueError(f"Invalid mode: {mode}. Must be 'empty' or 'all'")
|