|
|
#!/usr/bin/env python
|
|
|
#
|
|
|
# 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.
|
|
|
#
|
|
|
from __future__ import absolute_import
|
|
|
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"
|
|
|
)
|
|
|
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]
|
|
|
out.write(b" ".join(vars))
|
|
|
out.write(b"\n")
|
|
|
out.close()
|
|
|
|
|
|
sys.exit(args.exitcode)
|
|
|
|
