mirror of https://github.com/fail2ban/fail2ban
Merge 5045f05107
into 86b9adb2f5
commit
ab536e6bec
|
@ -231,12 +231,8 @@ def with_tmpdir(f):
|
||||||
"""
|
"""
|
||||||
@wraps(f)
|
@wraps(f)
|
||||||
def wrapper(self, *args, **kwargs):
|
def wrapper(self, *args, **kwargs):
|
||||||
tmp = tempfile.mkdtemp(prefix="f2b-temp")
|
with tempfile.TemporaryDirectory(prefix="f2b-temp") as tmp:
|
||||||
try:
|
|
||||||
return f(self, tmp, *args, **kwargs)
|
return f(self, tmp, *args, **kwargs)
|
||||||
finally:
|
|
||||||
# clean up
|
|
||||||
shutil.rmtree(tmp)
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
def with_alt_time(f):
|
def with_alt_time(f):
|
||||||
|
|
Loading…
Reference in New Issue