This is an archive of the discontinued Mercurial Phabricator instance.

Differential D5452

tests: update printenv.py argument parsing
ClosedPublic

Authored by lothiraldan on Dec 19 2018, 11:25 AM.

Details

Reviewers
None
Group Reviewers
hg-reviewers
Commits
rHG3c5aaea9638f: tests: update printenv.py argument parsing
rHG07b15f64e560: tests: update printenv.py argument parsing
Summary

We are about to introduce a new flag for printing the HG environment variables
one per line and it's easier to do when using the argparse module for argument
parsing.

Diff Detail

Repository
rHG Mercurial
Lint
Automatic diff as part of commit; lint not applicable.
Unit
Automatic diff as part of commit; unit tests not applicable.

Revision Contents
Changeset List

PathPackages
Mtests/printenv.py (31 lines)

Diff 12927

tests/printenv.py

#!/usr/bin/env python #!/usr/bin/env python
# #
# simple script to be used in hooks # simple script to be used in hooks
# #
# put something like this in the repo .hg/hgrc: # put something like this in the repo .hg/hgrc:
# #
# [hooks] # [hooks]
# changegroup = python "$TESTDIR/printenv.py" <hookname> [exit] [output] # changegroup = python "$TESTDIR/printenv.py" <hookname> [exit] [output]
# #
# - <hookname> is a mandatory argument (e.g. "changegroup") # - <hookname> is a mandatory argument (e.g. "changegroup")
# - [exit] is the exit code of the hook (default: 0) # - [exit] is the exit code of the hook (default: 0)
# - [output] is the name of the output file (default: use sys.stdout) # - [output] is the name of the output file (default: use sys.stdout)
# the file will be opened in append mode. # the file will be opened in append mode.
# #
from __future__ import absolute_import from __future__ import absolute_import
import argparse
import os import os
import sys import sys
try: try:
import msvcrt import msvcrt
msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY) msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
except ImportError: except ImportError:
pass pass
exitcode = 0 parser = argparse.ArgumentParser()
out = sys.stdout parser.add_argument("name", help="the hook name, used for display")
out = getattr(out, 'buffer', out) 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"
)
args = parser.parse_args()
name = sys.argv[1] if args.out is None:
if len(sys.argv) > 2: out = sys.stdout
exitcode = int(sys.argv[2]) out = getattr(out, "buffer", out)
if len(sys.argv) > 3: else:
out = open(sys.argv[3], "ab") out = open(args.out, "ab")
# variables with empty values may not exist on all platforms, filter # variables with empty values may not exist on all platforms, filter
# them now for portability sake. # them now for portability sake.
env = [(k, v) for k, v in os.environ.items() env = [(k, v) for k, v in os.environ.items()
if k.startswith("HG_") and v] if k.startswith("HG_") and v]
env.sort() env.sort()
out.write(b"%s hook: " % name.encode('ascii')) out.write(b"%s hook: " % args.name.encode('ascii'))
if os.name == 'nt': if os.name == 'nt':
filter = lambda x: x.replace('\\', '/') filter = lambda x: x.replace('\\', '/')
else: else:
filter = lambda x: x filter = lambda x: x
vars = [b"%s=%s" % (k.encode('ascii'), filter(v).encode('ascii')) vars = [b"%s=%s" % (k.encode('ascii'), filter(v).encode('ascii'))
for k, v in env] for k, v in env]
out.write(b" ".join(vars)) out.write(b" ".join(vars))
out.write(b"\n") out.write(b"\n")
out.close() out.close()
sys.exit(exitcode) sys.exit(args.exitcode)