tests/printenv.py

tests: treat `select` as a built-in module on Windows

This fixes:

    --- C:/Users/Matt/hg/tests/test-check-module-imports.t
    +++ C:/Users/Matt/hg/tests/test-check-module-imports.t.err
    @@ -43,3 +43,15 @@
       > -X tests/test-verify-repo-operations.py \
       > -X tests/test-extension.t \
       > | sed 's-\\-/-g' | "$PYTHON" "$import_checker" -
    +  hgext/zeroconf/Zeroconf.py:86: stdlib import "socket" follows local import: select\r (esc)
    +  hgext/zeroconf/Zeroconf.py:87: stdlib import "struct" follows local import: select\r (esc)
    +  hgext/zeroconf/Zeroconf.py:88: stdlib import "threading" follows local import: select\r (esc)
    +  hgext/zeroconf/Zeroconf.py:89: stdlib import "time" follows local import: select\r (esc)
    +  hgext/zeroconf/Zeroconf.py:90: stdlib import "traceback" follows local import: select\r (esc)
    +  mercurial/posix.py:18: stdlib import "stat" follows local import: select\r (esc)
    +  mercurial/posix.py:19: stdlib import "sys" follows local import: select\r (esc)
    +  mercurial/posix.py:20: stdlib import "tempfile" follows local import: select\r (esc)
    +  mercurial/posix.py:21: stdlib import "typing" follows local import: select\r (esc)
    +  tests/tinyproxy.py:19: stdlib import "socket" follows local import: select\r (esc)
    +  tests/tinyproxy.py:20: stdlib import "sys" follows local import: select\r (esc)
    +  [1]

    ERROR: test-check-module-imports.t output changed

#!/usr/bin/env python3
#
# simple script to be used in hooks
#
# put something like this in the repo .hg/hgrc:
#
#     [hooks]
#     changegroup = python "$TESTDIR/printenv.py" <hookname> [exit] [output]
#
#   - <hookname> is a mandatory argument (e.g. "changegroup")
#   - [exit] is the exit code of the hook (default: 0)
#   - [output] is the name of the output file (default: use sys.stdout)
#              the file will be opened in append mode.
#
import argparse
import os
import sys

try:
    import msvcrt

    msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
    msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
    msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
except ImportError:
    pass

parser = argparse.ArgumentParser()
parser.add_argument("name", help="the hook name, used for display")
parser.add_argument(
    "exitcode",
    nargs="?",
    default=0,
    type=int,
    help="the exit code for the hook",
)
parser.add_argument(
    "out", nargs="?", default=None, help="where to write the output"
)
parser.add_argument(
    "--line",
    action="store_true",
    help="print environment variables one per line instead of on a single line",
)
args = parser.parse_args()

if args.out is None:
    out = sys.stdout
    out = getattr(out, "buffer", out)
else:
    out = open(args.out, "ab")

# variables with empty values may not exist on all platforms, filter
# them now for portability sake.
env = [(k, v) for k, v in os.environ.items() if k.startswith("HG_") and v]
env.sort()

out.write(b"%s hook: " % args.name.encode('ascii'))
if os.name == 'nt':
    filter = lambda x: x.replace('\\', '/')
else:
    filter = lambda x: x

vars = [
    b"%s=%s" % (k.encode('ascii'), filter(v).encode('ascii')) for k, v in env
]

# Print variables on out
if not args.line:
    out.write(b" ".join(vars))
else:
    for var in vars:
        out.write(var)
        out.write(b"\n")

out.write(b"\n")
out.close()

sys.exit(args.exitcode)