improved @tmpdir decorator
All checks were successful
CI / build (push) Successful in 12s

This commit is contained in:
2024-06-24 21:24:48 +08:00
parent 436bd737fa
commit e89e306e96

View File

@@ -155,10 +155,22 @@ def async_test(coro):
@decorator_with_kwargs @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) @wraps(f)
def result(*args, **kwargs): 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, **{ f(*args, **kwargs, **{
argument_name: Path(temp_dir) argument_name: Path(temp_dir)
}) })