printenv.py
48 lines
| 1.2 KiB
| text/x-python
|
PythonLexer
/ tests / printenv.py
|
r4285 | # simple script to be used in hooks | ||
# copy it to the current directory when the test starts: | ||||
# | ||||
# cp "$TESTDIR"/printenv.py . | ||||
# | ||||
# put something like this in the repo .hg/hgrc: | ||||
# | ||||
# [hooks] | ||||
# changegroup = python ../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 os | ||||
import sys | ||||
|
r7080 | try: | ||
import msvcrt | ||||
msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY) | ||||
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) | ||||
|
r7186 | msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) | ||
|
r7080 | except ImportError: | ||
pass | ||||
|
r4285 | exitcode = 0 | ||
out = sys.stdout | ||||
name = sys.argv[1] | ||||
if len(sys.argv) > 2: | ||||
exitcode = int(sys.argv[2]) | ||||
if len(sys.argv) > 3: | ||||
out = open(sys.argv[3], "ab") | ||||
|
r4643 | # variables with empty values may not exist on all platforms, filter | ||
# them now for portability sake. | ||||
|
r4659 | env = [k for k, v in os.environ.iteritems() | ||
|
r4643 | if k.startswith("HG_") and v] | ||
|
r4285 | env.sort() | ||
out.write("%s hook: " % name) | ||||
for v in env: | ||||
|
r12642 | out.write("%s=%s " % (v, os.environ[v])) | ||
|
r4285 | out.write("\n") | ||
out.close() | ||||
sys.exit(exitcode) | ||||