gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2026-04-28 11:53:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: use tempfile.TemporaryFile as context manager, drop tfile=None guard

Browse Source
This commit is contained in:
Willi Ballenthin
2026-04-22 16:19:24 +03:00
committed by Willi Ballenthin
parent fe4fece1bc
commit 30b97dc9af
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
capa/helpers.py
View File

@@ -142,21 +142,15 @@ def stdout_redirector(stream):
# Save a copy of the original stdout fd in saved_stdout_fd
saved_stdout_fd = os.dup(original_stdout_fd)
tfile = None
try:
# Create a temporary file and redirect stdout to it
tfile = tempfile.TemporaryFile(mode="w+b")
_redirect_stdout(tfile.fileno())
# Yield to caller, then redirect stdout back to the saved fd
yield
_redirect_stdout(saved_stdout_fd)
# Copy contents of temporary file to the given stream
tfile.flush()
tfile.seek(0, io.SEEK_SET)
stream.write(tfile.read())
with tempfile.TemporaryFile(mode="w+b") as tfile:
_redirect_stdout(tfile.fileno())
yield
_redirect_stdout(saved_stdout_fd)
tfile.flush()
tfile.seek(0, io.SEEK_SET)
stream.write(tfile.read())
finally:
if tfile is not None:
tfile.close()
os.close(saved_stdout_fd)