[PATCH 4 of 5] procutil: unroll uin/uout loop in protectstdio()

Yuya Nishihara yuya at tcha.org
Sun Mar 25 08:54:15 EDT 2018


# HG changeset patch
# User Yuya Nishihara <yuya at tcha.org>
# Date 1521947238 -32400
#      Sun Mar 25 12:07:18 2018 +0900
# Node ID 768b03c733d00da88fa99a698c0274ec5aa2cf5a
# Parent  a315bc2695f7faff23e71acea37d2c4a13961b14
procutil: unroll uin/uout loop in protectstdio()

I'll change uout to be redirected to stderr.

diff --git a/mercurial/utils/procutil.py b/mercurial/utils/procutil.py
--- a/mercurial/utils/procutil.py
+++ b/mercurial/utils/procutil.py
@@ -221,17 +221,18 @@ def protectstdio(uin, uout):
     "owned" in that print(), exec(), etc. never reach to them.
     """
     uout.flush()
-    newfiles = []
     nullfd = os.open(os.devnull, os.O_RDWR)
-    for f, sysf, mode in [(uin, stdin, r'rb'),
-                          (uout, stdout, r'wb')]:
-        if f is sysf:
-            newfd = os.dup(f.fileno())
-            os.dup2(nullfd, f.fileno())
-            f = os.fdopen(newfd, mode)
-        newfiles.append(f)
+    fin, fout = uin, uout
+    if uin is stdin:
+        newfd = os.dup(uin.fileno())
+        os.dup2(nullfd, uin.fileno())
+        fin = os.fdopen(newfd, r'rb')
+    if uout is stdout:
+        newfd = os.dup(uout.fileno())
+        os.dup2(nullfd, uout.fileno())
+        fout = os.fdopen(newfd, r'wb')
     os.close(nullfd)
-    return tuple(newfiles)
+    return fin, fout
 
 def restorestdio(uin, uout, fin, fout):
     """Restore (uin, uout) streams from possibly duplicated (fin, fout)"""


More information about the Mercurial-devel mailing list