From e89e306e96ba5ee19d4b95b6c016afe40c414af0 Mon Sep 17 00:00:00 2001
From: Walter Oggioni <oggioni.walter@gmail.com>
Date: Mon, 24 Jun 2024 21:24:48 +0800
Subject: [PATCH] improved @tmpdir decorator

---
 src/pwo/private.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/pwo/private.py b/src/pwo/private.py
index f6fc84a..5209414 100644
--- a/src/pwo/private.py
+++ b/src/pwo/private.py
@@ -155,10 +155,22 @@ def async_test(coro):
 
 
 @decorator_with_kwargs
-def tmpdir(f, argument_name='tmpdir'):
+def tmpdir(f,
+           argument_name='temp_dir',
+           suffix=None,
+           prefix=None,
+           dir=None,
+           ignore_cleanup_errors=False,
+           delete=True):
+
     @wraps(f)
     def result(*args, **kwargs):
-        with TemporaryDirectory() as temp_dir:
+        with TemporaryDirectory(
+                suffix=suffix,
+                prefix=prefix,
+                dir=dir,
+                ignore_cleanup_errors=ignore_cleanup_errors,
+                delete=delete) as temp_dir:
             f(*args, **kwargs, **{
                 argument_name: Path(temp_dir)
             })