prehistoric-systems/fail2ban
1
0
Fork 0
mirror of https://github.com/fail2ban/fail2ban.git synced 2026-03-11 08:55:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use tempfile contextmanager in with_tmpdir

Browse source
This commit is contained in:
Tomer Shalev 2022-08-07 08:44:44 +03:00
parent 3c22ae161c
commit 5045f05107
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
fail2ban/tests/utils.py
View file

@ -235,12 +235,8 @@ def with_tmpdir(f):
"""
@wraps(f)
def wrapper(self, *args, **kwargs):
tmp = tempfile.mkdtemp(prefix="f2b-temp")
try:
with tempfile.TemporaryDirectory(prefix="f2b-temp") as tmp:
return f(self, tmp, *args, **kwargs)
finally:
# clean up
shutil.rmtree(tmp)
return wrapper
def with_alt_time(f):