Repository: Parquery/temppathlib

Issue:
Support `prefix` option also with `TmpDirIfNecessary`?
Maybe `TmpDirIfNecessary` should also support the `prefix` option, to give a temporary directory a certain prefix (if no directory is provided already?

Comments:

Patch:
diff --git a/pylint.rc b/pylint.rc
index a70a9ba..9ecb8f0 100644
--- a/pylint.rc
+++ b/pylint.rc
@@ -7,5 +7,5 @@ generated-members=bottle\.request\.forms\.decode,bottle\.request\.query\.decode
 max-line-length=120
 
 [MESSAGES CONTROL]
-disable=too-few-public-methods,abstract-class-little-used,len-as-condition,bad-continuation,bad-whitespace
+disable=too-few-public-methods,abstract-class-little-used,len-as-condition,bad-continuation,bad-whitespace,too-many-arguments
 
diff --git a/temppathlib/__init__.py b/temppathlib/__init__.py
index 442e321..e23b513 100644
--- a/temppathlib/__init__.py
+++ b/temppathlib/__init__.py
@@ -41,18 +41,32 @@ class TmpDirIfNecessary:
     def __init__(self,
                  path: Union[None, str, pathlib.Path],
                  base_tmp_dir: Union[None, str, pathlib.Path] = None,
-                 dont_delete_tmp_dir: bool = False) -> None:
+                 dont_delete_tmp_dir: bool = False,
+                 prefix: Optional[str] = None,
+                 suffix: Optional[str] = None) -> None:
         """
         Initialize with the given values.
 
-        :param path: provided path to the directory; if specified, no temporary directory is created.
-        :param base_tmp_dir: parent directory of the temporary directories; if not set,
-        the default is used (usually '/tmp'). This path is only used if a temporary directory needs to be created
-        and has no effect if 'path' was provided.
+        :param path:
+            provided path to the directory; if specified, no temporary directory is created.
+
+        :param base_tmp_dir:
+            parent directory of the temporary directories; if not set,
+            the default is used (usually '/tmp'). This path is only used if a temporary directory needs to be created
+            and has no effect if 'path' was provided.
+
+        :param dont_delete_tmp_dir:
+            if set, the temporary directory is not deleted upon close.
 
-        :param dont_delete_tmp_dir: if set, the temporary directory is not deleted upon close.
+            If the 'path' was provided, this argument has no effect.
 
-        If the 'path' was provided, this argument has no effect.
+        :param prefix:
+            If 'prefix' is not None, the name will begin with that prefix,
+            otherwise a default prefix is used.
+
+        :param suffix:
+            If 'suffix' is not None, the name will end with that suffix,
+            otherwise a default suffix is used.
         """
         if base_tmp_dir is None:
             self.base_tmp_dir = base_tmp_dir
@@ -76,6 +90,9 @@ class TmpDirIfNecessary:
 
         self.dont_delete = dont_delete_tmp_dir
 
+        self._prefix = prefix
+        self._suffix = suffix
+
         self.__use_tmp_dir = path is None
 
         self.exited = False
@@ -96,9 +113,10 @@ class TmpDirIfNecessary:
 
         if self._path is None:
             if self.base_tmp_dir is None:
-                self._path = pathlib.Path(tempfile.mkdtemp())
+                self._path = pathlib.Path(tempfile.mkdtemp(prefix=self._prefix, suffix=self._suffix))
             else:
-                self._path = pathlib.Path(tempfile.mkdtemp(dir=str(self.base_tmp_dir)))
+                self._path = pathlib.Path(
+                    tempfile.mkdtemp(dir=str(self.base_tmp_dir), prefix=self._prefix, suffix=self._suffix))
         else:
             self._path.mkdir(exist_ok=True, parents=True)
 
@@ -213,7 +231,6 @@ class NamedTemporaryFile:
 
         :param delete: whether the file is deleted on close (default True).
         """
-        # pylint: disable=too-many-arguments
         self.__tmpfile = tempfile.NamedTemporaryFile(
             mode=mode,
             buffering=buffering,


Test patch:
diff --git a/tests/test_temppathlib.py b/tests/test_temppathlib.py
index 6cd1e7c..cdf6988 100644
--- a/tests/test_temppathlib.py
+++ b/tests/test_temppathlib.py
@@ -94,6 +94,14 @@ class TestTmpDirIfNecessary(unittest.TestCase):
         finally:
             shutil.rmtree(str(basedir))
 
+    def test_prefix(self) -> None:
+        with temppathlib.TmpDirIfNecessary(path=None, prefix="some_prefix") as tmp_dir:
+            self.assertTrue(tmp_dir.path.name.startswith("some_prefix"))
+
+    def test_suffix(self) -> None:
+        with temppathlib.TmpDirIfNecessary(path=None, suffix="some_suffix") as tmp_dir:
+            self.assertTrue(tmp_dir.path.name.endswith("some_suffix"))
+
 
 class TestTemporaryDirectory(unittest.TestCase):
     def test_that_it_works(self) -> None: