SPath: Add append_to_stem (#4)

Setsugennoao · Jun 3, 2024 · 35c3125 · 35c3125
1 parent bd2a89f
commit 35c3125
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/stgpytools/types/file.py b/stgpytools/types/file.py
@@ -4,6 +4,8 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, Iterable, Literal, TypeAlias, Union
 
+from ..functions import to_arr
+
 __all__ = [
     'FilePathType', 'FileDescriptor',
     'FileOpener',
@@ -113,5 +115,10 @@ def write_lines(
 
         return super().write_text('\n'.join(data), encoding, errors, newline)
 
+    def append_to_stem(self, suffixes: str | Iterable[str], sep: str = '_') -> SPath:
+        """Append a suffix to the stem of the path"""
+
+        return self.with_stem(sep.join([self.stem, *to_arr(suffixes)]))  # type:ignore[list-item]
+
 
 SPathLike = Union[str, Path, SPath]