Patchwork [4,of,8] util: add public isstdin/isstdout() functions

login
register
mail settings
Submitter Yuya Nishihara
Date March 9, 2018, 12:35 p.m.
Message ID <10e3ea3d8b1dc671f630.1520598937@mimosa>
Download mbox | patch
Permalink /patch/29164/
State Accepted
Headers show

Comments

Yuya Nishihara - March 9, 2018, 12:35 p.m.
# HG changeset patch
# User Yuya Nishihara <yuya@tcha.org>
# Date 1520324939 21600
#      Tue Mar 06 02:28:59 2018 -0600
# Node ID 10e3ea3d8b1dc671f6302ebed4489773c7d79458
# Parent  c9ac3a3b0a31abc37a76e5ae3be1b5f14a206cf3
util: add public isstdin/isstdout() functions

Patch

diff --git a/mercurial/util.py b/mercurial/util.py
--- a/mercurial/util.py
+++ b/mercurial/util.py
@@ -1430,13 +1430,19 @@  def _sethgexecutable(path):
     global _hgexecutable
     _hgexecutable = path
 
-def _isstdout(f):
+def _testfileno(f, stdf):
     fileno = getattr(f, 'fileno', None)
     try:
-        return fileno and fileno() == sys.__stdout__.fileno()
+        return fileno and fileno() == stdf.fileno()
     except io.UnsupportedOperation:
         return False # fileno() raised UnsupportedOperation
 
+def isstdin(f):
+    return _testfileno(f, sys.__stdin__)
+
+def isstdout(f):
+    return _testfileno(f, sys.__stdout__)
+
 def shellenviron(environ=None):
     """return environ with optional override, useful for shelling out"""
     def py2shell(val):
@@ -1464,7 +1470,7 @@  def system(cmd, environ=None, cwd=None, 
         pass
     cmd = quotecommand(cmd)
     env = shellenviron(environ)
-    if out is None or _isstdout(out):
+    if out is None or isstdout(out):
         rc = subprocess.call(cmd, shell=True, close_fds=closefds,
                              env=env, cwd=cwd)
     else: