mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 17:58:27 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
95 lines
2.4 KiB
Python
Executable File
95 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""This does HTTP GET requests given a host:port and path and returns
|
|
a subset of the headers plus the body of the result."""
|
|
|
|
from __future__ import absolute_import, print_function
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
from edenscm.mercurial import util
|
|
|
|
|
|
httplib = util.httplib
|
|
|
|
try:
|
|
import msvcrt
|
|
|
|
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
|
|
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
|
|
except ImportError:
|
|
pass
|
|
|
|
twice = False
|
|
if "--twice" in sys.argv:
|
|
sys.argv.remove("--twice")
|
|
twice = True
|
|
headeronly = False
|
|
if "--headeronly" in sys.argv:
|
|
sys.argv.remove("--headeronly")
|
|
headeronly = True
|
|
formatjson = False
|
|
if "--json" in sys.argv:
|
|
sys.argv.remove("--json")
|
|
formatjson = True
|
|
|
|
hgproto = None
|
|
if "--hgproto" in sys.argv:
|
|
idx = sys.argv.index("--hgproto")
|
|
hgproto = sys.argv[idx + 1]
|
|
sys.argv.pop(idx)
|
|
sys.argv.pop(idx)
|
|
|
|
tag = None
|
|
|
|
|
|
def request(host, path, show):
|
|
assert not path.startswith("/"), path
|
|
global tag
|
|
headers = {}
|
|
if tag:
|
|
headers["If-None-Match"] = tag
|
|
if hgproto:
|
|
headers["X-HgProto-1"] = hgproto
|
|
|
|
conn = httplib.HTTPConnection(host)
|
|
conn.request("GET", "/" + path, None, headers)
|
|
response = conn.getresponse()
|
|
print(response.status, response.reason)
|
|
if show[:1] == ["-"]:
|
|
show = sorted(h for h, v in response.getheaders() if h.lower() not in show)
|
|
for h in [h.lower() for h in show]:
|
|
if response.getheader(h, None) is not None:
|
|
print("%s: %s" % (h, response.getheader(h)))
|
|
if not headeronly:
|
|
print()
|
|
data = response.read()
|
|
|
|
# Pretty print JSON. This also has the beneficial side-effect
|
|
# of verifying emitted JSON is well-formed.
|
|
if formatjson:
|
|
# json.dumps() will print trailing newlines. Eliminate them
|
|
# to make tests easier to write.
|
|
data = json.loads(data)
|
|
lines = json.dumps(data, sort_keys=True, indent=2).splitlines()
|
|
for line in lines:
|
|
print(line.rstrip())
|
|
else:
|
|
sys.stdout.write(data)
|
|
|
|
if twice and response.getheader("ETag", None):
|
|
tag = response.getheader("ETag")
|
|
|
|
return response.status
|
|
|
|
|
|
status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
|
|
if twice:
|
|
status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
|
|
|
|
if 200 <= status <= 305:
|
|
sys.exit(0)
|
|
sys.exit(1)
|