extern/httpie-cli
1
0
Fork 1
mirror of https://github.com/httpie/cli.git synced 2024-11-25 17:23:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Show redirects WIP

Browse Source
This commit is contained in:
Jakub Roztocil 2016-02-29 12:56:40 +08:00
parent dc1371d4d6
commit c6d4f6cdf6
2 changed files with 55 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
httpie/core.py
View File

@ -99,7 +99,6 @@ def main(args=sys.argv[1:], env=Environment(), error=None):
return exit_status
download = None
try:
args = parser.parse_args(args=args, env=env)
@ -112,60 +111,64 @@ def main(args=sys.argv[1:], env=Environment(), error=None):
)
download.pre_request(args.headers)
response = get_response(args, config_dir=env.config.directory)
last_response = get_response(args, config_dir=env.config.directory)
if args.check_status or download:
redirect_chain = last_response.history + [last_response]
for response in redirect_chain:
exit_status = get_exit_status(
http_status=response.status_code,
follow=args.follow
)
if exit_status != ExitStatus.OK:
break
if not env.stdout_isatty and exit_status != ExitStatus.OK:
error('HTTP %s %s',
response.raw.status,
response.raw.reason,
level='warning')
if args.check_status or download:
write_kwargs = {
'stream': build_output_stream(
args, env, response.request, response),
# This will in fact be `stderr` with `--download`
'outfile': env.stdout,
'flush': env.stdout_isatty or args.stream
}
try:
if env.is_windows and is_py3 and 'colors' in args.prettify:
write_with_colors_win_py3(**write_kwargs)
else:
write(**write_kwargs)
if download and exit_status == ExitStatus.OK:
# Response body download.
download_stream, download_to = download.start(response)
write(
stream=download_stream,
outfile=download_to,
flush=False,
exit_status = get_exit_status(
http_status=response.status_code,
follow=args.follow
)
download.finish()
if download.interrupted:
exit_status = ExitStatus.ERROR
error('Incomplete download: size=%d; downloaded=%d' % (
download.status.total_size,
download.status.downloaded
))
except IOError as e:
if not traceback and e.errno == errno.EPIPE:
# Ignore broken pipes unless --traceback.
env.stderr.write('\n')
else:
raise
if not env.stdout_isatty and exit_status != ExitStatus.OK:
error('HTTP %s %s',
response.raw.status,
response.raw.reason,
level='warning')
write_kwargs = {
'stream': build_output_stream(args, env,
response.request,
response),
# This will in fact be `stderr` with `--download`
'outfile': env.stdout,
'flush': env.stdout_isatty or args.stream
}
try:
if env.is_windows and is_py3 and 'colors' in args.prettify:
write_with_colors_win_py3(**write_kwargs)
else:
write(**write_kwargs)
except IOError as e:
if not traceback and e.errno == errno.EPIPE:
# Ignore broken pipes unless --traceback.
env.stderr.write('\n')
else:
raise
if download and exit_status == ExitStatus.OK:
# Last response body download.
download_stream, download_to = download.start(last_response)
write(
stream=download_stream,
outfile=download_to,
flush=False,
)
download.finish()
if download.interrupted:
exit_status = ExitStatus.ERROR
error('Incomplete download: size=%d; downloaded=%d' % (
download.status.total_size,
download.status.downloaded
))
except KeyboardInterrupt:
if traceback:
raise

6
httpie/sessions.py
View File

@ -22,7 +22,7 @@ SESSION_IGNORED_HEADER_PREFIXES = ['Content-', 'If-']
def get_response(requests_session, session_name,
config_dir, args, read_only=False):
"""Like `client.get_response`, but applies permanent
"""Like `client.get_responses`, but applies permanent
aspects of the session to the request.
"""
@ -30,8 +30,8 @@ def get_response(requests_session, session_name,
if os.path.sep in session_name:
path = os.path.expanduser(session_name)
else:
hostname = (args.headers.get('Host', None)
or urlsplit(args.url).netloc.split('@')[-1])
hostname = (args.headers.get('Host', None) or
urlsplit(args.url).netloc.split('@')[-1])
if not hostname:
# HACK/FIXME: httpie-unixsocket's URLs have no hostname.
hostname = 'localhost'