mirror of
https://github.com/httpie/cli.git
synced 2025-08-12 13:47:51 +02:00
Compare commits
92 Commits
Author | SHA1 | Date | |
---|---|---|---|
71e7061014 | |||
bc756cb6a2 | |||
63ed4d32a7 | |||
d1b91bfa9c | |||
dac79a8efc | |||
1fc8396c4b | |||
6c3b983c18 | |||
cfa7199f0b | |||
5a1177d57e | |||
c63a92f9b7 | |||
d17e02792b | |||
fc4f70a900 | |||
1681a4ddd0 | |||
289e9b844e | |||
72cf7c2cb7 | |||
4d84d77851 | |||
1b98505537 | |||
d32acfe2fa | |||
e8d79c4d8c | |||
38206e9e92 | |||
55d5e78324 | |||
341272db1e | |||
464b7a36da | |||
9d043eb745 | |||
40bd8f65af | |||
347653b369 | |||
ebfce6fb93 | |||
674acfe2c2 | |||
7ccdece39f | |||
e53dcba03e | |||
486657afa3 | |||
599bc0519f | |||
21613faa5a | |||
36bc64e02f | |||
6e5c696ac9 | |||
9b2a293e6e | |||
b0dd463687 | |||
bffaee13ff | |||
30afcea72d | |||
631c54b711 | |||
99f82bbd32 | |||
6f64b437b7 | |||
7774eac3df | |||
8e6c765be2 | |||
f0c42cd089 | |||
5c6cea79a1 | |||
2bed81059a | |||
be0b2f21d2 | |||
d97a610f7c | |||
5cc5b13555 | |||
3043f24733 | |||
093dab5896 | |||
5f42a21cfb | |||
4c45f0d91f | |||
d7ec7b2217 | |||
7817dfbbcc | |||
238b2e0441 | |||
a93d57b58b | |||
79c412064a | |||
0ae9d7af58 | |||
80e317fe24 | |||
1481749c22 | |||
d84d94dd55 | |||
1913b0d438 | |||
fe16f425a9 | |||
7ff71a7f10 | |||
4a37d10245 | |||
e5edb66ae8 | |||
1766dd8291 | |||
675a8b17ad | |||
69e26b8bc8 | |||
291f520e0c | |||
9ec328ff6f | |||
f2d59ba6bd | |||
53caf6ae72 | |||
8175366f27 | |||
8190a7c0c6 | |||
4a615e762f | |||
7426b4b493 | |||
2cdcadd9d5 | |||
18510a9396 | |||
acf5f063c7 | |||
2cf379df78 | |||
dd100c2cc4 | |||
444a9fa929 | |||
4a24cd25b9 | |||
1c5fb89001 | |||
466e1dbedf | |||
d87b2aa0e5 | |||
5d969852c7 | |||
bbc702fa11 | |||
e25d64a610 |
3
.gitignore
vendored
3
.gitignore
vendored
@ -6,4 +6,5 @@ build
|
|||||||
README.html
|
README.html
|
||||||
.coverage
|
.coverage
|
||||||
htmlcov
|
htmlcov
|
||||||
|
.idea
|
||||||
|
.DS_Store
|
||||||
|
@ -3,7 +3,7 @@ python:
|
|||||||
- 2.6
|
- 2.6
|
||||||
- 2.7
|
- 2.7
|
||||||
- pypy
|
- pypy
|
||||||
- 3.2
|
- 3.3
|
||||||
script: python setup.py test
|
script: python setup.py test
|
||||||
install:
|
install:
|
||||||
- pip install . --use-mirrors
|
- pip install . --use-mirrors
|
||||||
|
@ -8,6 +8,7 @@ HTTPie authors
|
|||||||
Patches and ideas
|
Patches and ideas
|
||||||
-----------------
|
-----------------
|
||||||
|
|
||||||
|
* `Cláudia T. Delgado <https://github.com/claudiatd>`_ (logo)
|
||||||
* `Hank Gay <https://github.com/gthank>`_
|
* `Hank Gay <https://github.com/gthank>`_
|
||||||
* `Jake Basile <https://github.com/jakebasile>`_
|
* `Jake Basile <https://github.com/jakebasile>`_
|
||||||
* `Vladimir Berkutov <https://github.com/dair-targ>`_
|
* `Vladimir Berkutov <https://github.com/dair-targ>`_
|
||||||
|
274
README.rst
274
README.rst
@ -1,12 +1,11 @@
|
|||||||
***********************
|
****************************************
|
||||||
HTTPie: cURL for Humans
|
HTTPie: a CLI, cURL-like tool for humans
|
||||||
***********************
|
****************************************
|
||||||
|
|
||||||
v0.2.8-alpha (`stable version`_)
|
|
||||||
|
|
||||||
HTTPie is a **command line HTTP client** whose goal is to make CLI interaction
|
HTTPie is a **command line HTTP client**. Its goal is to make CLI interaction
|
||||||
with web services as **human-friendly** as possible. It provides a
|
with web services as **human-friendly** as possible. It provides a
|
||||||
simple ``http`` command that allows for sending arbitrary HTTP requests with a
|
simple ``http`` command that allows for sending arbitrary HTTP requests using a
|
||||||
simple and natural syntax, and displays colorized responses. HTTPie can be used
|
simple and natural syntax, and displays colorized responses. HTTPie can be used
|
||||||
for **testing, debugging**, and generally **interacting** with HTTP servers.
|
for **testing, debugging**, and generally **interacting** with HTTP servers.
|
||||||
|
|
||||||
@ -15,10 +14,18 @@ for **testing, debugging**, and generally **interacting** with HTTP servers.
|
|||||||
:alt: HTTPie compared to cURL
|
:alt: HTTPie compared to cURL
|
||||||
:width: 835
|
:width: 835
|
||||||
:height: 835
|
:height: 835
|
||||||
|
:align: center
|
||||||
|
|
||||||
|
|
||||||
|
------
|
||||||
|
|
||||||
|
|
||||||
|
.. image:: https://raw.github.com/claudiatd/httpie-artwork/master/images/httpie_logo_simple.png
|
||||||
|
:alt: HTTPie logo
|
||||||
|
:align: center
|
||||||
|
|
||||||
HTTPie is written in Python, and under the hood it uses the excellent
|
HTTPie is written in Python, and under the hood it uses the excellent
|
||||||
`Requests`_ for HTTP and `Pygments`_ for colorizing.
|
`Requests`_ and `Pygments`_ libraries.
|
||||||
|
|
||||||
|
|
||||||
**Table of Contents**
|
**Table of Contents**
|
||||||
@ -30,7 +37,6 @@ HTTPie is written in Python, and under the hood it uses the excellent
|
|||||||
:backlinks: none
|
:backlinks: none
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
=============
|
=============
|
||||||
Main Features
|
Main Features
|
||||||
=============
|
=============
|
||||||
@ -42,6 +48,8 @@ Main Features
|
|||||||
* HTTPS, proxies, and authentication
|
* HTTPS, proxies, and authentication
|
||||||
* Arbitrary request data
|
* Arbitrary request data
|
||||||
* Custom headers
|
* Custom headers
|
||||||
|
* Persistent sessions
|
||||||
|
* Wget-like downloads
|
||||||
* Python 2.6, 2.7 and 3.x support
|
* Python 2.6, 2.7 and 3.x support
|
||||||
* Linux, Mac OS X and Windows support
|
* Linux, Mac OS X and Windows support
|
||||||
* Documentation
|
* Documentation
|
||||||
@ -58,9 +66,11 @@ or ``easy_install``:
|
|||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ pip install -U httpie
|
$ pip install --upgrade httpie
|
||||||
|
|
||||||
|
|
||||||
|
Alternatively:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ easy_install httpie
|
$ easy_install httpie
|
||||||
@ -76,12 +86,12 @@ Or, you can install the **development version** directly from GitHub:
|
|||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ pip install -U https://github.com/jkbr/httpie/tarball/master
|
$ pip install --upgrade https://github.com/jkbr/httpie/tarball/master
|
||||||
|
|
||||||
|
|
||||||
There are also packages available for `Ubuntu`_, `Debian`_, and possibly other
|
There are also packages available for `Ubuntu`_, `Debian`_, and possibly other
|
||||||
Linux distributions as well. However, they may be a significant delay between
|
Linux distributions as well. However, there may be a significant delay between
|
||||||
releases and package updates.
|
official HTTPie releases and package updates.
|
||||||
|
|
||||||
|
|
||||||
=====
|
=====
|
||||||
@ -126,14 +136,16 @@ Submitting `forms`_:
|
|||||||
$ http -f POST example.org hello=World
|
$ http -f POST example.org hello=World
|
||||||
|
|
||||||
|
|
||||||
See the request that is being sent using on of the `output options`_:
|
See the request that is being sent using one of the `output options`_:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ http -v example.org
|
$ http -v example.org
|
||||||
|
|
||||||
|
|
||||||
Use `Github API`_ to post a comment on an issue with `authentication`_:
|
Use `Github API`_ to post a comment on an
|
||||||
|
`issue <https://github.com/jkbr/httpie/issues/83>`_
|
||||||
|
with `authentication`_:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
@ -154,10 +166,27 @@ Download a file and save it via `redirected output`_:
|
|||||||
$ http example.org/file > file
|
$ http example.org/file > file
|
||||||
|
|
||||||
|
|
||||||
|
Download a file ``wget`` style:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http --download example.org/file
|
||||||
|
|
||||||
|
Use named `sessions`_ to make certain aspects or the communication persistent
|
||||||
|
between requests to the same host:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http --session=logged-in -a username:password httpbin.org/get API-Key:123
|
||||||
|
|
||||||
|
$ http --session=logged-in httpbin.org/headers
|
||||||
|
|
||||||
|
..
|
||||||
|
|
||||||
--------
|
--------
|
||||||
|
|
||||||
*What follows is a detailed documentation. It covers the command syntax,
|
*What follows is a detailed documentation. It covers the command syntax,
|
||||||
advances usage, and also features additional examples.*
|
advanced usage, and also features additional examples.*
|
||||||
|
|
||||||
|
|
||||||
============
|
============
|
||||||
@ -230,8 +259,8 @@ their type is distinguished only by the separator used:
|
|||||||
| | The ``==`` separator is used |
|
| | The ``==`` separator is used |
|
||||||
+-----------------------+-----------------------------------------------------+
|
+-----------------------+-----------------------------------------------------+
|
||||||
| Data Fields | Request data fields to be serialized as a JSON |
|
| Data Fields | Request data fields to be serialized as a JSON |
|
||||||
| ``field=value`` | object (default), or to be form encoded (``--form`` |
|
| ``field=value`` | object (default), or to be form encoded |
|
||||||
| | / ``-f``). |
|
| | (``--form, -f``). |
|
||||||
+-----------------------+-----------------------------------------------------+
|
+-----------------------+-----------------------------------------------------+
|
||||||
| Raw JSON fields | Useful when sending JSON and one or |
|
| Raw JSON fields | Useful when sending JSON and one or |
|
||||||
| ``field:=json`` | more fields need to be a ``Boolean``, ``Number``, |
|
| ``field:=json`` | more fields need to be a ``Boolean``, ``Number``, |
|
||||||
@ -239,18 +268,19 @@ their type is distinguished only by the separator used:
|
|||||||
| | ``meals:='["ham","spam"]'`` or ``pies:=[1,2,3]`` |
|
| | ``meals:='["ham","spam"]'`` or ``pies:=[1,2,3]`` |
|
||||||
| | (note the quotes). |
|
| | (note the quotes). |
|
||||||
+-----------------------+-----------------------------------------------------+
|
+-----------------------+-----------------------------------------------------+
|
||||||
| Files | Only available with ``-f`` / ``--form``. |
|
| Files | Only available with ``--form, -f``. |
|
||||||
| ``field@/dir/file`` | For example ``screenshot@~/Pictures/img.png``. |
|
| ``field@/dir/file`` | For example ``screenshot@~/Pictures/img.png``. |
|
||||||
| | The presence of a file field results |
|
| | The presence of a file field results |
|
||||||
| | in a ``multipart/form-data`` request. |
|
| | in a ``multipart/form-data`` request. |
|
||||||
+-----------------------+-----------------------------------------------------+
|
+-----------------------+-----------------------------------------------------+
|
||||||
|
|
||||||
You can use ``\`` to escape characters that shouldn't be used as separators
|
You can use ``\`` to escape characters that shouldn't be used as separators
|
||||||
(or parts thereof). e.g., ``foo\==bar`` will become a data key/value
|
(or parts thereof). For instance, ``foo\==bar`` will become a data key/value
|
||||||
pair (``foo=`` and ``bar``) instead of a URL parameter.
|
pair (``foo=`` and ``bar``) instead of a URL parameter.
|
||||||
|
|
||||||
Note that data fields aren't the only way to specify request data,
|
Note that data fields aren't the only way to specify request data:
|
||||||
`redirected input`_ allows passing arbitrary data to be sent with the request.
|
`Redirected input`_ allows for passing arbitrary data to be sent with the
|
||||||
|
request.
|
||||||
|
|
||||||
|
|
||||||
====
|
====
|
||||||
@ -269,7 +299,7 @@ both of which can be overwritten:
|
|||||||
``Accept`` ``application/json``
|
``Accept`` ``application/json``
|
||||||
================ =======================================
|
================ =======================================
|
||||||
|
|
||||||
You can use ``--json`` / ``-j`` to explicitly set ``Accept``
|
You can use ``--json, -j`` to explicitly set ``Accept``
|
||||||
to ``application/json`` regardless of whether you are sending data
|
to ``application/json`` regardless of whether you are sending data
|
||||||
(it's a shortcut for setting the header via the usual header notation –
|
(it's a shortcut for setting the header via the usual header notation –
|
||||||
``http url Accept:application/json``).
|
``http url Accept:application/json``).
|
||||||
@ -287,7 +317,6 @@ Simple example:
|
|||||||
Accept-Encoding: identity, deflate, compress, gzip
|
Accept-Encoding: identity, deflate, compress, gzip
|
||||||
Content-Type: application/json; charset=utf-8
|
Content-Type: application/json; charset=utf-8
|
||||||
Host: example.org
|
Host: example.org
|
||||||
User-Agent: HTTPie/0.2.7dev
|
|
||||||
|
|
||||||
{
|
{
|
||||||
"name": "John",
|
"name": "John",
|
||||||
@ -309,7 +338,6 @@ into the resulting object:
|
|||||||
Accept: application/json
|
Accept: application/json
|
||||||
Content-Type: application/json; charset=utf-8
|
Content-Type: application/json; charset=utf-8
|
||||||
Host: api.example.com
|
Host: api.example.com
|
||||||
User-Agent: HTTPie/0.2.7dev
|
|
||||||
|
|
||||||
{
|
{
|
||||||
"age": 29,
|
"age": 29,
|
||||||
@ -334,12 +362,12 @@ Forms
|
|||||||
=====
|
=====
|
||||||
|
|
||||||
Submitting forms is very similar to sending `JSON`_ requests. Often the only
|
Submitting forms is very similar to sending `JSON`_ requests. Often the only
|
||||||
difference is in adding the ``--form`` / ``-f`` option, which ensures that
|
difference is in adding the ``--form, -f`` option, which ensures that
|
||||||
data fields are serialized as, and ``Content-Type`` is set to,
|
data fields are serialized as, and ``Content-Type`` is set to,
|
||||||
``application/x-www-form-urlencoded; charset=utf-8``.
|
``application/x-www-form-urlencoded; charset=utf-8``.
|
||||||
|
|
||||||
It is possible to make form data the implicit content type via the `config`_
|
It is possible to make form data the implicit content type instead of JSON
|
||||||
file.
|
via the `config`_ file.
|
||||||
|
|
||||||
|
|
||||||
-------------
|
-------------
|
||||||
@ -354,7 +382,6 @@ Regular Forms
|
|||||||
.. code-block:: http
|
.. code-block:: http
|
||||||
|
|
||||||
POST /person/1 HTTP/1.1
|
POST /person/1 HTTP/1.1
|
||||||
User-Agent: HTTPie/0.2.7dev
|
|
||||||
Content-Type: application/x-www-form-urlencoded; charset=utf-8
|
Content-Type: application/x-www-form-urlencoded; charset=utf-8
|
||||||
|
|
||||||
name=John+Smith&email=john%40example.org
|
name=John+Smith&email=john%40example.org
|
||||||
@ -433,13 +460,16 @@ come). There are two flags that control authentication:
|
|||||||
(``-a username``), you'll be prompted for
|
(``-a username``), you'll be prompted for
|
||||||
the password before the request is sent.
|
the password before the request is sent.
|
||||||
To send a an empty password, pass ``username:``.
|
To send a an empty password, pass ``username:``.
|
||||||
|
The ``username:password@hostname`` URL syntax is
|
||||||
|
supported as well (but credentials passed via ``-a``
|
||||||
|
have higher priority).
|
||||||
|
|
||||||
``--auth-type`` Specify the auth mechanism. Possible values are
|
``--auth-type`` Specify the auth mechanism. Possible values are
|
||||||
``basic`` and ``digest``. The default value is
|
``basic`` and ``digest``. The default value is
|
||||||
``basic`` so it can often be omitted.
|
``basic`` so it can often be omitted.
|
||||||
=================== ======================================================
|
=================== ======================================================
|
||||||
|
|
||||||
Authorization information from ``.netrc`` is respected as well.
|
|
||||||
|
|
||||||
Basic auth:
|
Basic auth:
|
||||||
|
|
||||||
@ -464,22 +494,36 @@ With password prompt:
|
|||||||
$ http -a username example.org
|
$ http -a username example.org
|
||||||
|
|
||||||
|
|
||||||
|
Authorization information from your ``.netrc`` file is honored as well:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ cat .netrc
|
||||||
|
machine httpbin.org
|
||||||
|
login httpie
|
||||||
|
password test
|
||||||
|
$ http httpbin.org/basic-auth/httpie/test
|
||||||
|
HTTP/1.1 200 OK
|
||||||
|
[...]
|
||||||
|
|
||||||
|
|
||||||
=======
|
=======
|
||||||
Proxies
|
Proxies
|
||||||
=======
|
=======
|
||||||
|
|
||||||
You can specify proxies to be used through the ``--proxy`` argument:
|
You can specify proxies to be used through the ``--proxy`` argument for each
|
||||||
|
protocol (which is included in the value in case of redirects across protocols):
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
http --proxy=http:10.10.1.10:3128 --https:10.10.1.10:1080 example.org
|
$ http --proxy=http:10.10.1.10:3128 --proxy=https:10.10.1.10:1080 example.org
|
||||||
|
|
||||||
|
|
||||||
With Basic authentication:
|
With Basic authentication:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
http --proxy=http:http://user:pass@10.10.1.10:3128 example.org
|
$ http --proxy=http:http://user:pass@10.10.1.10:3128 example.org
|
||||||
|
|
||||||
You can also configure proxies by environment variables ``HTTP_PROXY`` and
|
You can also configure proxies by environment variables ``HTTP_PROXY`` and
|
||||||
``HTTPS_PROXY``, and the underlying Requests library will pick them up as well.
|
``HTTPS_PROXY``, and the underlying Requests library will pick them up as well.
|
||||||
@ -551,7 +595,7 @@ documentation examples:
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
All the other options are just a shortcut for ``--print`` / ``-p``.
|
All the other options are just a shortcut for ``--print, -p``.
|
||||||
It accepts a string of characters each of which represents a specific part of
|
It accepts a string of characters each of which represents a specific part of
|
||||||
the HTTP exchange:
|
the HTTP exchange:
|
||||||
|
|
||||||
@ -723,7 +767,7 @@ that the response body is binary,
|
|||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
http example.org/Movie.mov
|
$ http example.org/Movie.mov
|
||||||
|
|
||||||
|
|
||||||
You will nearly instantly see something like this:
|
You will nearly instantly see something like this:
|
||||||
@ -781,6 +825,76 @@ Force colorizing and formatting, and show both the request and the response in
|
|||||||
The ``-R`` flag tells ``less`` to interpret color escape sequences included
|
The ``-R`` flag tells ``less`` to interpret color escape sequences included
|
||||||
HTTPie`s output.
|
HTTPie`s output.
|
||||||
|
|
||||||
|
You can create a shortcut for invoking HTTPie with colorized and paged output
|
||||||
|
by adding the following to your ``~/.bash_profile``:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
function httpless {
|
||||||
|
# `httpless example.org'
|
||||||
|
http --pretty=all "$@" | less -R;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
=============
|
||||||
|
Download Mode
|
||||||
|
=============
|
||||||
|
|
||||||
|
HTTPie features a download mode in which it acts similarly to ``wget``.
|
||||||
|
|
||||||
|
When enabled using the ``--download, -d`` flag, response headers are printed to
|
||||||
|
the terminal (``stderr``), and a progress bar is shown while the response body
|
||||||
|
is being saved to a file.
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http --download https://github.com/jkbr/httpie/tarball/master
|
||||||
|
|
||||||
|
.. code-block:: http
|
||||||
|
|
||||||
|
HTTP/1.1 200 OK
|
||||||
|
Connection: keep-alive
|
||||||
|
Content-Disposition: attachment; filename=jkbr-httpie-0.4.1-33-gfc4f70a.tar.gz
|
||||||
|
Content-Length: 505530
|
||||||
|
Content-Type: application/x-gzip
|
||||||
|
Server: GitHub.com
|
||||||
|
Vary: Accept-Encoding
|
||||||
|
|
||||||
|
Downloading 494.89 kB to "jkbr-httpie-0.4.1-33-gfc4f70a.tar.gz"
|
||||||
|
/ 21.01% 104.00 kB 47.55 kB/s 0:00:08 ETA
|
||||||
|
|
||||||
|
|
||||||
|
If not provided via ``--output, -o``, the output filename will be determined
|
||||||
|
from ``Content-Disposition`` (if available), or from the URL and
|
||||||
|
``Content-Type``. If the guessed filename already exists, HTTPie adds a unique
|
||||||
|
suffix to it.
|
||||||
|
|
||||||
|
You can also redirect the response body to another program while the response
|
||||||
|
headers and progress are still shown in the terminal:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http -d https://github.com/jkbr/httpie/tarball/master | tar zxf -
|
||||||
|
|
||||||
|
|
||||||
|
If ``--output, -o`` is specified, you can resume a partial download using the
|
||||||
|
``--continue, -c`` option. This only works with servers that support
|
||||||
|
``Range`` requests and ``206 Partial Content`` responses. If the server doesn't
|
||||||
|
support that, the whole file will simply be downloaded:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http -dco file.zip example.org/file
|
||||||
|
|
||||||
|
Other notes:
|
||||||
|
|
||||||
|
* The ``--download`` option only changes how the response body is treated.
|
||||||
|
* You can still set custom headers, use sessions, ``--verbose, -v``, etc.
|
||||||
|
* ``--download`` always implies ``--follow`` (redirects are followed).
|
||||||
|
* HTTPie exits with status code ``1`` (error) if the body hasn't been fully
|
||||||
|
downloaded.
|
||||||
|
* ``Accept-Encoding`` cannot be set with ``--download``.
|
||||||
|
|
||||||
|
|
||||||
==================
|
==================
|
||||||
Streamed Responses
|
Streamed Responses
|
||||||
@ -819,54 +933,59 @@ Streamed output by small chunks alá ``tail -f``:
|
|||||||
$ http --stream -f -a YOUR-TWITTER-NAME https://stream.twitter.com/1/statuses/filter.json track=Apple \
|
$ http --stream -f -a YOUR-TWITTER-NAME https://stream.twitter.com/1/statuses/filter.json track=Apple \
|
||||||
| while read tweet; do echo "$tweet" | http POST example.org/tweets ; done
|
| while read tweet; do echo "$tweet" | http POST example.org/tweets ; done
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
========
|
========
|
||||||
Sessions
|
Sessions
|
||||||
========
|
========
|
||||||
|
|
||||||
HTTPie supports named, per-host sessions, where custom headers, authorization,
|
By default, every request is completely independent of the previous ones.
|
||||||
and cookies (manually specified or sent by the server) persist between requests:
|
HTTPie also supports persistent sessions, where custom headers, authorization,
|
||||||
|
and cookies (manually specified or sent by the server) persist between
|
||||||
|
requests to the same host.
|
||||||
|
|
||||||
|
Create a new session named ``user1``:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ http --session user1 -a user1:password example.org X-Foo:Bar
|
$ http --session=user1 -a user1:password example.org X-Foo:Bar
|
||||||
|
|
||||||
Now you can refer to the session by its name:
|
Now you can refer to the session by its name, and the previously used
|
||||||
|
authorization and HTTP headers will automatically be set:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ http --session user1 example.org
|
$ http --session=user1 example.org
|
||||||
|
|
||||||
|
To create or reuse a different session, simple specify a different name:
|
||||||
To switch to another session simple pass a different name:
|
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ http --session user2 -a user2:password example.org X-Bar:Foo
|
$ http --session=user2 -a user2:password example.org X-Bar:Foo
|
||||||
|
|
||||||
To use a session without updating it from the request/response exchange
|
To use a session without updating it from the request/response exchange
|
||||||
once it is created, specify the session name via
|
once it is created, specify the session name via
|
||||||
``--session-read-only=SESSION_NAME`` instead.
|
``--session-read-only=SESSION_NAME`` instead.
|
||||||
|
|
||||||
You can view and manipulate existing sessions via the ``httpie`` management
|
Session data are stored in JSON files in the directory
|
||||||
command, see ``httpie --help``.
|
``~/.httpie/sessions/<host>/<name>.json``
|
||||||
|
|
||||||
Sessions are stored as JSON in ``~/.httpie/sessions/<host>/<name>.json``
|
|
||||||
(``%APPDATA%\httpie\sessions\<host>\<name>.json`` on Windows).
|
(``%APPDATA%\httpie\sessions\<host>\<name>.json`` on Windows).
|
||||||
|
**Warning:** All session data, including credentials, cookie data,
|
||||||
|
and custom headers are stored in plain text.
|
||||||
|
|
||||||
See also `config`_.
|
Session files can also be created and edited manually in a text editor.
|
||||||
|
|
||||||
|
|
||||||
|
See also `Config`_.
|
||||||
|
|
||||||
|
|
||||||
======
|
======
|
||||||
Config
|
Config
|
||||||
======
|
======
|
||||||
|
|
||||||
HTTPie provides a simple configuration file containing a JSON
|
HTTPie uses a simple configuration file that contains a JSON object with the
|
||||||
object with the following keys:
|
following keys:
|
||||||
|
|
||||||
========================= =================================================
|
========================= =================================================
|
||||||
``__version__`` HTTPie automatically sets this to its version.
|
``__meta__`` HTTPie automatically stores some metadata here.
|
||||||
Do not change.
|
Do not change.
|
||||||
|
|
||||||
``implicit_content_type`` A ``String`` specifying the implicit content type
|
``implicit_content_type`` A ``String`` specifying the implicit content type
|
||||||
@ -876,10 +995,24 @@ object with the following keys:
|
|||||||
|
|
||||||
``default_options`` An ``Array`` (by default empty) of options
|
``default_options`` An ``Array`` (by default empty) of options
|
||||||
that should be applied to every request.
|
that should be applied to every request.
|
||||||
|
|
||||||
|
For instance, you can use this option to change
|
||||||
|
the default style and output options:
|
||||||
|
``"default_options": ["--style=fruity", "--body"]``
|
||||||
|
|
||||||
|
Another useful default option is
|
||||||
|
``"--session=default"`` to make HTTPie always
|
||||||
|
use `sessions`_.
|
||||||
|
|
||||||
|
Default options from config file can be unset
|
||||||
|
for a particular invocation via
|
||||||
|
``--no-OPTION`` arguments passed on the
|
||||||
|
command line (e.g., ``--no-style``
|
||||||
|
or ``--no-session``).
|
||||||
========================= =================================================
|
========================= =================================================
|
||||||
|
|
||||||
The default location is ``~/.httpie/config.json``
|
The default location of the configuration file is ``~/.httpie/config.json``
|
||||||
(``%APPDATA%\httpie\config.json`` on Windows).
|
(or ``%APPDATA%\httpie\config.json`` on Windows).
|
||||||
|
|
||||||
The config directory location can be changed by setting the
|
The config directory location can be changed by setting the
|
||||||
``HTTPIE_CONFIG_DIR`` environment variable.
|
``HTTPIE_CONFIG_DIR`` environment variable.
|
||||||
@ -1003,6 +1136,11 @@ Please run the existing suite of tests before a pull request is submitted:
|
|||||||
|
|
||||||
Don't forget to add yourself to `AUTHORS.rst`_.
|
Don't forget to add yourself to `AUTHORS.rst`_.
|
||||||
|
|
||||||
|
=======
|
||||||
|
Logo
|
||||||
|
=======
|
||||||
|
|
||||||
|
See `claudiatd/httpie-artwork`_
|
||||||
|
|
||||||
=======
|
=======
|
||||||
Authors
|
Authors
|
||||||
@ -1011,7 +1149,6 @@ Authors
|
|||||||
`Jakub Roztocil`_ (`@jakubroztocil`_) created HTTPie and `these fine people`_
|
`Jakub Roztocil`_ (`@jakubroztocil`_) created HTTPie and `these fine people`_
|
||||||
have contributed.
|
have contributed.
|
||||||
|
|
||||||
|
|
||||||
=======
|
=======
|
||||||
Licence
|
Licence
|
||||||
=======
|
=======
|
||||||
@ -1025,9 +1162,22 @@ Changelog
|
|||||||
|
|
||||||
*You can click a version name to see a diff with the previous one.*
|
*You can click a version name to see a diff with the previous one.*
|
||||||
|
|
||||||
* `0.2.8-alpha`_
|
* `0.6.0-dev`_
|
||||||
|
* `0.5.0`_ (2013-04-27)
|
||||||
|
* Added a `download mode`_ via ``--download``.
|
||||||
|
* Bugfixes.
|
||||||
|
* `0.4.1`_ (2013-02-26)
|
||||||
|
* Fixed ``setup.py``.
|
||||||
|
* `0.4.0`_ (2013-02-22)
|
||||||
|
* Python 3.3 compatibility.
|
||||||
|
* Requests >= v1.0.4 compatibility.
|
||||||
|
* Added support for credentials in URL.
|
||||||
|
* Added ``--no-option`` for every ``--option`` to be config-friendly.
|
||||||
|
* Mutually exclusive arguments can be specified multiple times. The
|
||||||
|
last value is used.
|
||||||
|
* `0.3.0`_ (2012-09-21)
|
||||||
* Allow output redirection on Windows.
|
* Allow output redirection on Windows.
|
||||||
* Added config file.
|
* Added configuration file.
|
||||||
* Added persistent session support.
|
* Added persistent session support.
|
||||||
* Renamed ``--allow-redirects`` to ``--follow``.
|
* Renamed ``--allow-redirects`` to ``--follow``.
|
||||||
* Improved the usability of ``http --help``.
|
* Improved the usability of ``http --help``.
|
||||||
@ -1040,7 +1190,7 @@ Changelog
|
|||||||
``--ugly`` has bee removed in favor of ``--pretty=none``.
|
``--ugly`` has bee removed in favor of ``--pretty=none``.
|
||||||
* `0.2.7`_ (2012-08-07)
|
* `0.2.7`_ (2012-08-07)
|
||||||
* Compatibility with Requests 0.13.6.
|
* Compatibility with Requests 0.13.6.
|
||||||
* Streamed terminal output. ``--stream`` / ``-S`` can be used to enable
|
* Streamed terminal output. ``--stream, -S`` can be used to enable
|
||||||
streaming also with ``--pretty`` and to ensure a more frequent output
|
streaming also with ``--pretty`` and to ensure a more frequent output
|
||||||
flushing.
|
flushing.
|
||||||
* Support for efficient large file downloads.
|
* Support for efficient large file downloads.
|
||||||
@ -1110,6 +1260,7 @@ Changelog
|
|||||||
.. _Jakub Roztocil: http://roztocil.name
|
.. _Jakub Roztocil: http://roztocil.name
|
||||||
.. _@jakubroztocil: https://twitter.com/jakubroztocil
|
.. _@jakubroztocil: https://twitter.com/jakubroztocil
|
||||||
.. _existing issues: https://github.com/jkbr/httpie/issues?state=open
|
.. _existing issues: https://github.com/jkbr/httpie/issues?state=open
|
||||||
|
.. _claudiatd/httpie-artwork: https://github.com/claudiatd/httpie-artwork
|
||||||
.. _0.1.6: https://github.com/jkbr/httpie/compare/0.1.4...0.1.6
|
.. _0.1.6: https://github.com/jkbr/httpie/compare/0.1.4...0.1.6
|
||||||
.. _0.2.0: https://github.com/jkbr/httpie/compare/0.1.6...0.2.0
|
.. _0.2.0: https://github.com/jkbr/httpie/compare/0.1.6...0.2.0
|
||||||
.. _0.2.1: https://github.com/jkbr/httpie/compare/0.2.0...0.2.1
|
.. _0.2.1: https://github.com/jkbr/httpie/compare/0.2.0...0.2.1
|
||||||
@ -1117,7 +1268,10 @@ Changelog
|
|||||||
.. _0.2.5: https://github.com/jkbr/httpie/compare/0.2.2...0.2.5
|
.. _0.2.5: https://github.com/jkbr/httpie/compare/0.2.2...0.2.5
|
||||||
.. _0.2.6: https://github.com/jkbr/httpie/compare/0.2.5...0.2.6
|
.. _0.2.6: https://github.com/jkbr/httpie/compare/0.2.5...0.2.6
|
||||||
.. _0.2.7: https://github.com/jkbr/httpie/compare/0.2.5...0.2.7
|
.. _0.2.7: https://github.com/jkbr/httpie/compare/0.2.5...0.2.7
|
||||||
.. _0.2.8-alpha: https://github.com/jkbr/httpie/compare/0.2.7...master
|
.. _0.3.0: https://github.com/jkbr/httpie/compare/0.2.7...0.3.0
|
||||||
.. _stable version: https://github.com/jkbr/httpie/tree/0.2.7#readme
|
.. _0.4.0: https://github.com/jkbr/httpie/compare/0.3.0...0.4.0
|
||||||
|
.. _0.4.1: https://github.com/jkbr/httpie/compare/0.4.0...0.4.1
|
||||||
|
.. _0.5.0: https://github.com/jkbr/httpie/compare/0.4.1...0.5.0
|
||||||
|
.. _0.6.0-dev: https://github.com/jkbr/httpie/compare/0.5.0...master
|
||||||
.. _AUTHORS.rst: https://github.com/jkbr/httpie/blob/master/AUTHORS.rst
|
.. _AUTHORS.rst: https://github.com/jkbr/httpie/blob/master/AUTHORS.rst
|
||||||
.. _LICENSE: https://github.com/jkbr/httpie/blob/master/LICENSE
|
.. _LICENSE: https://github.com/jkbr/httpie/blob/master/LICENSE
|
||||||
|
@ -1,13 +1,14 @@
|
|||||||
"""
|
"""
|
||||||
HTTPie - cURL for humans.
|
HTTPie - a CLI, cURL-like tool for humans.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
__author__ = 'Jakub Roztocil'
|
__author__ = 'Jakub Roztocil'
|
||||||
__version__ = '0.2.8-alpha'
|
__version__ = '0.5.0'
|
||||||
__licence__ = 'BSD'
|
__licence__ = 'BSD'
|
||||||
|
|
||||||
|
|
||||||
class EXIT:
|
class ExitStatus:
|
||||||
|
"""Exit status code constants."""
|
||||||
OK = 0
|
OK = 0
|
||||||
ERROR = 1
|
ERROR = 1
|
||||||
ERROR_TIMEOUT = 2
|
ERROR_TIMEOUT = 2
|
||||||
|
209
httpie/cli.py
209
httpie/cli.py
@ -5,17 +5,15 @@ NOTE: the CLI interface may change before reaching v1.0.
|
|||||||
"""
|
"""
|
||||||
from argparse import FileType, OPTIONAL, ZERO_OR_MORE, SUPPRESS
|
from argparse import FileType, OPTIONAL, ZERO_OR_MORE, SUPPRESS
|
||||||
|
|
||||||
from requests.compat import is_windows
|
|
||||||
|
|
||||||
from . import __doc__
|
from . import __doc__
|
||||||
from . import __version__
|
from . import __version__
|
||||||
from .config import DEFAULT_CONFIG_DIR
|
from .sessions import DEFAULT_SESSIONS_DIR, Session
|
||||||
from .output import AVAILABLE_STYLES, DEFAULT_STYLE
|
from .output import AVAILABLE_STYLES, DEFAULT_STYLE
|
||||||
from .input import (Parser, AuthCredentialsArgType, KeyValueArgType,
|
from .input import (Parser, AuthCredentialsArgType, KeyValueArgType,
|
||||||
SEP_PROXY, SEP_CREDENTIALS, SEP_GROUP_ITEMS,
|
SEP_PROXY, SEP_CREDENTIALS, SEP_GROUP_ITEMS,
|
||||||
OUT_REQ_HEAD, OUT_REQ_BODY, OUT_RESP_HEAD,
|
OUT_REQ_HEAD, OUT_REQ_BODY, OUT_RESP_HEAD,
|
||||||
OUT_RESP_BODY, OUTPUT_OPTIONS,
|
OUT_RESP_BODY, OUTPUT_OPTIONS,
|
||||||
PRETTY_MAP, PRETTY_STDOUT_TTY_ONLY)
|
PRETTY_MAP, PRETTY_STDOUT_TTY_ONLY, RegexValidator)
|
||||||
|
|
||||||
|
|
||||||
def _(text):
|
def _(text):
|
||||||
@ -25,14 +23,13 @@ def _(text):
|
|||||||
|
|
||||||
parser = Parser(
|
parser = Parser(
|
||||||
description='%s <http://httpie.org>' % __doc__.strip(),
|
description='%s <http://httpie.org>' % __doc__.strip(),
|
||||||
epilog=_('''
|
epilog='For every --option there is a --no-option'
|
||||||
Suggestions and bug reports are greatly appreciated:
|
' that reverts the option to its default value.\n\n'
|
||||||
https://github.com/jkbr/httpie/issues
|
'Suggestions and bug reports are greatly appreciated:\n'
|
||||||
''')
|
'https://github.com/jkbr/httpie/issues'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
# Positional arguments.
|
# Positional arguments.
|
||||||
###############################################################################
|
###############################################################################
|
||||||
@ -41,11 +38,12 @@ positional = parser.add_argument_group(
|
|||||||
title='Positional arguments',
|
title='Positional arguments',
|
||||||
description=_('''
|
description=_('''
|
||||||
These arguments come after any flags and in the
|
These arguments come after any flags and in the
|
||||||
order they are listed here. Only URL is required.'''
|
order they are listed here. Only URL is required.
|
||||||
)
|
''')
|
||||||
)
|
)
|
||||||
positional.add_argument(
|
positional.add_argument(
|
||||||
'method', metavar='METHOD',
|
'method',
|
||||||
|
metavar='METHOD',
|
||||||
nargs=OPTIONAL,
|
nargs=OPTIONAL,
|
||||||
default=None,
|
default=None,
|
||||||
help=_('''
|
help=_('''
|
||||||
@ -57,14 +55,16 @@ positional.add_argument(
|
|||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
positional.add_argument(
|
positional.add_argument(
|
||||||
'url', metavar='URL',
|
'url',
|
||||||
|
metavar='URL',
|
||||||
help=_('''
|
help=_('''
|
||||||
The protocol defaults to http:// if the
|
The protocol defaults to http:// if the
|
||||||
URL does not include one.
|
URL does not include one.
|
||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
positional.add_argument(
|
positional.add_argument(
|
||||||
'items', metavar='REQUEST ITEM',
|
'items',
|
||||||
|
metavar='REQUEST ITEM',
|
||||||
nargs=ZERO_OR_MORE,
|
nargs=ZERO_OR_MORE,
|
||||||
type=KeyValueArgType(*SEP_GROUP_ITEMS),
|
type=KeyValueArgType(*SEP_GROUP_ITEMS),
|
||||||
help=_('''
|
help=_('''
|
||||||
@ -87,10 +87,11 @@ positional.add_argument(
|
|||||||
content_type = parser.add_argument_group(
|
content_type = parser.add_argument_group(
|
||||||
title='Predefined content types',
|
title='Predefined content types',
|
||||||
description=None
|
description=None
|
||||||
).add_mutually_exclusive_group(required=False)
|
)
|
||||||
|
|
||||||
content_type.add_argument(
|
content_type.add_argument(
|
||||||
'--json', '-j', action='store_true',
|
'--json', '-j',
|
||||||
|
action='store_true',
|
||||||
help=_('''
|
help=_('''
|
||||||
(default) Data items from the command
|
(default) Data items from the command
|
||||||
line are serialized as a JSON object.
|
line are serialized as a JSON object.
|
||||||
@ -99,7 +100,8 @@ content_type.add_argument(
|
|||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
content_type.add_argument(
|
content_type.add_argument(
|
||||||
'--form', '-f', action='store_true',
|
'--form', '-f',
|
||||||
|
action='store_true',
|
||||||
help=_('''
|
help=_('''
|
||||||
Data items from the command line are serialized as form fields.
|
Data items from the command line are serialized as form fields.
|
||||||
The Content-Type is set to application/x-www-form-urlencoded
|
The Content-Type is set to application/x-www-form-urlencoded
|
||||||
@ -117,20 +119,9 @@ content_type.add_argument(
|
|||||||
output_processing = parser.add_argument_group(title='Output processing')
|
output_processing = parser.add_argument_group(title='Output processing')
|
||||||
|
|
||||||
output_processing.add_argument(
|
output_processing.add_argument(
|
||||||
'--output', '-o', type=FileType('w+b'),
|
'--pretty',
|
||||||
metavar='FILE',
|
dest='prettify',
|
||||||
help= SUPPRESS if not is_windows else _(
|
default=PRETTY_STDOUT_TTY_ONLY,
|
||||||
'''
|
|
||||||
Save output to FILE.
|
|
||||||
This option is a replacement for piping output to FILE,
|
|
||||||
which would on Windows result in corrupted data
|
|
||||||
being saved.
|
|
||||||
|
|
||||||
'''
|
|
||||||
)
|
|
||||||
)
|
|
||||||
output_processing.add_argument(
|
|
||||||
'--pretty', dest='prettify', default=PRETTY_STDOUT_TTY_ONLY,
|
|
||||||
choices=sorted(PRETTY_MAP.keys()),
|
choices=sorted(PRETTY_MAP.keys()),
|
||||||
help=_('''
|
help=_('''
|
||||||
Controls output processing. The value can be "none" to not prettify
|
Controls output processing. The value can be "none" to not prettify
|
||||||
@ -140,7 +131,10 @@ output_processing.add_argument(
|
|||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
output_processing.add_argument(
|
output_processing.add_argument(
|
||||||
'--style', '-s', dest='style', default=DEFAULT_STYLE, metavar='STYLE',
|
'--style', '-s',
|
||||||
|
dest='style',
|
||||||
|
metavar='STYLE',
|
||||||
|
default=DEFAULT_STYLE,
|
||||||
choices=AVAILABLE_STYLES,
|
choices=AVAILABLE_STYLES,
|
||||||
help=_('''
|
help=_('''
|
||||||
Output coloring style. One of %s. Defaults to "%s".
|
Output coloring style. One of %s. Defaults to "%s".
|
||||||
@ -151,14 +145,14 @@ output_processing.add_argument(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
# Output options
|
# Output options
|
||||||
###############################################################################
|
###############################################################################
|
||||||
output_options = parser.add_argument_group(title='Output options')
|
output_options = parser.add_argument_group(title='Output options')
|
||||||
|
|
||||||
output_print = output_options.add_mutually_exclusive_group(required=False)
|
output_options.add_argument(
|
||||||
output_print.add_argument('--print', '-p', dest='output_options',
|
'--print', '-p',
|
||||||
|
dest='output_options',
|
||||||
metavar='WHAT',
|
metavar='WHAT',
|
||||||
help=_('''
|
help=_('''
|
||||||
String specifying what the output should contain:
|
String specifying what the output should contain:
|
||||||
@ -170,39 +164,46 @@ output_print.add_argument('--print', '-p', dest='output_options',
|
|||||||
headers and body is printed), if standard output is not redirected.
|
headers and body is printed), if standard output is not redirected.
|
||||||
If the output is piped to another program or to a file,
|
If the output is piped to another program or to a file,
|
||||||
then only the body is printed by default.
|
then only the body is printed by default.
|
||||||
'''.format(
|
'''.format(request_headers=OUT_REQ_HEAD,
|
||||||
request_headers=OUT_REQ_HEAD,
|
request_body=OUT_REQ_BODY,
|
||||||
request_body=OUT_REQ_BODY,
|
response_headers=OUT_RESP_HEAD,
|
||||||
response_headers=OUT_RESP_HEAD,
|
response_body=OUT_RESP_BODY,))
|
||||||
response_body=OUT_RESP_BODY,
|
|
||||||
))
|
|
||||||
)
|
)
|
||||||
output_print.add_argument(
|
output_options.add_argument(
|
||||||
'--verbose', '-v', dest='output_options',
|
'--verbose', '-v',
|
||||||
action='store_const', const=''.join(OUTPUT_OPTIONS),
|
dest='output_options',
|
||||||
|
action='store_const',
|
||||||
|
const=''.join(OUTPUT_OPTIONS),
|
||||||
help=_('''
|
help=_('''
|
||||||
Print the whole request as well as the response.
|
Print the whole request as well as the response.
|
||||||
Shortcut for --print={0}.
|
Shortcut for --print={0}.
|
||||||
'''.format(''.join(OUTPUT_OPTIONS)))
|
'''.format(''.join(OUTPUT_OPTIONS)))
|
||||||
)
|
)
|
||||||
output_print.add_argument(
|
output_options.add_argument(
|
||||||
'--headers', '-h', dest='output_options',
|
'--headers', '-h',
|
||||||
action='store_const', const=OUT_RESP_HEAD,
|
dest='output_options',
|
||||||
|
action='store_const',
|
||||||
|
const=OUT_RESP_HEAD,
|
||||||
help=_('''
|
help=_('''
|
||||||
Print only the response headers.
|
Print only the response headers.
|
||||||
Shortcut for --print={0}.
|
Shortcut for --print={0}.
|
||||||
'''.format(OUT_RESP_HEAD))
|
'''.format(OUT_RESP_HEAD))
|
||||||
)
|
)
|
||||||
output_print.add_argument(
|
output_options.add_argument(
|
||||||
'--body', '-b', dest='output_options',
|
'--body', '-b',
|
||||||
action='store_const', const=OUT_RESP_BODY,
|
dest='output_options',
|
||||||
|
action='store_const',
|
||||||
|
const=OUT_RESP_BODY,
|
||||||
help=_('''
|
help=_('''
|
||||||
Print only the response body.
|
Print only the response body.
|
||||||
Shortcut for --print={0}.
|
Shortcut for --print={0}.
|
||||||
'''.format(OUT_RESP_BODY))
|
'''.format(OUT_RESP_BODY))
|
||||||
)
|
)
|
||||||
|
|
||||||
output_options.add_argument('--stream', '-S', action='store_true', default=False,
|
output_options.add_argument(
|
||||||
|
'--stream', '-S',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
help=_('''
|
help=_('''
|
||||||
Always stream the output by line, i.e., behave like `tail -f'.
|
Always stream the output by line, i.e., behave like `tail -f'.
|
||||||
|
|
||||||
@ -215,9 +216,45 @@ output_options.add_argument('--stream', '-S', action='store_true', default=False
|
|||||||
It is useful also without --pretty: It ensures that the output is flushed
|
It is useful also without --pretty: It ensures that the output is flushed
|
||||||
more often and in smaller chunks.
|
more often and in smaller chunks.
|
||||||
|
|
||||||
'''
|
''')
|
||||||
))
|
)
|
||||||
|
output_processing.add_argument(
|
||||||
|
'--output', '-o',
|
||||||
|
type=FileType('a+b'),
|
||||||
|
dest='output_file',
|
||||||
|
metavar='FILE',
|
||||||
|
help=_(
|
||||||
|
'''
|
||||||
|
Save output to FILE. If --download is set, then only the response
|
||||||
|
body is saved to the file. Other parts of the HTTP exchange are
|
||||||
|
printed to stderr.
|
||||||
|
|
||||||
|
'''
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
output_options.add_argument(
|
||||||
|
'--download', '-d',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help=_('''
|
||||||
|
Do not print the response body to stdout. Rather, download it and store it
|
||||||
|
in a file. The filename is guessed unless specified with --output
|
||||||
|
[filename]. This action is similar to the default behaviour of wget.
|
||||||
|
|
||||||
|
''')
|
||||||
|
)
|
||||||
|
|
||||||
|
output_options.add_argument(
|
||||||
|
'--continue', '-c',
|
||||||
|
dest='download_resume',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help=_('''
|
||||||
|
Resume an interrupted download.
|
||||||
|
The --output option needs to be specified as well.
|
||||||
|
''')
|
||||||
|
)
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
# Sessions
|
# Sessions
|
||||||
@ -226,19 +263,25 @@ sessions = parser.add_argument_group(title='Sessions')\
|
|||||||
.add_mutually_exclusive_group(required=False)
|
.add_mutually_exclusive_group(required=False)
|
||||||
|
|
||||||
sessions.add_argument(
|
sessions.add_argument(
|
||||||
'--session', metavar='SESSION_NAME',
|
'--session',
|
||||||
|
metavar='SESSION_NAME',
|
||||||
|
type=RegexValidator(
|
||||||
|
Session.VALID_NAME_PATTERN,
|
||||||
|
'Session name contains invalid characters.'
|
||||||
|
),
|
||||||
help=_('''
|
help=_('''
|
||||||
Create, or reuse and update a session.
|
Create, or reuse and update a session.
|
||||||
Withing a session, custom headers, auth credential, as well as any
|
Within a session, custom headers, auth credential, as well as any
|
||||||
cookies sent by the server persist between requests.
|
cookies sent by the server persist between requests.
|
||||||
You can use the `httpie' management command to manipulate
|
Session files are stored in %s/<HOST>/<SESSION_NAME>.json.
|
||||||
and inspect existing sessions. See `httpie --help'.
|
''' % DEFAULT_SESSIONS_DIR)
|
||||||
''')
|
|
||||||
)
|
)
|
||||||
sessions.add_argument(
|
sessions.add_argument(
|
||||||
'--session-read-only', metavar='SESSION_NAME',
|
'--session-read-only',
|
||||||
|
metavar='SESSION_NAME',
|
||||||
help=_('''
|
help=_('''
|
||||||
Create or reuse a session, but do not update it once saved.
|
Create or read a session without updating it form the
|
||||||
|
request/response exchange.
|
||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -249,7 +292,8 @@ sessions.add_argument(
|
|||||||
# ``requests.request`` keyword arguments.
|
# ``requests.request`` keyword arguments.
|
||||||
auth = parser.add_argument_group(title='Authentication')
|
auth = parser.add_argument_group(title='Authentication')
|
||||||
auth.add_argument(
|
auth.add_argument(
|
||||||
'--auth', '-a', metavar='USER[:PASS]',
|
'--auth', '-a',
|
||||||
|
metavar='USER[:PASS]',
|
||||||
type=AuthCredentialsArgType(SEP_CREDENTIALS),
|
type=AuthCredentialsArgType(SEP_CREDENTIALS),
|
||||||
help=_('''
|
help=_('''
|
||||||
If only the username is provided (-a username),
|
If only the username is provided (-a username),
|
||||||
@ -258,7 +302,9 @@ auth.add_argument(
|
|||||||
)
|
)
|
||||||
|
|
||||||
auth.add_argument(
|
auth.add_argument(
|
||||||
'--auth-type', choices=['basic', 'digest'], default='basic',
|
'--auth-type',
|
||||||
|
choices=['basic', 'digest'],
|
||||||
|
default='basic',
|
||||||
help=_('''
|
help=_('''
|
||||||
The authentication mechanism to be used.
|
The authentication mechanism to be used.
|
||||||
Defaults to "basic".
|
Defaults to "basic".
|
||||||
@ -266,14 +312,16 @@ auth.add_argument(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# Network
|
# Network
|
||||||
#############################################
|
#############################################
|
||||||
|
|
||||||
network = parser.add_argument_group(title='Network')
|
network = parser.add_argument_group(title='Network')
|
||||||
|
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--proxy', default=[], action='append', metavar='PROTOCOL:HOST',
|
'--proxy',
|
||||||
|
default=[],
|
||||||
|
action='append',
|
||||||
|
metavar='PROTOCOL:HOST',
|
||||||
type=KeyValueArgType(SEP_PROXY),
|
type=KeyValueArgType(SEP_PROXY),
|
||||||
help=_('''
|
help=_('''
|
||||||
String mapping protocol to the URL of the proxy
|
String mapping protocol to the URL of the proxy
|
||||||
@ -282,14 +330,17 @@ network.add_argument(
|
|||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--follow', default=False, action='store_true',
|
'--follow',
|
||||||
|
default=False,
|
||||||
|
action='store_true',
|
||||||
help=_('''
|
help=_('''
|
||||||
Set this flag if full redirects are allowed
|
Set this flag if full redirects are allowed
|
||||||
(e.g. re-POST-ing of data at new ``Location``)
|
(e.g. re-POST-ing of data at new ``Location``)
|
||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--verify', default='yes',
|
'--verify',
|
||||||
|
default='yes',
|
||||||
help=_('''
|
help=_('''
|
||||||
Set to "no" to skip checking the host\'s SSL certificate.
|
Set to "no" to skip checking the host\'s SSL certificate.
|
||||||
You can also pass the path to a CA_BUNDLE
|
You can also pass the path to a CA_BUNDLE
|
||||||
@ -300,14 +351,19 @@ network.add_argument(
|
|||||||
)
|
)
|
||||||
|
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--timeout', type=float, default=30, metavar='SECONDS',
|
'--timeout',
|
||||||
|
type=float,
|
||||||
|
default=30,
|
||||||
|
metavar='SECONDS',
|
||||||
help=_('''
|
help=_('''
|
||||||
The connection timeout of the request in seconds.
|
The connection timeout of the request in seconds.
|
||||||
The default value is 30 seconds.
|
The default value is 30 seconds.
|
||||||
''')
|
''')
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--check-status', default=False, action='store_true',
|
'--check-status',
|
||||||
|
default=False,
|
||||||
|
action='store_true',
|
||||||
help=_('''
|
help=_('''
|
||||||
By default, HTTPie exits with 0 when no network or other fatal
|
By default, HTTPie exits with 0 when no network or other fatal
|
||||||
errors occur.
|
errors occur.
|
||||||
@ -334,16 +390,25 @@ troubleshooting = parser.add_argument_group(title='Troubleshooting')
|
|||||||
|
|
||||||
troubleshooting.add_argument(
|
troubleshooting.add_argument(
|
||||||
'--help',
|
'--help',
|
||||||
action='help', default=SUPPRESS,
|
action='help',
|
||||||
|
default=SUPPRESS,
|
||||||
help='Show this help message and exit'
|
help='Show this help message and exit'
|
||||||
)
|
)
|
||||||
troubleshooting.add_argument('--version', action='version', version=__version__)
|
|
||||||
troubleshooting.add_argument(
|
troubleshooting.add_argument(
|
||||||
'--traceback', action='store_true', default=False,
|
'--version',
|
||||||
|
action='version',
|
||||||
|
version=__version__
|
||||||
|
)
|
||||||
|
troubleshooting.add_argument(
|
||||||
|
'--traceback',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
help='Prints exception traceback should one occur.'
|
help='Prints exception traceback should one occur.'
|
||||||
)
|
)
|
||||||
troubleshooting.add_argument(
|
troubleshooting.add_argument(
|
||||||
'--debug', action='store_true', default=False,
|
'--debug',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
help=_('''
|
help=_('''
|
||||||
Prints exception traceback should one occur, and also other
|
Prints exception traceback should one occur, and also other
|
||||||
information that is useful for debugging HTTPie itself and
|
information that is useful for debugging HTTPie itself and
|
||||||
|
@ -4,7 +4,6 @@ from pprint import pformat
|
|||||||
|
|
||||||
import requests
|
import requests
|
||||||
import requests.auth
|
import requests.auth
|
||||||
from requests.defaults import defaults
|
|
||||||
|
|
||||||
from . import sessions
|
from . import sessions
|
||||||
from . import __version__
|
from . import __version__
|
||||||
@ -21,41 +20,52 @@ def get_response(args, config_dir):
|
|||||||
requests_kwargs = get_requests_kwargs(args)
|
requests_kwargs = get_requests_kwargs(args)
|
||||||
|
|
||||||
if args.debug:
|
if args.debug:
|
||||||
sys.stderr.write(
|
sys.stderr.write('\n>>> requests.request(%s)\n\n'
|
||||||
'\n>>> requests.request(%s)\n\n' % pformat(requests_kwargs))
|
% pformat(requests_kwargs))
|
||||||
|
|
||||||
if not args.session and not args.session_read_only:
|
if not args.session and not args.session_read_only:
|
||||||
return requests.request(**requests_kwargs)
|
response = requests.request(**requests_kwargs)
|
||||||
else:
|
else:
|
||||||
return sessions.get_response(
|
response = sessions.get_response(
|
||||||
config_dir=config_dir,
|
config_dir=config_dir,
|
||||||
name=args.session or args.session_read_only,
|
name=args.session or args.session_read_only,
|
||||||
request_kwargs=requests_kwargs,
|
request_kwargs=requests_kwargs,
|
||||||
read_only=bool(args.session_read_only),
|
read_only=bool(args.session_read_only),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
def get_requests_kwargs(args):
|
def get_requests_kwargs(args):
|
||||||
"""Translate our `args` into `requests.request` keyword arguments."""
|
"""Translate our `args` into `requests.request` keyword arguments."""
|
||||||
|
|
||||||
base_headers = defaults['base_headers'].copy()
|
implicit_headers = {
|
||||||
base_headers['User-Agent'] = DEFAULT_UA
|
'User-Agent': DEFAULT_UA
|
||||||
|
}
|
||||||
|
|
||||||
auto_json = args.data and not args.form
|
auto_json = args.data and not args.form
|
||||||
|
# FIXME: Accept is set to JSON with `http url @./file.txt`.
|
||||||
if args.json or auto_json:
|
if args.json or auto_json:
|
||||||
base_headers['Accept'] = 'application/json'
|
implicit_headers['Accept'] = 'application/json'
|
||||||
if args.data:
|
if args.json or (auto_json and args.data):
|
||||||
base_headers['Content-Type'] = JSON
|
implicit_headers['Content-Type'] = JSON
|
||||||
|
|
||||||
if isinstance(args.data, dict):
|
if isinstance(args.data, dict):
|
||||||
# If not empty, serialize the data `dict` parsed from arguments.
|
if args.data:
|
||||||
# Otherwise set it to `None` avoid sending "{}".
|
args.data = json.dumps(args.data)
|
||||||
args.data = json.dumps(args.data) if args.data else None
|
else:
|
||||||
|
# We need to set data to an empty string to prevent requests
|
||||||
|
# from assigning an empty list to `response.request.data`.
|
||||||
|
args.data = ''
|
||||||
|
|
||||||
elif args.form and not args.files:
|
elif args.form and not args.files:
|
||||||
# If sending files, `requests` will set
|
# If sending files, `requests` will set
|
||||||
# the `Content-Type` for us.
|
# the `Content-Type` for us.
|
||||||
base_headers['Content-Type'] = FORM
|
implicit_headers['Content-Type'] = FORM
|
||||||
|
|
||||||
|
for name, value in implicit_headers.items():
|
||||||
|
if name not in args.headers:
|
||||||
|
args.headers[name] = value
|
||||||
|
|
||||||
credentials = None
|
credentials = None
|
||||||
if args.auth:
|
if args.auth:
|
||||||
@ -65,7 +75,7 @@ def get_requests_kwargs(args):
|
|||||||
}[args.auth_type](args.auth.key, args.auth.value)
|
}[args.auth_type](args.auth.key, args.auth.value)
|
||||||
|
|
||||||
kwargs = {
|
kwargs = {
|
||||||
'prefetch': False,
|
'stream': True,
|
||||||
'method': args.method.lower(),
|
'method': args.method.lower(),
|
||||||
'url': args.url,
|
'url': args.url,
|
||||||
'headers': args.headers,
|
'headers': args.headers,
|
||||||
@ -80,9 +90,6 @@ def get_requests_kwargs(args):
|
|||||||
'files': args.files,
|
'files': args.files,
|
||||||
'allow_redirects': args.follow,
|
'allow_redirects': args.follow,
|
||||||
'params': args.params,
|
'params': args.params,
|
||||||
'config': {
|
|
||||||
'base_headers': base_headers
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return kwargs
|
return kwargs
|
||||||
|
18
httpie/compat.py
Normal file
18
httpie/compat.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
"""
|
||||||
|
Python 2/3 compatibility.
|
||||||
|
|
||||||
|
"""
|
||||||
|
#noinspection PyUnresolvedReferences
|
||||||
|
from requests.compat import (
|
||||||
|
is_windows,
|
||||||
|
bytes,
|
||||||
|
str,
|
||||||
|
is_py3,
|
||||||
|
is_py26,
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
from urllib.parse import urlsplit
|
||||||
|
except ImportError:
|
||||||
|
from urlparse import urlsplit
|
||||||
|
|
@ -3,7 +3,7 @@ import json
|
|||||||
import errno
|
import errno
|
||||||
|
|
||||||
from . import __version__
|
from . import __version__
|
||||||
from requests.compat import is_windows
|
from .compat import is_windows
|
||||||
|
|
||||||
|
|
||||||
DEFAULT_CONFIG_DIR = os.environ.get(
|
DEFAULT_CONFIG_DIR = os.environ.get(
|
||||||
@ -17,7 +17,9 @@ class BaseConfigDict(dict):
|
|||||||
|
|
||||||
name = None
|
name = None
|
||||||
help = None
|
help = None
|
||||||
directory=DEFAULT_CONFIG_DIR
|
about = None
|
||||||
|
|
||||||
|
directory = DEFAULT_CONFIG_DIR
|
||||||
|
|
||||||
def __init__(self, directory=None, *args, **kwargs):
|
def __init__(self, directory=None, *args, **kwargs):
|
||||||
super(BaseConfigDict, self).__init__(*args, **kwargs)
|
super(BaseConfigDict, self).__init__(*args, **kwargs)
|
||||||
@ -56,7 +58,15 @@ class BaseConfigDict(dict):
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
def save(self):
|
def save(self):
|
||||||
self['__version__'] = __version__
|
self['__meta__'] = {
|
||||||
|
'httpie': __version__
|
||||||
|
}
|
||||||
|
if self.help:
|
||||||
|
self['__meta__']['help'] = self.help
|
||||||
|
|
||||||
|
if self.about:
|
||||||
|
self['__meta__']['about'] = self.about
|
||||||
|
|
||||||
with open(self.path, 'w') as f:
|
with open(self.path, 'w') as f:
|
||||||
json.dump(self, f, indent=4, sort_keys=True, ensure_ascii=True)
|
json.dump(self, f, indent=4, sort_keys=True, ensure_ascii=True)
|
||||||
f.write('\n')
|
f.write('\n')
|
||||||
@ -72,6 +82,8 @@ class BaseConfigDict(dict):
|
|||||||
class Config(BaseConfigDict):
|
class Config(BaseConfigDict):
|
||||||
|
|
||||||
name = 'config'
|
name = 'config'
|
||||||
|
help = 'https://github.com/jkbr/httpie#config'
|
||||||
|
about = 'HTTPie configuration file'
|
||||||
|
|
||||||
DEFAULTS = {
|
DEFAULTS = {
|
||||||
'implicit_content_type': 'json',
|
'implicit_content_type': 'json',
|
||||||
|
105
httpie/core.py
105
httpie/core.py
@ -14,31 +14,32 @@ import sys
|
|||||||
import errno
|
import errno
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
from requests.compat import str, is_py3
|
|
||||||
from httpie import __version__ as httpie_version
|
from httpie import __version__ as httpie_version
|
||||||
from requests import __version__ as requests_version
|
from requests import __version__ as requests_version
|
||||||
from pygments import __version__ as pygments_version
|
from pygments import __version__ as pygments_version
|
||||||
|
|
||||||
from .cli import parser
|
from .cli import parser
|
||||||
|
from .compat import str, is_py3
|
||||||
from .client import get_response
|
from .client import get_response
|
||||||
|
from .downloads import Download
|
||||||
from .models import Environment
|
from .models import Environment
|
||||||
from .output import output_stream, write, write_with_colors_win_p3k
|
from .output import build_output_stream, write, write_with_colors_win_py3
|
||||||
from . import EXIT
|
from . import ExitStatus
|
||||||
|
|
||||||
|
|
||||||
def get_exist_status(code, follow=False):
|
def get_exit_status(http_status, follow=False):
|
||||||
"""Translate HTTP status code to exit status."""
|
"""Translate HTTP status code to exit status code."""
|
||||||
if 300 <= code <= 399 and not follow:
|
if 300 <= http_status <= 399 and not follow:
|
||||||
# Redirect
|
# Redirect
|
||||||
return EXIT.ERROR_HTTP_3XX
|
return ExitStatus.ERROR_HTTP_3XX
|
||||||
elif 400 <= code <= 499:
|
elif 400 <= http_status <= 499:
|
||||||
# Client Error
|
# Client Error
|
||||||
return EXIT.ERROR_HTTP_4XX
|
return ExitStatus.ERROR_HTTP_4XX
|
||||||
elif 500 <= code <= 599:
|
elif 500 <= http_status <= 599:
|
||||||
# Server Error
|
# Server Error
|
||||||
return EXIT.ERROR_HTTP_5XX
|
return ExitStatus.ERROR_HTTP_5XX
|
||||||
else:
|
else:
|
||||||
return EXIT.OK
|
return ExitStatus.OK
|
||||||
|
|
||||||
|
|
||||||
def print_debug_info(env):
|
def print_debug_info(env):
|
||||||
@ -54,70 +55,110 @@ def print_debug_info(env):
|
|||||||
def main(args=sys.argv[1:], env=Environment()):
|
def main(args=sys.argv[1:], env=Environment()):
|
||||||
"""Run the main program and write the output to ``env.stdout``.
|
"""Run the main program and write the output to ``env.stdout``.
|
||||||
|
|
||||||
Return exit status.
|
Return exit status code.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
if env.config.default_options:
|
if env.config.default_options:
|
||||||
args = env.config.default_options + args
|
args = env.config.default_options + args
|
||||||
|
|
||||||
def error(msg, *args):
|
def error(msg, *args, **kwargs):
|
||||||
msg = msg % args
|
msg = msg % args
|
||||||
env.stderr.write('\nhttp: error: %s\n' % msg)
|
level = kwargs.get('level', 'error')
|
||||||
|
env.stderr.write('\nhttp: %s: %s\n' % (level, msg))
|
||||||
|
|
||||||
debug = '--debug' in args
|
debug = '--debug' in args
|
||||||
traceback = debug or '--traceback' in args
|
traceback = debug or '--traceback' in args
|
||||||
status = EXIT.OK
|
exit_status = ExitStatus.OK
|
||||||
|
|
||||||
if debug:
|
if debug:
|
||||||
print_debug_info(env)
|
print_debug_info(env)
|
||||||
if args == ['--debug']:
|
if args == ['--debug']:
|
||||||
sys.exit(EXIT.OK)
|
return exit_status
|
||||||
|
|
||||||
|
download = None
|
||||||
|
|
||||||
try:
|
try:
|
||||||
args = parser.parse_args(args=args, env=env)
|
args = parser.parse_args(args=args, env=env)
|
||||||
|
|
||||||
|
if args.download:
|
||||||
|
args.follow = True # --download implies --follow.
|
||||||
|
download = Download(
|
||||||
|
output_file=args.output_file,
|
||||||
|
progress_file=env.stderr,
|
||||||
|
resume=args.download_resume
|
||||||
|
)
|
||||||
|
download.pre_request(args.headers)
|
||||||
|
|
||||||
response = get_response(args, config_dir=env.config.directory)
|
response = get_response(args, config_dir=env.config.directory)
|
||||||
|
|
||||||
if args.check_status:
|
if args.check_status or download:
|
||||||
status = get_exist_status(response.status_code,
|
|
||||||
args.follow)
|
|
||||||
if status and not env.stdout_isatty:
|
|
||||||
error('%s %s', response.raw.status, response.raw.reason)
|
|
||||||
|
|
||||||
stream = output_stream(args, env, response.request, response)
|
exit_status = get_exit_status(
|
||||||
|
http_status=response.status_code,
|
||||||
|
follow=args.follow
|
||||||
|
)
|
||||||
|
|
||||||
|
if not env.stdout_isatty and exit_status != ExitStatus.OK:
|
||||||
|
error('HTTP %s %s',
|
||||||
|
response.raw.status,
|
||||||
|
response.raw.reason,
|
||||||
|
level='warning')
|
||||||
|
|
||||||
write_kwargs = {
|
write_kwargs = {
|
||||||
'stream': stream,
|
'stream': build_output_stream(
|
||||||
|
args, env, response.request, response),
|
||||||
|
|
||||||
|
# This will in fact be `stderr` with `--download`
|
||||||
'outfile': env.stdout,
|
'outfile': env.stdout,
|
||||||
|
|
||||||
'flush': env.stdout_isatty or args.stream
|
'flush': env.stdout_isatty or args.stream
|
||||||
}
|
}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
||||||
if env.is_windows and is_py3 and 'colors' in args.prettify:
|
if env.is_windows and is_py3 and 'colors' in args.prettify:
|
||||||
write_with_colors_win_p3k(**write_kwargs)
|
write_with_colors_win_py3(**write_kwargs)
|
||||||
else:
|
else:
|
||||||
write(**write_kwargs)
|
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,
|
||||||
|
)
|
||||||
|
download.finish()
|
||||||
|
if download.interrupted:
|
||||||
|
exit_status = ExitStatus.ERROR
|
||||||
|
|
||||||
except IOError as e:
|
except IOError as e:
|
||||||
if not traceback and e.errno == errno.EPIPE:
|
if not traceback and e.errno == errno.EPIPE:
|
||||||
# Ignore broken pipes unless --traceback.
|
# Ignore broken pipes unless --traceback.
|
||||||
env.stderr.write('\n')
|
env.stderr.write('\n')
|
||||||
else:
|
else:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
except (KeyboardInterrupt, SystemExit):
|
except (KeyboardInterrupt, SystemExit):
|
||||||
if traceback:
|
if traceback:
|
||||||
raise
|
raise
|
||||||
env.stderr.write('\n')
|
env.stderr.write('\n')
|
||||||
status = EXIT.ERROR
|
exit_status = ExitStatus.ERROR
|
||||||
|
|
||||||
except requests.Timeout:
|
except requests.Timeout:
|
||||||
status = EXIT.ERROR_TIMEOUT
|
exit_status = ExitStatus.ERROR_TIMEOUT
|
||||||
error('Request timed out (%ss).', args.timeout)
|
error('Request timed out (%ss).', args.timeout)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
# TODO: distinguish between expected and unexpected errors.
|
# TODO: Better distinction between expected and unexpected errors.
|
||||||
# network errors vs. bugs, etc.
|
# Network errors vs. bugs, etc.
|
||||||
if traceback:
|
if traceback:
|
||||||
raise
|
raise
|
||||||
error('%s: %s', type(e).__name__, str(e))
|
error('%s: %s', type(e).__name__, str(e))
|
||||||
status = EXIT.ERROR
|
exit_status = ExitStatus.ERROR
|
||||||
|
|
||||||
return status
|
finally:
|
||||||
|
if download and not download.finished:
|
||||||
|
download.failed()
|
||||||
|
|
||||||
|
return exit_status
|
||||||
|
423
httpie/downloads.py
Normal file
423
httpie/downloads.py
Normal file
@ -0,0 +1,423 @@
|
|||||||
|
# coding=utf-8
|
||||||
|
"""
|
||||||
|
Download mode implementation.
|
||||||
|
|
||||||
|
"""
|
||||||
|
from __future__ import division
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
import mimetypes
|
||||||
|
import threading
|
||||||
|
from time import time
|
||||||
|
|
||||||
|
from .output import RawStream
|
||||||
|
from .models import HTTPResponse
|
||||||
|
from .utils import humanize_bytes
|
||||||
|
from .compat import urlsplit
|
||||||
|
|
||||||
|
|
||||||
|
PARTIAL_CONTENT = 206
|
||||||
|
|
||||||
|
|
||||||
|
CLEAR_LINE = '\r\033[K'
|
||||||
|
PROGRESS = (
|
||||||
|
'{percentage: 6.2f} %'
|
||||||
|
' {downloaded: >10}'
|
||||||
|
' {speed: >10}/s'
|
||||||
|
' {eta: >8} ETA'
|
||||||
|
)
|
||||||
|
PROGRESS_NO_CONTENT_LENGTH = '{downloaded: >10} {speed: >10}/s'
|
||||||
|
SUMMARY = 'Done. {downloaded} in {time:0.5f}s ({speed}/s)\n'
|
||||||
|
SPINNER = '|/-\\'
|
||||||
|
|
||||||
|
|
||||||
|
class ContentRangeError(ValueError):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def parse_content_range(content_range, resumed_from):
|
||||||
|
"""
|
||||||
|
Parse and validate Content-Range header.
|
||||||
|
|
||||||
|
<http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html>
|
||||||
|
|
||||||
|
:param content_range: the value of a Content-Range response header
|
||||||
|
eg. "bytes 21010-47021/47022"
|
||||||
|
:param resumed_from: first byte pos. from the Range request header
|
||||||
|
:return: total size of the response body when fully downloaded.
|
||||||
|
|
||||||
|
"""
|
||||||
|
if content_range is None:
|
||||||
|
raise ContentRangeError('Missing Content-Range')
|
||||||
|
|
||||||
|
pattern = (
|
||||||
|
'^bytes (?P<first_byte_pos>\d+)-(?P<last_byte_pos>\d+)'
|
||||||
|
'/(\*|(?P<instance_length>\d+))$'
|
||||||
|
)
|
||||||
|
match = re.match(pattern, content_range)
|
||||||
|
|
||||||
|
if not match:
|
||||||
|
raise ContentRangeError(
|
||||||
|
'Invalid Content-Range format %r' % content_range)
|
||||||
|
|
||||||
|
content_range_dict = match.groupdict()
|
||||||
|
first_byte_pos = int(content_range_dict['first_byte_pos'])
|
||||||
|
last_byte_pos = int(content_range_dict['last_byte_pos'])
|
||||||
|
instance_length = (
|
||||||
|
int(content_range_dict['instance_length'])
|
||||||
|
if content_range_dict['instance_length']
|
||||||
|
else None
|
||||||
|
)
|
||||||
|
|
||||||
|
# "A byte-content-range-spec with a byte-range-resp-spec whose
|
||||||
|
# last- byte-pos value is less than its first-byte-pos value,
|
||||||
|
# or whose instance-length value is less than or equal to its
|
||||||
|
# last-byte-pos value, is invalid. The recipient of an invalid
|
||||||
|
# byte-content-range- spec MUST ignore it and any content
|
||||||
|
# transferred along with it."
|
||||||
|
if (first_byte_pos >= last_byte_pos
|
||||||
|
or (instance_length is not None
|
||||||
|
and instance_length <= last_byte_pos)):
|
||||||
|
raise ContentRangeError(
|
||||||
|
'Invalid Content-Range returned: %r' % content_range)
|
||||||
|
|
||||||
|
if (first_byte_pos != resumed_from
|
||||||
|
or (instance_length is not None
|
||||||
|
and last_byte_pos + 1 != instance_length)):
|
||||||
|
# Not what we asked for.
|
||||||
|
raise ContentRangeError(
|
||||||
|
'Unexpected Content-Range returned (%r)'
|
||||||
|
' for the requested Range ("bytes=%d-")'
|
||||||
|
% (content_range, resumed_from)
|
||||||
|
)
|
||||||
|
|
||||||
|
return last_byte_pos + 1
|
||||||
|
|
||||||
|
|
||||||
|
def filename_from_content_disposition(content_disposition):
|
||||||
|
"""
|
||||||
|
Extract and validate filename from a Content-Disposition header.
|
||||||
|
|
||||||
|
:param content_disposition: Content-Disposition value
|
||||||
|
:return: the filename if present and valid, otherwise `None`
|
||||||
|
|
||||||
|
"""
|
||||||
|
# attachment; filename=jkbr-httpie-0.4.1-20-g40bd8f6.tar.gz
|
||||||
|
match = re.search('filename=(\S+)', content_disposition)
|
||||||
|
if match and match.group(1):
|
||||||
|
fn = match.group(1).strip('."')
|
||||||
|
if re.match('^[a-zA-Z0-9._-]+$', fn):
|
||||||
|
return fn
|
||||||
|
|
||||||
|
|
||||||
|
def filename_from_url(url, content_type):
|
||||||
|
fn = urlsplit(url).path.rstrip('/')
|
||||||
|
fn = os.path.basename(fn) if fn else 'index'
|
||||||
|
if '.' not in fn and content_type:
|
||||||
|
content_type = content_type.split(';')[0]
|
||||||
|
if content_type == 'text/plain':
|
||||||
|
# mimetypes returns '.ksh'
|
||||||
|
ext = '.txt'
|
||||||
|
else:
|
||||||
|
ext = mimetypes.guess_extension(content_type)
|
||||||
|
|
||||||
|
if ext == '.htm': # Python 3
|
||||||
|
ext = '.html'
|
||||||
|
|
||||||
|
if ext:
|
||||||
|
fn += ext
|
||||||
|
|
||||||
|
return fn
|
||||||
|
|
||||||
|
|
||||||
|
def get_unique_filename(fn, exists=os.path.exists):
|
||||||
|
attempt = 0
|
||||||
|
while True:
|
||||||
|
suffix = '-' + str(attempt) if attempt > 0 else ''
|
||||||
|
if not exists(fn + suffix):
|
||||||
|
return fn + suffix
|
||||||
|
attempt += 1
|
||||||
|
|
||||||
|
|
||||||
|
class Download(object):
|
||||||
|
|
||||||
|
def __init__(self, output_file=None,
|
||||||
|
resume=False, progress_file=sys.stderr):
|
||||||
|
"""
|
||||||
|
:param resume: Should the download resume if partial download
|
||||||
|
already exists.
|
||||||
|
:type resume: bool
|
||||||
|
|
||||||
|
:param output_file: The file to store response body in. If not
|
||||||
|
provided, it will be guessed from the response.
|
||||||
|
:type output_file: file
|
||||||
|
|
||||||
|
:param progress_file: Where to report download progress.
|
||||||
|
:type progress_file: file
|
||||||
|
|
||||||
|
"""
|
||||||
|
self._output_file = output_file
|
||||||
|
self._resume = resume
|
||||||
|
self._resumed_from = 0
|
||||||
|
self.finished = False
|
||||||
|
|
||||||
|
self._status = Status()
|
||||||
|
self._progress_reporter = ProgressReporter(
|
||||||
|
status=self._status,
|
||||||
|
output=progress_file
|
||||||
|
)
|
||||||
|
|
||||||
|
def pre_request(self, request_headers):
|
||||||
|
"""Called just before the HTTP request is sent.
|
||||||
|
|
||||||
|
Might alter `request_headers`.
|
||||||
|
|
||||||
|
:type request_headers: dict
|
||||||
|
|
||||||
|
"""
|
||||||
|
# Disable content encoding so that we can resume, etc.
|
||||||
|
request_headers['Accept-Encoding'] = None
|
||||||
|
if self._resume:
|
||||||
|
bytes_have = os.path.getsize(self._output_file.name)
|
||||||
|
if bytes_have:
|
||||||
|
# Set ``Range`` header to resume the download
|
||||||
|
# TODO: Use "If-Range: mtime" to make sure it's fresh?
|
||||||
|
request_headers['Range'] = 'bytes=%d-' % bytes_have
|
||||||
|
self._resumed_from = bytes_have
|
||||||
|
|
||||||
|
def start(self, response):
|
||||||
|
"""
|
||||||
|
Initiate and return a stream for `response` body with progress
|
||||||
|
callback attached. Can be called only once.
|
||||||
|
|
||||||
|
:param response: Initiated response object with headers already fetched
|
||||||
|
:type response: requests.models.Response
|
||||||
|
|
||||||
|
:return: RawStream, output_file
|
||||||
|
|
||||||
|
"""
|
||||||
|
assert not self._status.time_started
|
||||||
|
|
||||||
|
try:
|
||||||
|
total_size = int(response.headers['Content-Length'])
|
||||||
|
except (KeyError, ValueError, TypeError):
|
||||||
|
total_size = None
|
||||||
|
|
||||||
|
if self._output_file:
|
||||||
|
if self._resume and response.status_code == PARTIAL_CONTENT:
|
||||||
|
total_size = parse_content_range(
|
||||||
|
response.headers.get('Content-Range'),
|
||||||
|
self._resumed_from
|
||||||
|
)
|
||||||
|
|
||||||
|
else:
|
||||||
|
self._resumed_from = 0
|
||||||
|
try:
|
||||||
|
self._output_file.seek(0)
|
||||||
|
self._output_file.truncate()
|
||||||
|
except IOError:
|
||||||
|
pass # stdout
|
||||||
|
else:
|
||||||
|
# TODO: Should the filename be taken from response.history[0].url?
|
||||||
|
# Output file not specified. Pick a name that doesn't exist yet.
|
||||||
|
fn = None
|
||||||
|
if 'Content-Disposition' in response.headers:
|
||||||
|
fn = filename_from_content_disposition(
|
||||||
|
response.headers['Content-Disposition'])
|
||||||
|
if not fn:
|
||||||
|
fn = filename_from_url(
|
||||||
|
url=response.url,
|
||||||
|
content_type=response.headers.get('Content-Type'),
|
||||||
|
)
|
||||||
|
self._output_file = open(get_unique_filename(fn), mode='a+b')
|
||||||
|
|
||||||
|
self._status.started(
|
||||||
|
resumed_from=self._resumed_from,
|
||||||
|
total_size=total_size
|
||||||
|
)
|
||||||
|
|
||||||
|
stream = RawStream(
|
||||||
|
msg=HTTPResponse(response),
|
||||||
|
with_headers=False,
|
||||||
|
with_body=True,
|
||||||
|
on_body_chunk_downloaded=self._chunk_downloaded,
|
||||||
|
chunk_size=1024 * 8
|
||||||
|
)
|
||||||
|
|
||||||
|
self._progress_reporter.output.write(
|
||||||
|
'Downloading %sto "%s"\n' % (
|
||||||
|
(humanize_bytes(total_size) + ' '
|
||||||
|
if total_size is not None
|
||||||
|
else ''),
|
||||||
|
self._output_file.name
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self._progress_reporter.report()
|
||||||
|
|
||||||
|
return stream, self._output_file
|
||||||
|
|
||||||
|
def finish(self):
|
||||||
|
assert not self.finished
|
||||||
|
self.finished = True
|
||||||
|
self._status.finished()
|
||||||
|
|
||||||
|
def failed(self):
|
||||||
|
self.finish()
|
||||||
|
self._progress_reporter.stop()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def interrupted(self):
|
||||||
|
return (
|
||||||
|
self.finished
|
||||||
|
and self._status.total_size
|
||||||
|
and self._status.total_size != self._status.downloaded
|
||||||
|
)
|
||||||
|
|
||||||
|
def _chunk_downloaded(self, chunk):
|
||||||
|
"""
|
||||||
|
A download progress callback.
|
||||||
|
|
||||||
|
:param chunk: A chunk of response body data that has just
|
||||||
|
been downloaded and written to the output.
|
||||||
|
:type chunk: bytes
|
||||||
|
|
||||||
|
"""
|
||||||
|
self._status.chunk_downloaded(len(chunk))
|
||||||
|
|
||||||
|
|
||||||
|
class Status(object):
|
||||||
|
"""Holds details about the downland status."""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.downloaded = 0
|
||||||
|
self.total_size = None
|
||||||
|
self.resumed_from = 0
|
||||||
|
self.time_started = None
|
||||||
|
self.time_finished = None
|
||||||
|
|
||||||
|
def started(self, resumed_from=0, total_size=None):
|
||||||
|
assert self.time_started is None
|
||||||
|
if total_size is not None:
|
||||||
|
self.total_size = total_size
|
||||||
|
self.downloaded = self.resumed_from = resumed_from
|
||||||
|
self.time_started = time()
|
||||||
|
|
||||||
|
def chunk_downloaded(self, size):
|
||||||
|
assert self.time_finished is None
|
||||||
|
self.downloaded += size
|
||||||
|
|
||||||
|
@property
|
||||||
|
def has_finished(self):
|
||||||
|
return self.time_finished is not None
|
||||||
|
|
||||||
|
def finished(self):
|
||||||
|
assert self.time_started is not None
|
||||||
|
assert self.time_finished is None
|
||||||
|
self.time_finished = time()
|
||||||
|
|
||||||
|
|
||||||
|
class ProgressReporter(object):
|
||||||
|
"""
|
||||||
|
Reports download progress based on its status.
|
||||||
|
|
||||||
|
Uses threading to periodically update the status (speed, ETA, etc.).
|
||||||
|
|
||||||
|
"""
|
||||||
|
def __init__(self, status, output, tick=.1, update_interval=1):
|
||||||
|
"""
|
||||||
|
|
||||||
|
:type status: Status
|
||||||
|
:type output: file
|
||||||
|
"""
|
||||||
|
self.status = status
|
||||||
|
self.output = output
|
||||||
|
self._tick = tick
|
||||||
|
self._update_interval = update_interval
|
||||||
|
self._spinner_pos = 0
|
||||||
|
self._status_line = ''
|
||||||
|
self._prev_bytes = 0
|
||||||
|
self._prev_time = time()
|
||||||
|
self._stop = False
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
"""Stop reporting on next tick."""
|
||||||
|
self._stop = True
|
||||||
|
|
||||||
|
def report(self):
|
||||||
|
if self._stop:
|
||||||
|
return
|
||||||
|
if self.status.has_finished:
|
||||||
|
self.sum_up()
|
||||||
|
else:
|
||||||
|
self.report_speed()
|
||||||
|
threading.Timer(self._tick, self.report).start()
|
||||||
|
|
||||||
|
def report_speed(self):
|
||||||
|
|
||||||
|
now = time()
|
||||||
|
|
||||||
|
if now - self._prev_time >= self._update_interval:
|
||||||
|
downloaded = self.status.downloaded
|
||||||
|
try:
|
||||||
|
speed = ((downloaded - self._prev_bytes)
|
||||||
|
/ (now - self._prev_time))
|
||||||
|
except ZeroDivisionError:
|
||||||
|
speed = 0
|
||||||
|
|
||||||
|
if not self.status.total_size:
|
||||||
|
self._status_line = PROGRESS_NO_CONTENT_LENGTH.format(
|
||||||
|
downloaded=humanize_bytes(downloaded),
|
||||||
|
speed=humanize_bytes(speed),
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
percentage = downloaded / self.status.total_size * 100
|
||||||
|
except ZeroDivisionError:
|
||||||
|
percentage = 0
|
||||||
|
|
||||||
|
if not speed:
|
||||||
|
eta = '-:--:--'
|
||||||
|
else:
|
||||||
|
s = int((self.status.total_size - downloaded) / speed)
|
||||||
|
h, s = divmod(s, 60 * 60)
|
||||||
|
m, s = divmod(s, 60)
|
||||||
|
eta = '{}:{:0>2}:{:0>2}'.format(h, m, s)
|
||||||
|
|
||||||
|
self._status_line = PROGRESS.format(
|
||||||
|
percentage=percentage,
|
||||||
|
downloaded=humanize_bytes(downloaded),
|
||||||
|
speed=humanize_bytes(speed),
|
||||||
|
eta=eta,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._prev_time = now
|
||||||
|
self._prev_bytes = downloaded
|
||||||
|
|
||||||
|
self.output.write(
|
||||||
|
CLEAR_LINE
|
||||||
|
+ ' '
|
||||||
|
+ SPINNER[self._spinner_pos]
|
||||||
|
+ ' '
|
||||||
|
+ self._status_line
|
||||||
|
)
|
||||||
|
self.output.flush()
|
||||||
|
|
||||||
|
self._spinner_pos = (self._spinner_pos + 1
|
||||||
|
if self._spinner_pos + 1 != len(SPINNER)
|
||||||
|
else 0)
|
||||||
|
|
||||||
|
def sum_up(self):
|
||||||
|
actually_downloaded = (self.status.downloaded
|
||||||
|
- self.status.resumed_from)
|
||||||
|
time_taken = self.status.time_finished - self.status.time_started
|
||||||
|
|
||||||
|
self.output.write(CLEAR_LINE)
|
||||||
|
self.output.write(SUMMARY.format(
|
||||||
|
downloaded=humanize_bytes(actually_downloaded),
|
||||||
|
total=(self.status.total_size
|
||||||
|
and humanize_bytes(self.status.total_size)),
|
||||||
|
speed=humanize_bytes(actually_downloaded / time_taken),
|
||||||
|
time=time_taken,
|
||||||
|
))
|
||||||
|
self.output.flush()
|
292
httpie/input.py
292
httpie/input.py
@ -8,15 +8,18 @@ import json
|
|||||||
import mimetypes
|
import mimetypes
|
||||||
import getpass
|
import getpass
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
from argparse import ArgumentParser, ArgumentTypeError
|
from argparse import ArgumentParser, ArgumentTypeError, ArgumentError
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
except ImportError:
|
except ImportError:
|
||||||
OrderedDict = dict
|
OrderedDict = dict
|
||||||
|
|
||||||
|
# TODO: Use MultiDict for headers once added to `requests`.
|
||||||
|
# https://github.com/jkbr/httpie/issues/130
|
||||||
from requests.structures import CaseInsensitiveDict
|
from requests.structures import CaseInsensitiveDict
|
||||||
from requests.compat import str, urlparse
|
|
||||||
|
from .compat import urlsplit, str
|
||||||
|
|
||||||
|
|
||||||
HTTP_POST = 'POST'
|
HTTP_POST = 'POST'
|
||||||
@ -95,37 +98,32 @@ class Parser(ArgumentParser):
|
|||||||
def parse_args(self, env, args=None, namespace=None):
|
def parse_args(self, env, args=None, namespace=None):
|
||||||
|
|
||||||
self.env = env
|
self.env = env
|
||||||
|
self.args, no_options = super(Parser, self)\
|
||||||
|
.parse_known_args(args, namespace)
|
||||||
|
|
||||||
args = super(Parser, self).parse_args(args, namespace)
|
if self.args.debug:
|
||||||
|
self.args.traceback = True
|
||||||
if not args.json and env.config.implicit_content_type == 'form':
|
|
||||||
args.form = True
|
|
||||||
|
|
||||||
if args.debug:
|
|
||||||
args.traceback = True
|
|
||||||
|
|
||||||
if args.output:
|
|
||||||
env.stdout = args.output
|
|
||||||
env.stdout_isatty = False
|
|
||||||
|
|
||||||
self._process_output_options(args, env)
|
|
||||||
self._process_pretty_options(args, env)
|
|
||||||
self._guess_method(args, env)
|
|
||||||
self._parse_items(args)
|
|
||||||
|
|
||||||
|
# Arguments processing and environment setup.
|
||||||
|
self._apply_no_options(no_options)
|
||||||
|
self._apply_config()
|
||||||
|
self._validate_download_options()
|
||||||
|
self._setup_standard_streams()
|
||||||
|
self._process_output_options()
|
||||||
|
self._process_pretty_options()
|
||||||
|
self._guess_method()
|
||||||
|
self._parse_items()
|
||||||
if not env.stdin_isatty:
|
if not env.stdin_isatty:
|
||||||
self._body_from_file(args, env.stdin)
|
self._body_from_file(self.env.stdin)
|
||||||
|
if not (self.args.url.startswith((HTTP, HTTPS))):
|
||||||
|
# Default to 'https://' if invoked as `https args`.
|
||||||
|
scheme = HTTPS if self.env.progname == 'https' else HTTP
|
||||||
|
self.args.url = scheme + self.args.url
|
||||||
|
self._process_auth()
|
||||||
|
|
||||||
if not (args.url.startswith(HTTP) or args.url.startswith(HTTPS)):
|
return self.args
|
||||||
scheme = HTTPS if env.progname == 'https' else HTTP
|
|
||||||
args.url = scheme + args.url
|
|
||||||
|
|
||||||
if args.auth and not args.auth.has_password():
|
|
||||||
# Stdin already read (if not a tty) so it's save to prompt.
|
|
||||||
args.auth.prompt_password(urlparse(args.url).netloc)
|
|
||||||
|
|
||||||
return args
|
|
||||||
|
|
||||||
|
# noinspection PyShadowingBuiltins
|
||||||
def _print_message(self, message, file=None):
|
def _print_message(self, message, file=None):
|
||||||
# Sneak in our stderr/stdout.
|
# Sneak in our stderr/stdout.
|
||||||
file = {
|
file = {
|
||||||
@ -136,113 +134,226 @@ class Parser(ArgumentParser):
|
|||||||
|
|
||||||
super(Parser, self)._print_message(message, file)
|
super(Parser, self)._print_message(message, file)
|
||||||
|
|
||||||
def _body_from_file(self, args, fd):
|
def _setup_standard_streams(self):
|
||||||
|
"""
|
||||||
|
Modify `env.stdout` and `env.stdout_isatty` based on args, if needed.
|
||||||
|
|
||||||
|
"""
|
||||||
|
if not self.env.stdout_isatty and self.args.output_file:
|
||||||
|
self.error('Cannot use --output, -o with redirected output.')
|
||||||
|
|
||||||
|
# FIXME: Come up with a cleaner solution.
|
||||||
|
if self.args.download:
|
||||||
|
|
||||||
|
if not self.env.stdout_isatty:
|
||||||
|
# Use stdout as tge download output file.
|
||||||
|
self.args.output_file = self.env.stdout
|
||||||
|
|
||||||
|
# With `--download`, we write everything that would normally go to
|
||||||
|
# `stdout` to `stderr` instead. Let's replace the stream so that
|
||||||
|
# we don't have to use many `if`s throughout the codebase.
|
||||||
|
# The response body will be treated separately.
|
||||||
|
self.env.stdout = self.env.stderr
|
||||||
|
self.env.stdout_isatty = self.env.stderr_isatty
|
||||||
|
|
||||||
|
elif self.args.output_file:
|
||||||
|
# When not `--download`ing, then `--output` simply replaces
|
||||||
|
# `stdout`. The file is opened for appending, which isn't what
|
||||||
|
# we want in this case.
|
||||||
|
self.args.output_file.seek(0)
|
||||||
|
self.args.output_file.truncate()
|
||||||
|
|
||||||
|
self.env.stdout = self.args.output_file
|
||||||
|
self.env.stdout_isatty = False
|
||||||
|
|
||||||
|
def _apply_config(self):
|
||||||
|
if (not self.args.json
|
||||||
|
and self.env.config.implicit_content_type == 'form'):
|
||||||
|
self.args.form = True
|
||||||
|
|
||||||
|
def _process_auth(self):
|
||||||
|
"""
|
||||||
|
If only a username provided via --auth, then ask for a password.
|
||||||
|
Or, take credentials from the URL, if provided.
|
||||||
|
|
||||||
|
"""
|
||||||
|
url = urlsplit(self.args.url)
|
||||||
|
|
||||||
|
if self.args.auth:
|
||||||
|
if not self.args.auth.has_password():
|
||||||
|
# Stdin already read (if not a tty) so it's save to prompt.
|
||||||
|
self.args.auth.prompt_password(url.netloc)
|
||||||
|
|
||||||
|
elif url.username is not None:
|
||||||
|
# Handle http://username:password@hostname/
|
||||||
|
username, password = url.username, url.password
|
||||||
|
self.args.auth = AuthCredentials(
|
||||||
|
key=username,
|
||||||
|
value=password,
|
||||||
|
sep=SEP_CREDENTIALS,
|
||||||
|
orig=SEP_CREDENTIALS.join([username, password])
|
||||||
|
)
|
||||||
|
|
||||||
|
def _apply_no_options(self, no_options):
|
||||||
|
"""For every `--no-OPTION` in `no_options`, set `args.OPTION` to
|
||||||
|
its default value. This allows for un-setting of options, e.g.,
|
||||||
|
specified in config.
|
||||||
|
|
||||||
|
"""
|
||||||
|
invalid = []
|
||||||
|
|
||||||
|
for option in no_options:
|
||||||
|
if not option.startswith('--no-'):
|
||||||
|
invalid.append(option)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# --no-option => --option
|
||||||
|
inverted = '--' + option[5:]
|
||||||
|
for action in self._actions:
|
||||||
|
if inverted in action.option_strings:
|
||||||
|
setattr(self.args, action.dest, action.default)
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
invalid.append(option)
|
||||||
|
|
||||||
|
if invalid:
|
||||||
|
msg = 'unrecognized arguments: %s'
|
||||||
|
self.error(msg % ' '.join(invalid))
|
||||||
|
|
||||||
|
def _body_from_file(self, fd):
|
||||||
"""There can only be one source of request data.
|
"""There can only be one source of request data.
|
||||||
|
|
||||||
Bytes are always read.
|
Bytes are always read.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
if args.data:
|
if self.args.data:
|
||||||
self.error('Request body (from stdin or a file) and request '
|
self.error('Request body (from stdin or a file) and request '
|
||||||
'data (key=value) cannot be mixed.')
|
'data (key=value) cannot be mixed.')
|
||||||
args.data = getattr(fd, 'buffer', fd).read()
|
self.args.data = getattr(fd, 'buffer', fd).read()
|
||||||
|
|
||||||
def _guess_method(self, args, env):
|
def _guess_method(self):
|
||||||
"""Set `args.method` if not specified to either POST or GET
|
"""Set `args.method` if not specified to either POST or GET
|
||||||
based on whether the request has data or not.
|
based on whether the request has data or not.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
if args.method is None:
|
if self.args.method is None:
|
||||||
# Invoked as `http URL'.
|
# Invoked as `http URL'.
|
||||||
assert not args.items
|
assert not self.args.items
|
||||||
if not env.stdin_isatty:
|
if not self.env.stdin_isatty:
|
||||||
args.method = HTTP_POST
|
self.args.method = HTTP_POST
|
||||||
else:
|
else:
|
||||||
args.method = HTTP_GET
|
self.args.method = HTTP_GET
|
||||||
|
|
||||||
# FIXME: False positive, e.g., "localhost" matches but is a valid URL.
|
# FIXME: False positive, e.g., "localhost" matches but is a valid URL.
|
||||||
elif not re.match('^[a-zA-Z]+$', args.method):
|
elif not re.match('^[a-zA-Z]+$', self.args.method):
|
||||||
# Invoked as `http URL item+'. The URL is now in `args.method`
|
# Invoked as `http URL item+'. The URL is now in `args.method`
|
||||||
# and the first ITEM is now incorrectly in `args.url`.
|
# and the first ITEM is now incorrectly in `args.url`.
|
||||||
try:
|
try:
|
||||||
# Parse the URL as an ITEM and store it as the first ITEM arg.
|
# Parse the URL as an ITEM and store it as the first ITEM arg.
|
||||||
args.items.insert(
|
self.args.items.insert(
|
||||||
0, KeyValueArgType(*SEP_GROUP_ITEMS).__call__(args.url))
|
0,
|
||||||
|
KeyValueArgType(*SEP_GROUP_ITEMS).__call__(self.args.url)
|
||||||
|
)
|
||||||
|
|
||||||
except ArgumentTypeError as e:
|
except ArgumentTypeError as e:
|
||||||
if args.traceback:
|
if self.args.traceback:
|
||||||
raise
|
raise
|
||||||
self.error(e.message)
|
self.error(e.message)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# Set the URL correctly
|
# Set the URL correctly
|
||||||
args.url = args.method
|
self.args.url = self.args.method
|
||||||
# Infer the method
|
# Infer the method
|
||||||
has_data = not env.stdin_isatty or any(
|
has_data = not self.env.stdin_isatty or any(
|
||||||
item.sep in SEP_GROUP_DATA_ITEMS for item in args.items)
|
item.sep in SEP_GROUP_DATA_ITEMS
|
||||||
args.method = HTTP_POST if has_data else HTTP_GET
|
for item in self.args.items
|
||||||
|
)
|
||||||
|
self.args.method = HTTP_POST if has_data else HTTP_GET
|
||||||
|
|
||||||
def _parse_items(self, args):
|
def _parse_items(self):
|
||||||
"""Parse `args.items` into `args.headers`, `args.data`,
|
"""Parse `args.items` into `args.headers`, `args.data`, `args.params`,
|
||||||
`args.`, and `args.files`.
|
and `args.files`.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
args.headers = CaseInsensitiveDict()
|
self.args.headers = CaseInsensitiveDict()
|
||||||
args.data = ParamDict() if args.form else OrderedDict()
|
self.args.data = ParamDict() if self.args.form else OrderedDict()
|
||||||
args.files = OrderedDict()
|
self.args.files = OrderedDict()
|
||||||
args.params = ParamDict()
|
self.args.params = ParamDict()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
parse_items(items=args.items,
|
parse_items(items=self.args.items,
|
||||||
headers=args.headers,
|
headers=self.args.headers,
|
||||||
data=args.data,
|
data=self.args.data,
|
||||||
files=args.files,
|
files=self.args.files,
|
||||||
params=args.params)
|
params=self.args.params)
|
||||||
except ParseError as e:
|
except ParseError as e:
|
||||||
if args.traceback:
|
if self.args.traceback:
|
||||||
raise
|
raise
|
||||||
self.error(e.message)
|
self.error(e.message)
|
||||||
|
|
||||||
if args.files and not args.form:
|
if self.args.files and not self.args.form:
|
||||||
# `http url @/path/to/file`
|
# `http url @/path/to/file`
|
||||||
file_fields = list(args.files.keys())
|
file_fields = list(self.args.files.keys())
|
||||||
if file_fields != ['']:
|
if file_fields != ['']:
|
||||||
self.error(
|
self.error(
|
||||||
'Invalid file fields (perhaps you meant --form?): %s'
|
'Invalid file fields (perhaps you meant --form?): %s'
|
||||||
% ','.join(file_fields))
|
% ','.join(file_fields))
|
||||||
|
|
||||||
fn, fd = args.files['']
|
fn, fd = self.args.files['']
|
||||||
args.files = {}
|
self.args.files = {}
|
||||||
self._body_from_file(args, fd)
|
|
||||||
if 'Content-Type' not in args.headers:
|
self._body_from_file(fd)
|
||||||
|
|
||||||
|
if 'Content-Type' not in self.args.headers:
|
||||||
mime, encoding = mimetypes.guess_type(fn, strict=False)
|
mime, encoding = mimetypes.guess_type(fn, strict=False)
|
||||||
if mime:
|
if mime:
|
||||||
content_type = mime
|
content_type = mime
|
||||||
if encoding:
|
if encoding:
|
||||||
content_type = '%s; charset=%s' % (mime, encoding)
|
content_type = '%s; charset=%s' % (mime, encoding)
|
||||||
args.headers['Content-Type'] = content_type
|
self.args.headers['Content-Type'] = content_type
|
||||||
|
|
||||||
def _process_output_options(self, args, env):
|
def _process_output_options(self):
|
||||||
"""Apply defaults to output options or validate the provided ones.
|
"""Apply defaults to output options, or validate the provided ones.
|
||||||
|
|
||||||
The default output options are stdout-type-sensitive.
|
The default output options are stdout-type-sensitive.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
if not args.output_options:
|
if not self.args.output_options:
|
||||||
args.output_options = (OUTPUT_OPTIONS_DEFAULT if env.stdout_isatty
|
self.args.output_options = (
|
||||||
else OUTPUT_OPTIONS_DEFAULT_STDOUT_REDIRECTED)
|
OUTPUT_OPTIONS_DEFAULT
|
||||||
|
if self.env.stdout_isatty
|
||||||
|
else OUTPUT_OPTIONS_DEFAULT_STDOUT_REDIRECTED
|
||||||
|
)
|
||||||
|
|
||||||
unknown = set(args.output_options) - OUTPUT_OPTIONS
|
unknown_output_options = set(self.args.output_options) - OUTPUT_OPTIONS
|
||||||
if unknown:
|
if unknown_output_options:
|
||||||
self.error('Unknown output options: %s' % ','.join(unknown))
|
self.error(
|
||||||
|
'Unknown output options: %s' % ','.join(unknown_output_options)
|
||||||
|
)
|
||||||
|
|
||||||
def _process_pretty_options(self, args, env):
|
if self.args.download and OUT_RESP_BODY in self.args.output_options:
|
||||||
if args.prettify == PRETTY_STDOUT_TTY_ONLY:
|
# Response body is always downloaded with --download and it goes
|
||||||
args.prettify = PRETTY_MAP['all' if env.stdout_isatty else 'none']
|
# through a different routine, so we remove it.
|
||||||
elif args.prettify and env.is_windows:
|
self.args.output_options = str(
|
||||||
|
set(self.args.output_options) - set(OUT_RESP_BODY))
|
||||||
|
|
||||||
|
def _process_pretty_options(self):
|
||||||
|
if self.args.prettify == PRETTY_STDOUT_TTY_ONLY:
|
||||||
|
self.args.prettify = PRETTY_MAP[
|
||||||
|
'all' if self.env.stdout_isatty else 'none']
|
||||||
|
elif self.args.prettify and self.env.is_windows:
|
||||||
self.error('Only terminal output can be colorized on Windows.')
|
self.error('Only terminal output can be colorized on Windows.')
|
||||||
else:
|
else:
|
||||||
args.prettify = PRETTY_MAP[args.prettify]
|
# noinspection PyTypeChecker
|
||||||
|
self.args.prettify = PRETTY_MAP[self.args.prettify]
|
||||||
|
|
||||||
|
def _validate_download_options(self):
|
||||||
|
if not self.args.download:
|
||||||
|
if self.args.download_resume:
|
||||||
|
self.error('--continue only works with --download')
|
||||||
|
if self.args.download_resume and not (
|
||||||
|
self.args.download and self.args.output_file):
|
||||||
|
self.error('--continue requires --output to be specified')
|
||||||
|
|
||||||
|
|
||||||
class ParseError(Exception):
|
class ParseError(Exception):
|
||||||
@ -262,6 +373,28 @@ class KeyValue(object):
|
|||||||
return self.__dict__ == other.__dict__
|
return self.__dict__ == other.__dict__
|
||||||
|
|
||||||
|
|
||||||
|
def session_name_arg_type(name):
|
||||||
|
from .sessions import Session
|
||||||
|
if not Session.is_valid_name(name):
|
||||||
|
raise ArgumentTypeError(
|
||||||
|
'special characters and spaces are not'
|
||||||
|
' allowed in session names: "%s"'
|
||||||
|
% name)
|
||||||
|
return name
|
||||||
|
|
||||||
|
|
||||||
|
class RegexValidator(object):
|
||||||
|
|
||||||
|
def __init__(self, pattern, error_message):
|
||||||
|
self.pattern = re.compile(pattern)
|
||||||
|
self.error_message = error_message
|
||||||
|
|
||||||
|
def __call__(self, value):
|
||||||
|
if not self.pattern.search(value):
|
||||||
|
raise ArgumentError(None, self.error_message)
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
class KeyValueArgType(object):
|
class KeyValueArgType(object):
|
||||||
"""A key-value pair argument type used with `argparse`.
|
"""A key-value pair argument type used with `argparse`.
|
||||||
|
|
||||||
@ -403,9 +536,6 @@ class ParamDict(OrderedDict):
|
|||||||
data and URL params.
|
data and URL params.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
# NOTE: Won't work when used for form data with multiple values
|
|
||||||
# for a field and a file field is present:
|
|
||||||
# https://github.com/kennethreitz/requests/issues/737
|
|
||||||
if key not in self:
|
if key not in self:
|
||||||
super(ParamDict, self).__setitem__(key, value)
|
super(ParamDict, self).__setitem__(key, value)
|
||||||
else:
|
else:
|
||||||
|
@ -1,30 +0,0 @@
|
|||||||
"""
|
|
||||||
Provides the `httpie' management command.
|
|
||||||
|
|
||||||
Note that the main `http' command points to `httpie.__main__.main()`.
|
|
||||||
|
|
||||||
"""
|
|
||||||
import argparse
|
|
||||||
|
|
||||||
from . import sessions
|
|
||||||
from . import __version__
|
|
||||||
|
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(
|
|
||||||
description='The HTTPie management command.',
|
|
||||||
version=__version__
|
|
||||||
)
|
|
||||||
subparsers = parser.add_subparsers()
|
|
||||||
|
|
||||||
|
|
||||||
# Only sessions as of now.
|
|
||||||
sessions.add_commands(subparsers)
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
args = parser.parse_args()
|
|
||||||
args.command(args)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
@ -1,8 +1,8 @@
|
|||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
from requests.compat import urlparse, is_windows, bytes, str
|
|
||||||
|
|
||||||
from .config import DEFAULT_CONFIG_DIR, Config
|
from .config import DEFAULT_CONFIG_DIR, Config
|
||||||
|
from .compat import urlsplit, is_windows, bytes, str
|
||||||
|
|
||||||
|
|
||||||
class Environment(object):
|
class Environment(object):
|
||||||
@ -13,30 +13,33 @@ class Environment(object):
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
#noinspection PyUnresolvedReferences
|
|
||||||
is_windows = is_windows
|
is_windows = is_windows
|
||||||
|
|
||||||
progname = os.path.basename(sys.argv[0])
|
progname = os.path.basename(sys.argv[0])
|
||||||
if progname not in ['http', 'https']:
|
if progname not in ['http', 'https']:
|
||||||
progname = 'http'
|
progname = 'http'
|
||||||
|
|
||||||
stdin_isatty = sys.stdin.isatty()
|
|
||||||
stdin = sys.stdin
|
|
||||||
stdout_isatty = sys.stdout.isatty()
|
|
||||||
|
|
||||||
config_dir = DEFAULT_CONFIG_DIR
|
config_dir = DEFAULT_CONFIG_DIR
|
||||||
|
|
||||||
if stdout_isatty and is_windows:
|
|
||||||
from colorama.initialise import wrap_stream
|
|
||||||
stdout = wrap_stream(sys.stdout, convert=None,
|
|
||||||
strip=None, autoreset=True, wrap=True)
|
|
||||||
else:
|
|
||||||
stdout = sys.stdout
|
|
||||||
stderr = sys.stderr
|
|
||||||
|
|
||||||
# Can be set to 0 to disable colors completely.
|
# Can be set to 0 to disable colors completely.
|
||||||
colors = 256 if '256color' in os.environ.get('TERM', '') else 88
|
colors = 256 if '256color' in os.environ.get('TERM', '') else 88
|
||||||
|
|
||||||
|
stdin = sys.stdin
|
||||||
|
stdin_isatty = sys.stdin.isatty()
|
||||||
|
|
||||||
|
stdout_isatty = sys.stdout.isatty()
|
||||||
|
stderr_isatty = sys.stderr.isatty()
|
||||||
|
if is_windows:
|
||||||
|
# noinspection PyUnresolvedReferences
|
||||||
|
from colorama.initialise import wrap_stream
|
||||||
|
stdout = wrap_stream(sys.stdout, convert=None,
|
||||||
|
strip=None, autoreset=True, wrap=True)
|
||||||
|
stderr = wrap_stream(sys.stderr, convert=None,
|
||||||
|
strip=None, autoreset=True, wrap=True)
|
||||||
|
else:
|
||||||
|
stdout = sys.stdout
|
||||||
|
stderr = sys.stderr
|
||||||
|
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
assert all(hasattr(type(self), attr)
|
assert all(hasattr(type(self), attr)
|
||||||
for attr in kwargs.keys())
|
for attr in kwargs.keys())
|
||||||
@ -85,7 +88,13 @@ class HTTPMessage(object):
|
|||||||
@property
|
@property
|
||||||
def content_type(self):
|
def content_type(self):
|
||||||
"""Return the message content type."""
|
"""Return the message content type."""
|
||||||
ct = self._orig.headers.get('Content-Type', '')
|
ct = self._orig.headers.get(
|
||||||
|
b'Content-Type',
|
||||||
|
self._orig.headers.get(
|
||||||
|
'Content-Type',
|
||||||
|
''
|
||||||
|
)
|
||||||
|
)
|
||||||
if isinstance(ct, bytes):
|
if isinstance(ct, bytes):
|
||||||
ct = ct.decode()
|
ct = ct.decode()
|
||||||
return ct
|
return ct
|
||||||
@ -143,33 +152,18 @@ class HTTPRequest(HTTPMessage):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def headers(self):
|
def headers(self):
|
||||||
"""Return Request-Line"""
|
url = urlsplit(self._orig.url)
|
||||||
url = urlparse(self._orig.url)
|
|
||||||
|
|
||||||
# Querystring
|
|
||||||
qs = ''
|
|
||||||
if url.query or self._orig.params:
|
|
||||||
qs = '?'
|
|
||||||
if url.query:
|
|
||||||
qs += url.query
|
|
||||||
# Requests doesn't make params part of ``request.url``.
|
|
||||||
if self._orig.params:
|
|
||||||
if url.query:
|
|
||||||
qs += '&'
|
|
||||||
#noinspection PyUnresolvedReferences
|
|
||||||
qs += type(self._orig)._encode_params(self._orig.params)
|
|
||||||
|
|
||||||
# Request-Line
|
|
||||||
request_line = '{method} {path}{query} HTTP/1.1'.format(
|
request_line = '{method} {path}{query} HTTP/1.1'.format(
|
||||||
method=self._orig.method,
|
method=self._orig.method,
|
||||||
path=url.path or '/',
|
path=url.path or '/',
|
||||||
query=qs
|
query='?' + url.query if url.query else ''
|
||||||
)
|
)
|
||||||
|
|
||||||
headers = dict(self._orig.headers)
|
headers = dict(self._orig.headers)
|
||||||
|
|
||||||
if 'Host' not in headers:
|
if 'Host' not in headers:
|
||||||
headers['Host'] = urlparse(self._orig.url).netloc
|
headers['Host'] = url.netloc
|
||||||
|
|
||||||
headers = ['%s: %s' % (name, value)
|
headers = ['%s: %s' % (name, value)
|
||||||
for name, value in headers.items()]
|
for name, value in headers.items()]
|
||||||
@ -184,26 +178,8 @@ class HTTPRequest(HTTPMessage):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def body(self):
|
def body(self):
|
||||||
"""Reconstruct and return the original request body bytes."""
|
body = self._orig.body
|
||||||
if self._orig.files:
|
if isinstance(body, str):
|
||||||
# TODO: would be nice if we didn't need to encode the files again
|
# Happens with JSON/form request data parsed from the command line.
|
||||||
# FIXME: Also the boundary header doesn't match the one used.
|
body = body.encode('utf8')
|
||||||
for fn, fd in self._orig.files.values():
|
return body or b''
|
||||||
# Rewind the files as they have already been read before.
|
|
||||||
fd.seek(0)
|
|
||||||
body, _ = self._orig._encode_files(self._orig.files)
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
body = self._orig.data
|
|
||||||
except AttributeError:
|
|
||||||
# requests < 0.12.1
|
|
||||||
body = self._orig._enc_data
|
|
||||||
|
|
||||||
if isinstance(body, dict):
|
|
||||||
#noinspection PyUnresolvedReferences
|
|
||||||
body = type(self._orig)._encode_params(body)
|
|
||||||
|
|
||||||
if isinstance(body, str):
|
|
||||||
body = body.encode('utf8')
|
|
||||||
|
|
||||||
return body
|
|
||||||
|
@ -12,8 +12,8 @@ from pygments.lexers import get_lexer_for_mimetype, get_lexer_by_name
|
|||||||
from pygments.formatters.terminal import TerminalFormatter
|
from pygments.formatters.terminal import TerminalFormatter
|
||||||
from pygments.formatters.terminal256 import Terminal256Formatter
|
from pygments.formatters.terminal256 import Terminal256Formatter
|
||||||
from pygments.util import ClassNotFound
|
from pygments.util import ClassNotFound
|
||||||
from requests.compat import is_windows
|
|
||||||
|
|
||||||
|
from .compat import is_windows
|
||||||
from .solarized import Solarized256Style
|
from .solarized import Solarized256Style
|
||||||
from .models import HTTPRequest, HTTPResponse, Environment
|
from .models import HTTPRequest, HTTPResponse, Environment
|
||||||
from .input import (OUT_REQ_BODY, OUT_REQ_HEAD,
|
from .input import (OUT_REQ_BODY, OUT_REQ_HEAD,
|
||||||
@ -61,7 +61,7 @@ def write(stream, outfile, flush):
|
|||||||
outfile.flush()
|
outfile.flush()
|
||||||
|
|
||||||
|
|
||||||
def write_with_colors_win_p3k(stream, outfile, flush):
|
def write_with_colors_win_py3(stream, outfile, flush):
|
||||||
"""Like `write`, but colorized chunks are written as text
|
"""Like `write`, but colorized chunks are written as text
|
||||||
directly to `outfile` to ensure it gets processed by colorama.
|
directly to `outfile` to ensure it gets processed by colorama.
|
||||||
Applies only to Windows with Python 3 and colorized terminal output.
|
Applies only to Windows with Python 3 and colorized terminal output.
|
||||||
@ -78,23 +78,21 @@ def write_with_colors_win_p3k(stream, outfile, flush):
|
|||||||
outfile.flush()
|
outfile.flush()
|
||||||
|
|
||||||
|
|
||||||
def output_stream(args, env, request, response):
|
def build_output_stream(args, env, request, response):
|
||||||
"""Build and return a chain of iterators over the `request`-`response`
|
"""Build and return a chain of iterators over the `request`-`response`
|
||||||
exchange each of which yields `bytes` chunks.
|
exchange each of which yields `bytes` chunks.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
Stream = make_stream(env, args)
|
|
||||||
|
|
||||||
req_h = OUT_REQ_HEAD in args.output_options
|
req_h = OUT_REQ_HEAD in args.output_options
|
||||||
req_b = OUT_REQ_BODY in args.output_options
|
req_b = OUT_REQ_BODY in args.output_options
|
||||||
resp_h = OUT_RESP_HEAD in args.output_options
|
resp_h = OUT_RESP_HEAD in args.output_options
|
||||||
resp_b = OUT_RESP_BODY in args.output_options
|
resp_b = OUT_RESP_BODY in args.output_options
|
||||||
|
|
||||||
req = req_h or req_b
|
req = req_h or req_b
|
||||||
resp = resp_h or resp_b
|
resp = resp_h or resp_b
|
||||||
|
|
||||||
output = []
|
output = []
|
||||||
|
Stream = get_stream_type(env, args)
|
||||||
|
|
||||||
if req:
|
if req:
|
||||||
output.append(Stream(
|
output.append(Stream(
|
||||||
@ -120,7 +118,7 @@ def output_stream(args, env, request, response):
|
|||||||
return chain(*output)
|
return chain(*output)
|
||||||
|
|
||||||
|
|
||||||
def make_stream(env, args):
|
def get_stream_type(env, args):
|
||||||
"""Pick the right stream type based on `env` and `args`.
|
"""Pick the right stream type based on `env` and `args`.
|
||||||
Wrap it in a partial with the type-specific args so that
|
Wrap it in a partial with the type-specific args so that
|
||||||
we don't need to think what stream we are dealing with.
|
we don't need to think what stream we are dealing with.
|
||||||
@ -147,37 +145,42 @@ def make_stream(env, args):
|
|||||||
|
|
||||||
|
|
||||||
class BaseStream(object):
|
class BaseStream(object):
|
||||||
"""Base HTTP message stream class."""
|
"""Base HTTP message output stream class."""
|
||||||
|
|
||||||
def __init__(self, msg, with_headers=True, with_body=True):
|
def __init__(self, msg, with_headers=True, with_body=True,
|
||||||
|
on_body_chunk_downloaded=None):
|
||||||
"""
|
"""
|
||||||
:param msg: a :class:`models.HTTPMessage` subclass
|
:param msg: a :class:`models.HTTPMessage` subclass
|
||||||
:param with_headers: if `True`, headers will be included
|
:param with_headers: if `True`, headers will be included
|
||||||
:param with_body: if `True`, body will be included
|
:param with_body: if `True`, body will be included
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
assert with_headers or with_body
|
||||||
self.msg = msg
|
self.msg = msg
|
||||||
self.with_headers = with_headers
|
self.with_headers = with_headers
|
||||||
self.with_body = with_body
|
self.with_body = with_body
|
||||||
|
self.on_body_chunk_downloaded = on_body_chunk_downloaded
|
||||||
|
|
||||||
def _headers(self):
|
def _get_headers(self):
|
||||||
"""Return the headers' bytes."""
|
"""Return the headers' bytes."""
|
||||||
return self.msg.headers.encode('ascii')
|
return self.msg.headers.encode('ascii')
|
||||||
|
|
||||||
def _body(self):
|
def _iter_body(self):
|
||||||
"""Return an iterator over the message body."""
|
"""Return an iterator over the message body."""
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
def __iter__(self):
|
def __iter__(self):
|
||||||
"""Return an iterator over `self.msg`."""
|
"""Return an iterator over `self.msg`."""
|
||||||
if self.with_headers:
|
if self.with_headers:
|
||||||
yield self._headers()
|
yield self._get_headers()
|
||||||
yield b'\r\n\r\n'
|
yield b'\r\n\r\n'
|
||||||
|
|
||||||
if self.with_body:
|
if self.with_body:
|
||||||
try:
|
try:
|
||||||
for chunk in self._body():
|
for chunk in self._iter_body():
|
||||||
yield chunk
|
yield chunk
|
||||||
|
if self.on_body_chunk_downloaded:
|
||||||
|
self.on_body_chunk_downloaded(chunk)
|
||||||
except BinarySuppressedError as e:
|
except BinarySuppressedError as e:
|
||||||
if self.with_headers:
|
if self.with_headers:
|
||||||
yield b'\n'
|
yield b'\n'
|
||||||
@ -188,13 +191,13 @@ class RawStream(BaseStream):
|
|||||||
"""The message is streamed in chunks with no processing."""
|
"""The message is streamed in chunks with no processing."""
|
||||||
|
|
||||||
CHUNK_SIZE = 1024 * 100
|
CHUNK_SIZE = 1024 * 100
|
||||||
CHUNK_SIZE_BY_LINE = 1024 * 5
|
CHUNK_SIZE_BY_LINE = 1
|
||||||
|
|
||||||
def __init__(self, chunk_size=CHUNK_SIZE, **kwargs):
|
def __init__(self, chunk_size=CHUNK_SIZE, **kwargs):
|
||||||
super(RawStream, self).__init__(**kwargs)
|
super(RawStream, self).__init__(**kwargs)
|
||||||
self.chunk_size = chunk_size
|
self.chunk_size = chunk_size
|
||||||
|
|
||||||
def _body(self):
|
def _iter_body(self):
|
||||||
return self.msg.iter_body(self.chunk_size)
|
return self.msg.iter_body(self.chunk_size)
|
||||||
|
|
||||||
|
|
||||||
@ -206,7 +209,7 @@ class EncodedStream(BaseStream):
|
|||||||
is suppressed. The body is always streamed by line.
|
is suppressed. The body is always streamed by line.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
CHUNK_SIZE = 1024 * 5
|
CHUNK_SIZE = 1
|
||||||
|
|
||||||
def __init__(self, env=Environment(), **kwargs):
|
def __init__(self, env=Environment(), **kwargs):
|
||||||
|
|
||||||
@ -222,7 +225,7 @@ class EncodedStream(BaseStream):
|
|||||||
# Default to utf8 when unsure.
|
# Default to utf8 when unsure.
|
||||||
self.output_encoding = output_encoding or 'utf8'
|
self.output_encoding = output_encoding or 'utf8'
|
||||||
|
|
||||||
def _body(self):
|
def _iter_body(self):
|
||||||
|
|
||||||
for line, lf in self.msg.iter_lines(self.CHUNK_SIZE):
|
for line, lf in self.msg.iter_lines(self.CHUNK_SIZE):
|
||||||
|
|
||||||
@ -242,17 +245,17 @@ class PrettyStream(EncodedStream):
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
CHUNK_SIZE = 1024 * 5
|
CHUNK_SIZE = 1
|
||||||
|
|
||||||
def __init__(self, processor, **kwargs):
|
def __init__(self, processor, **kwargs):
|
||||||
super(PrettyStream, self).__init__(**kwargs)
|
super(PrettyStream, self).__init__(**kwargs)
|
||||||
self.processor = processor
|
self.processor = processor
|
||||||
|
|
||||||
def _headers(self):
|
def _get_headers(self):
|
||||||
return self.processor.process_headers(
|
return self.processor.process_headers(
|
||||||
self.msg.headers).encode(self.output_encoding)
|
self.msg.headers).encode(self.output_encoding)
|
||||||
|
|
||||||
def _body(self):
|
def _iter_body(self):
|
||||||
for line, lf in self.msg.iter_lines(self.CHUNK_SIZE):
|
for line, lf in self.msg.iter_lines(self.CHUNK_SIZE):
|
||||||
if b'\0' in line:
|
if b'\0' in line:
|
||||||
raise BinarySuppressedError()
|
raise BinarySuppressedError()
|
||||||
@ -276,9 +279,8 @@ class BufferedPrettyStream(PrettyStream):
|
|||||||
|
|
||||||
CHUNK_SIZE = 1024 * 10
|
CHUNK_SIZE = 1024 * 10
|
||||||
|
|
||||||
def _body(self):
|
def _iter_body(self):
|
||||||
|
|
||||||
#noinspection PyArgumentList
|
|
||||||
# Read the whole body before prettifying it,
|
# Read the whole body before prettifying it,
|
||||||
# but bail out immediately if the body is binary.
|
# but bail out immediately if the body is binary.
|
||||||
body = bytearray()
|
body = bytearray()
|
||||||
|
@ -1,25 +1,22 @@
|
|||||||
"""Persistent, JSON-serialized sessions.
|
"""Persistent, JSON-serialized sessions.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
import re
|
||||||
import os
|
import os
|
||||||
import sys
|
|
||||||
import glob
|
import glob
|
||||||
import errno
|
import errno
|
||||||
import codecs
|
|
||||||
import shutil
|
import shutil
|
||||||
import subprocess
|
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
from requests.compat import urlparse
|
|
||||||
from requests.cookies import RequestsCookieJar, create_cookie
|
from requests.cookies import RequestsCookieJar, create_cookie
|
||||||
from requests.auth import HTTPBasicAuth, HTTPDigestAuth
|
from requests.auth import HTTPBasicAuth, HTTPDigestAuth
|
||||||
from argparse import OPTIONAL
|
|
||||||
|
|
||||||
|
from .compat import urlsplit
|
||||||
from .config import BaseConfigDict, DEFAULT_CONFIG_DIR
|
from .config import BaseConfigDict, DEFAULT_CONFIG_DIR
|
||||||
from .output import PygmentsProcessor
|
|
||||||
|
|
||||||
|
|
||||||
SESSIONS_DIR_NAME = 'sessions'
|
SESSIONS_DIR_NAME = 'sessions'
|
||||||
|
DEFAULT_SESSIONS_DIR = os.path.join(DEFAULT_CONFIG_DIR, SESSIONS_DIR_NAME)
|
||||||
|
|
||||||
|
|
||||||
def get_response(name, request_kwargs, config_dir, read_only=False):
|
def get_response(name, request_kwargs, config_dir, read_only=False):
|
||||||
@ -31,9 +28,8 @@ def get_response(name, request_kwargs, config_dir, read_only=False):
|
|||||||
host = Host(
|
host = Host(
|
||||||
root_dir=sessions_dir,
|
root_dir=sessions_dir,
|
||||||
name=request_kwargs['headers'].get('Host', None)
|
name=request_kwargs['headers'].get('Host', None)
|
||||||
or urlparse(request_kwargs['url']).netloc.split('@')[-1]
|
or urlsplit(request_kwargs['url']).netloc.split('@')[-1]
|
||||||
)
|
)
|
||||||
|
|
||||||
session = Session(host, name)
|
session = Session(host, name)
|
||||||
session.load()
|
session.load()
|
||||||
|
|
||||||
@ -48,15 +44,17 @@ def get_response(name, request_kwargs, config_dir, read_only=False):
|
|||||||
elif session.auth:
|
elif session.auth:
|
||||||
request_kwargs['auth'] = session.auth
|
request_kwargs['auth'] = session.auth
|
||||||
|
|
||||||
rsession = requests.Session(cookies=session.cookies)
|
requests_session = requests.Session()
|
||||||
|
requests_session.cookies = session.cookies
|
||||||
|
|
||||||
try:
|
try:
|
||||||
response = rsession.request(**request_kwargs)
|
response = requests_session.request(**request_kwargs)
|
||||||
except Exception:
|
except Exception:
|
||||||
raise
|
raise
|
||||||
else:
|
else:
|
||||||
# Existing sessions with `read_only=True` don't get updated.
|
# Existing sessions with `read_only=True` don't get updated.
|
||||||
if session.is_new or not read_only:
|
if session.is_new or not read_only:
|
||||||
session.cookies = rsession.cookies
|
session.cookies = requests_session.cookies
|
||||||
session.save()
|
session.save()
|
||||||
return response
|
return response
|
||||||
|
|
||||||
@ -64,24 +62,46 @@ def get_response(name, request_kwargs, config_dir, read_only=False):
|
|||||||
class Host(object):
|
class Host(object):
|
||||||
"""A host is a per-host directory on the disk containing sessions files."""
|
"""A host is a per-host directory on the disk containing sessions files."""
|
||||||
|
|
||||||
def __init__(self, name, root_dir=DEFAULT_CONFIG_DIR):
|
VALID_NAME_PATTERN = re.compile('^[a-zA-Z0-9_.:-]+$')
|
||||||
|
|
||||||
|
def __init__(self, name, root_dir=DEFAULT_SESSIONS_DIR):
|
||||||
|
assert self.VALID_NAME_PATTERN.match(name)
|
||||||
self.name = name
|
self.name = name
|
||||||
self.root_dir = root_dir
|
self.root_dir = root_dir
|
||||||
|
|
||||||
def __iter__(self):
|
def __iter__(self):
|
||||||
"""Return a iterator yielding `(session_name, session_path)`."""
|
"""Return an iterator yielding `Session` instances."""
|
||||||
for fn in sorted(glob.glob1(self.path, '*.json')):
|
for fn in sorted(glob.glob1(self.path, '*.json')):
|
||||||
yield os.path.splitext(fn)[0], os.path.join(self.path, fn)
|
session_name = os.path.splitext(fn)[0]
|
||||||
|
yield Session(host=self, name=session_name)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _quote_name(name):
|
||||||
|
"""host:port => host_port"""
|
||||||
|
return name.replace(':', '_')
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _unquote_name(name):
|
||||||
|
"""host_port => host:port"""
|
||||||
|
return re.sub(r'_(\d+)$', r':\1', name)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def all(cls, root_dir=DEFAULT_SESSIONS_DIR):
|
||||||
|
"""Return a generator yielding a host at a time."""
|
||||||
|
for name in sorted(glob.glob1(root_dir, '*')):
|
||||||
|
if os.path.isdir(os.path.join(root_dir, name)):
|
||||||
|
yield Host(cls._unquote_name(name), root_dir=root_dir)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def verbose_name(self):
|
||||||
|
return '%s %s' % (self.name, self.path)
|
||||||
|
|
||||||
def delete(self):
|
def delete(self):
|
||||||
shutil.rmtree(self.path)
|
shutil.rmtree(self.path)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def path(self):
|
def path(self):
|
||||||
# Name will include ':' if a port is specified, which is invalid
|
path = os.path.join(self.root_dir, self._quote_name(self.name))
|
||||||
# on windows. DNS does not allow '_' in a domain, or for it to end
|
|
||||||
# in a number (I think?)
|
|
||||||
path = os.path.join(self.root_dir, self.name.replace(':', '_'))
|
|
||||||
try:
|
try:
|
||||||
os.makedirs(path, mode=0o700)
|
os.makedirs(path, mode=0o700)
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
@ -89,28 +109,35 @@ class Host(object):
|
|||||||
raise
|
raise
|
||||||
return path
|
return path
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def all(cls):
|
|
||||||
"""Return a generator yielding a host at a time."""
|
|
||||||
for name in sorted(glob.glob1(SESSIONS_DIR, '*')):
|
|
||||||
if os.path.isdir(os.path.join(SESSIONS_DIR, name)):
|
|
||||||
yield Host(name)
|
|
||||||
|
|
||||||
|
|
||||||
class Session(BaseConfigDict):
|
class Session(BaseConfigDict):
|
||||||
""""""
|
|
||||||
|
help = 'https://github.com/jkbr/httpie#sessions'
|
||||||
|
about = 'HTTPie session file'
|
||||||
|
|
||||||
|
VALID_NAME_PATTERN = re.compile('^[a-zA-Z0-9_.-]+$')
|
||||||
|
|
||||||
def __init__(self, host, name, *args, **kwargs):
|
def __init__(self, host, name, *args, **kwargs):
|
||||||
|
assert self.VALID_NAME_PATTERN.match(name)
|
||||||
super(Session, self).__init__(*args, **kwargs)
|
super(Session, self).__init__(*args, **kwargs)
|
||||||
self.host = host
|
self.host = host
|
||||||
self.name = name
|
self.name = name
|
||||||
self['headers'] = {}
|
self['headers'] = {}
|
||||||
self['cookies'] = {}
|
self['cookies'] = {}
|
||||||
|
self['auth'] = {
|
||||||
|
'type': None,
|
||||||
|
'username': None,
|
||||||
|
'password': None
|
||||||
|
}
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def directory(self):
|
def directory(self):
|
||||||
return self.host.path
|
return self.host.path
|
||||||
|
|
||||||
|
@property
|
||||||
|
def verbose_name(self):
|
||||||
|
return '%s %s %s' % (self.host.name, self.name, self.path)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def cookies(self):
|
def cookies(self):
|
||||||
jar = RequestsCookieJar()
|
jar = RequestsCookieJar()
|
||||||
@ -122,26 +149,20 @@ class Session(BaseConfigDict):
|
|||||||
|
|
||||||
@cookies.setter
|
@cookies.setter
|
||||||
def cookies(self, jar):
|
def cookies(self, jar):
|
||||||
excluded = [
|
# http://docs.python.org/2/library/cookielib.html#cookie-objects
|
||||||
'_rest', 'name', 'port_specified',
|
stored_attrs = ['value', 'path', 'secure', 'expires']
|
||||||
'domain_specified', 'domain_initial_dot',
|
|
||||||
'path_specified', 'comment', 'comment_url'
|
|
||||||
]
|
|
||||||
self['cookies'] = {}
|
self['cookies'] = {}
|
||||||
for host in jar._cookies.values():
|
for cookie in jar:
|
||||||
for path in host.values():
|
self['cookies'][cookie.name] = dict(
|
||||||
for name, cookie in path.items():
|
(attname, getattr(cookie, attname))
|
||||||
cookie_dict = {}
|
for attname in stored_attrs
|
||||||
for k, v in cookie.__dict__.items():
|
)
|
||||||
if k not in excluded:
|
|
||||||
cookie_dict[k] = v
|
|
||||||
self['cookies'][name] = cookie_dict
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def auth(self):
|
def auth(self):
|
||||||
auth = self.get('auth', None)
|
auth = self.get('auth', None)
|
||||||
if not auth:
|
if not auth or not auth['type']:
|
||||||
return None
|
return
|
||||||
Auth = {'basic': HTTPBasicAuth,
|
Auth = {'basic': HTTPBasicAuth,
|
||||||
'digest': HTTPDigestAuth}[auth['type']]
|
'digest': HTTPDigestAuth}[auth['type']]
|
||||||
return Auth(auth['username'], auth['password'])
|
return Auth(auth['username'], auth['password'])
|
||||||
@ -154,76 +175,3 @@ class Session(BaseConfigDict):
|
|||||||
'username': cred.username,
|
'username': cred.username,
|
||||||
'password': cred.password,
|
'password': cred.password,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def list_command(args):
|
|
||||||
if args.host:
|
|
||||||
for name, path in Host(args.host):
|
|
||||||
print(name + ' [' + path + ']')
|
|
||||||
else:
|
|
||||||
for host in Host.all():
|
|
||||||
print(host.name)
|
|
||||||
for name, path in host:
|
|
||||||
print(' ' + name + ' [' + path + ']')
|
|
||||||
|
|
||||||
|
|
||||||
def show_command(args):
|
|
||||||
path = Session(Host(args.host), args.name).path
|
|
||||||
if not os.path.exists(path):
|
|
||||||
sys.stderr.write('Session "%s" does not exist [%s].\n'
|
|
||||||
% (args.name, path))
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
with codecs.open(path, encoding='utf8') as f:
|
|
||||||
print(path + ':\n')
|
|
||||||
proc = PygmentsProcessor()
|
|
||||||
print(proc.process_body(f.read(), 'application/json', 'json'))
|
|
||||||
print('')
|
|
||||||
|
|
||||||
|
|
||||||
def delete_command(args):
|
|
||||||
host = Host(args.host)
|
|
||||||
if not args.name:
|
|
||||||
host.delete()
|
|
||||||
else:
|
|
||||||
Session(host, args.name).delete()
|
|
||||||
|
|
||||||
|
|
||||||
def edit_command(args):
|
|
||||||
editor = os.environ.get('EDITOR', None)
|
|
||||||
if not editor:
|
|
||||||
sys.stderr.write(
|
|
||||||
'You need to configure the environment variable EDITOR.\n')
|
|
||||||
sys.exit(1)
|
|
||||||
command = editor.split()
|
|
||||||
command.append(Session(Host(args.host), args.name).path)
|
|
||||||
subprocess.call(command)
|
|
||||||
|
|
||||||
|
|
||||||
def add_commands(subparsers):
|
|
||||||
|
|
||||||
# List
|
|
||||||
list_ = subparsers.add_parser('session-list', help='list sessions')
|
|
||||||
list_.set_defaults(command=list_command)
|
|
||||||
list_.add_argument('host', nargs=OPTIONAL)
|
|
||||||
|
|
||||||
# Show
|
|
||||||
show = subparsers.add_parser('session-show', help='show a session')
|
|
||||||
show.set_defaults(command=show_command)
|
|
||||||
show.add_argument('host')
|
|
||||||
show.add_argument('name')
|
|
||||||
|
|
||||||
# Edit
|
|
||||||
edit = subparsers.add_parser(
|
|
||||||
'session-edit', help='edit a session in $EDITOR')
|
|
||||||
edit.set_defaults(command=edit_command)
|
|
||||||
edit.add_argument('host')
|
|
||||||
edit.add_argument('name')
|
|
||||||
|
|
||||||
# Delete
|
|
||||||
delete = subparsers.add_parser('session-delete', help='delete a session')
|
|
||||||
delete.set_defaults(command=delete_command)
|
|
||||||
delete.add_argument('host')
|
|
||||||
delete.add_argument('name', nargs=OPTIONAL,
|
|
||||||
help='The name of the session to be deleted.'
|
|
||||||
' If not specified, all host sessions are deleted.')
|
|
||||||
|
46
httpie/utils.py
Normal file
46
httpie/utils.py
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
from __future__ import division
|
||||||
|
|
||||||
|
|
||||||
|
def humanize_bytes(n, precision=2):
|
||||||
|
# Author: Doug Latornell
|
||||||
|
# Licence: MIT
|
||||||
|
# URL: http://code.activestate.com/recipes/577081/
|
||||||
|
"""Return a humanized string representation of a number of bytes.
|
||||||
|
|
||||||
|
Assumes `from __future__ import division`.
|
||||||
|
|
||||||
|
>>> humanize_bytes(1)
|
||||||
|
'1 byte'
|
||||||
|
>>> humanize_bytes(1024)
|
||||||
|
'1.0 kB'
|
||||||
|
>>> humanize_bytes(1024 * 123)
|
||||||
|
'123.0 kB'
|
||||||
|
>>> humanize_bytes(1024 * 12342)
|
||||||
|
'12.1 MB'
|
||||||
|
>>> humanize_bytes(1024 * 12342, 2)
|
||||||
|
'12.05 MB'
|
||||||
|
>>> humanize_bytes(1024 * 1234, 2)
|
||||||
|
'1.21 MB'
|
||||||
|
>>> humanize_bytes(1024 * 1234 * 1111, 2)
|
||||||
|
'1.31 GB'
|
||||||
|
>>> humanize_bytes(1024 * 1234 * 1111, 1)
|
||||||
|
'1.3 GB'
|
||||||
|
|
||||||
|
"""
|
||||||
|
abbrevs = [
|
||||||
|
(1 << 50, 'PB'),
|
||||||
|
(1 << 40, 'TB'),
|
||||||
|
(1 << 30, 'GB'),
|
||||||
|
(1 << 20, 'MB'),
|
||||||
|
(1 << 10, 'kB'),
|
||||||
|
(1, 'B')
|
||||||
|
]
|
||||||
|
|
||||||
|
if n == 1:
|
||||||
|
return '1 B'
|
||||||
|
|
||||||
|
for factor, suffix in abbrevs:
|
||||||
|
if n >= factor:
|
||||||
|
break
|
||||||
|
|
||||||
|
return '%.*f %s' % (precision, n / factor, suffix)
|
3
requirements-dev.txt
Normal file
3
requirements-dev.txt
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
tox
|
||||||
|
httpbin
|
||||||
|
docutils
|
4
setup.py
4
setup.py
@ -12,8 +12,7 @@ if sys.argv[-1] == 'test':
|
|||||||
|
|
||||||
|
|
||||||
requirements = [
|
requirements = [
|
||||||
# Debian has only requests==0.10.1 and httpie.deb depends on that.
|
'requests>=1.0.4',
|
||||||
'requests>=0.10.1',
|
|
||||||
'Pygments>=1.5'
|
'Pygments>=1.5'
|
||||||
]
|
]
|
||||||
if sys.version_info[:2] in ((2, 6), (3, 1)):
|
if sys.version_info[:2] in ((2, 6), (3, 1)):
|
||||||
@ -47,7 +46,6 @@ setup(
|
|||||||
entry_points={
|
entry_points={
|
||||||
'console_scripts': [
|
'console_scripts': [
|
||||||
'http = httpie.__main__:main',
|
'http = httpie.__main__:main',
|
||||||
'httpie = httpie.manage:main',
|
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
install_requires=requirements,
|
install_requires=requirements,
|
||||||
|
372
tests/tests.py
372
tests/tests.py
@ -27,15 +27,20 @@ import argparse
|
|||||||
import tempfile
|
import tempfile
|
||||||
import unittest
|
import unittest
|
||||||
import shutil
|
import shutil
|
||||||
|
import time
|
||||||
|
from requests.structures import CaseInsensitiveDict
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from urllib.request import urlopen
|
from urllib.request import urlopen
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
# noinspection PyUnresolvedReferences
|
||||||
from urllib2 import urlopen
|
from urllib2 import urlopen
|
||||||
try:
|
try:
|
||||||
from unittest import skipIf, skip
|
from unittest import skipIf, skip
|
||||||
except ImportError:
|
except ImportError:
|
||||||
skip = lambda msg: lambda self: None
|
skip = lambda msg: lambda self: None
|
||||||
|
|
||||||
|
# noinspection PyUnusedLocal
|
||||||
def skipIf(cond, reason):
|
def skipIf(cond, reason):
|
||||||
def decorator(test_method):
|
def decorator(test_method):
|
||||||
if cond:
|
if cond:
|
||||||
@ -44,7 +49,6 @@ except ImportError:
|
|||||||
return decorator
|
return decorator
|
||||||
|
|
||||||
from requests import __version__ as requests_version
|
from requests import __version__ as requests_version
|
||||||
from requests.compat import is_windows, is_py26, bytes, str
|
|
||||||
|
|
||||||
|
|
||||||
#################################################################
|
#################################################################
|
||||||
@ -55,12 +59,21 @@ from requests.compat import is_windows, is_py26, bytes, str
|
|||||||
TESTS_ROOT = os.path.abspath(os.path.dirname(__file__))
|
TESTS_ROOT = os.path.abspath(os.path.dirname(__file__))
|
||||||
sys.path.insert(0, os.path.realpath(os.path.join(TESTS_ROOT, '..')))
|
sys.path.insert(0, os.path.realpath(os.path.join(TESTS_ROOT, '..')))
|
||||||
|
|
||||||
from httpie import EXIT
|
from httpie import ExitStatus
|
||||||
from httpie import input
|
from httpie import input
|
||||||
from httpie.models import Environment
|
from httpie.models import Environment
|
||||||
from httpie.core import main
|
from httpie.core import main
|
||||||
from httpie.output import BINARY_SUPPRESSED_NOTICE
|
from httpie.output import BINARY_SUPPRESSED_NOTICE
|
||||||
from httpie.input import ParseError
|
from httpie.input import ParseError
|
||||||
|
from httpie.compat import is_windows, is_py26, bytes, str
|
||||||
|
from httpie.downloads import (
|
||||||
|
parse_content_range,
|
||||||
|
filename_from_content_disposition,
|
||||||
|
filename_from_url,
|
||||||
|
get_unique_filename,
|
||||||
|
ContentRangeError,
|
||||||
|
Download,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
CRLF = '\r\n'
|
CRLF = '\r\n'
|
||||||
@ -130,6 +143,7 @@ class TestEnvironment(Environment):
|
|||||||
if self.delete_config_dir:
|
if self.delete_config_dir:
|
||||||
self._shutil.rmtree(self.config_dir)
|
self._shutil.rmtree(self.config_dir)
|
||||||
|
|
||||||
|
|
||||||
def has_docutils():
|
def has_docutils():
|
||||||
try:
|
try:
|
||||||
#noinspection PyUnresolvedReferences
|
#noinspection PyUnresolvedReferences
|
||||||
@ -138,6 +152,7 @@ def has_docutils():
|
|||||||
except ImportError:
|
except ImportError:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
def get_readme_errors():
|
def get_readme_errors():
|
||||||
p = subprocess.Popen([
|
p = subprocess.Popen([
|
||||||
'rst2pseudoxml.py',
|
'rst2pseudoxml.py',
|
||||||
@ -152,6 +167,8 @@ def get_readme_errors():
|
|||||||
|
|
||||||
class BytesResponse(bytes):
|
class BytesResponse(bytes):
|
||||||
stderr = json = exit_status = None
|
stderr = json = exit_status = None
|
||||||
|
|
||||||
|
|
||||||
class StrResponse(str):
|
class StrResponse(str):
|
||||||
stderr = json = exit_status = None
|
stderr = json = exit_status = None
|
||||||
|
|
||||||
@ -175,20 +192,25 @@ def http(*args, **kwargs):
|
|||||||
if not env:
|
if not env:
|
||||||
env = kwargs['env'] = TestEnvironment()
|
env = kwargs['env'] = TestEnvironment()
|
||||||
|
|
||||||
|
stdout = env.stdout
|
||||||
|
stderr = env.stderr
|
||||||
try:
|
try:
|
||||||
|
|
||||||
try:
|
try:
|
||||||
exit_status = main(args=['--traceback'] + list(args), **kwargs)
|
exit_status = main(args=['--traceback'] + list(args), **kwargs)
|
||||||
|
if '--download' in args:
|
||||||
|
# Let the progress reporter thread finish.
|
||||||
|
time.sleep(.5)
|
||||||
except Exception:
|
except Exception:
|
||||||
sys.stderr.write(env.stderr.read())
|
sys.stderr.write(stderr.read())
|
||||||
raise
|
raise
|
||||||
except SystemExit:
|
except SystemExit:
|
||||||
exit_status = EXIT.ERROR
|
exit_status = ExitStatus.ERROR
|
||||||
|
|
||||||
env.stdout.seek(0)
|
stdout.seek(0)
|
||||||
env.stderr.seek(0)
|
stderr.seek(0)
|
||||||
|
|
||||||
output = env.stdout.read()
|
output = stdout.read()
|
||||||
try:
|
try:
|
||||||
r = StrResponse(output.decode('utf8'))
|
r = StrResponse(output.decode('utf8'))
|
||||||
except UnicodeDecodeError:
|
except UnicodeDecodeError:
|
||||||
@ -210,18 +232,20 @@ def http(*args, **kwargs):
|
|||||||
except ValueError:
|
except ValueError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
r.stderr = env.stderr.read()
|
r.stderr = stderr.read()
|
||||||
r.exit_status = exit_status
|
r.exit_status = exit_status
|
||||||
|
|
||||||
return r
|
return r
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
env.stdout.close()
|
stdout.close()
|
||||||
env.stderr.close()
|
stderr.close()
|
||||||
|
|
||||||
|
|
||||||
class BaseTestCase(unittest.TestCase):
|
class BaseTestCase(unittest.TestCase):
|
||||||
|
|
||||||
|
maxDiff = 100000
|
||||||
|
|
||||||
if is_py26:
|
if is_py26:
|
||||||
def assertIn(self, member, container, msg=None):
|
def assertIn(self, member, container, msg=None):
|
||||||
self.assertTrue(member in container, msg)
|
self.assertTrue(member in container, msg)
|
||||||
@ -233,11 +257,15 @@ class BaseTestCase(unittest.TestCase):
|
|||||||
self.assertEqual(set(d1.keys()), set(d2.keys()), msg)
|
self.assertEqual(set(d1.keys()), set(d2.keys()), msg)
|
||||||
self.assertEqual(sorted(d1.values()), sorted(d2.values()), msg)
|
self.assertEqual(sorted(d1.values()), sorted(d2.values()), msg)
|
||||||
|
|
||||||
|
def assertIsNone(self, obj, msg=None):
|
||||||
|
self.assertEqual(obj, None, msg=msg)
|
||||||
|
|
||||||
|
|
||||||
#################################################################
|
#################################################################
|
||||||
# High-level tests using httpbin.
|
# High-level tests using httpbin.
|
||||||
#################################################################
|
#################################################################
|
||||||
|
|
||||||
|
|
||||||
class HTTPieTest(BaseTestCase):
|
class HTTPieTest(BaseTestCase):
|
||||||
|
|
||||||
def test_GET(self):
|
def test_GET(self):
|
||||||
@ -261,7 +289,7 @@ class HTTPieTest(BaseTestCase):
|
|||||||
'foo=bar'
|
'foo=bar'
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertIn('"foo": "bar"', r)
|
self.assertIn(r'\"foo\": \"bar\"', r)
|
||||||
|
|
||||||
def test_POST_JSON_data(self):
|
def test_POST_JSON_data(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -270,7 +298,7 @@ class HTTPieTest(BaseTestCase):
|
|||||||
'foo=bar'
|
'foo=bar'
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertIn('"foo": "bar"', r)
|
self.assertIn(r'\"foo\": \"bar\"', r)
|
||||||
|
|
||||||
def test_POST_form(self):
|
def test_POST_form(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -429,7 +457,9 @@ class AutoContentTypeAndAcceptHeadersTest(BaseTestCase):
|
|||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertEqual(r.json['headers']['Accept'], 'application/json')
|
self.assertEqual(r.json['headers']['Accept'], 'application/json')
|
||||||
self.assertFalse(r.json['headers'].get('Content-Type'))
|
# Make sure Content-Type gets set even with no data.
|
||||||
|
# https://github.com/jkbr/httpie/issues/137
|
||||||
|
self.assertIn('application/json', r.json['headers']['Content-Type'])
|
||||||
|
|
||||||
def test_GET_explicit_JSON_explicit_headers(self):
|
def test_GET_explicit_JSON_explicit_headers(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -512,7 +542,7 @@ class ImplicitHTTPMethodTest(BaseTestCase):
|
|||||||
'hello=world'
|
'hello=world'
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertIn('"hello": "world"', r)
|
self.assertIn(r'\"hello\": \"world\"', r)
|
||||||
|
|
||||||
def test_implicit_POST_form(self):
|
def test_implicit_POST_form(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -651,8 +681,8 @@ class VerboseFlagTest(BaseTestCase):
|
|||||||
'baz=bar'
|
'baz=bar'
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
#noinspection PyUnresolvedReferences
|
self.assertIn('"baz": "bar"', r) # request
|
||||||
self.assertEqual(r.count('"baz": "bar"'), 2)
|
self.assertIn(r'\"baz\": \"bar\"', r) # response
|
||||||
|
|
||||||
|
|
||||||
class MultipartFormDataFileUploadTest(BaseTestCase):
|
class MultipartFormDataFileUploadTest(BaseTestCase):
|
||||||
@ -775,6 +805,8 @@ class RequestBodyFromFilePathTest(BaseTestCase):
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
def test_request_body_from_file_by_path(self):
|
def test_request_body_from_file_by_path(self):
|
||||||
|
# FIXME: *sometimes* fails on py33, the content-type is form.
|
||||||
|
# https://github.com/jkbr/httpie/issues/140
|
||||||
r = http(
|
r = http(
|
||||||
'POST',
|
'POST',
|
||||||
httpbin('/post'),
|
httpbin('/post'),
|
||||||
@ -785,6 +817,8 @@ class RequestBodyFromFilePathTest(BaseTestCase):
|
|||||||
self.assertIn('"Content-Type": "text/plain"', r)
|
self.assertIn('"Content-Type": "text/plain"', r)
|
||||||
|
|
||||||
def test_request_body_from_file_by_path_with_explicit_content_type(self):
|
def test_request_body_from_file_by_path_with_explicit_content_type(self):
|
||||||
|
# FIXME: *sometimes* fails on py33, the content-type is form.
|
||||||
|
# https://github.com/jkbr/httpie/issues/140
|
||||||
r = http(
|
r = http(
|
||||||
'POST',
|
'POST',
|
||||||
httpbin('/post'),
|
httpbin('/post'),
|
||||||
@ -838,8 +872,8 @@ class AuthTest(BaseTestCase):
|
|||||||
httpbin('/digest-auth/auth/user/password')
|
httpbin('/digest-auth/auth/user/password')
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertIn('"authenticated": true', r)
|
self.assertIn(r'"authenticated": true', r)
|
||||||
self.assertIn('"user": "user"', r)
|
self.assertIn(r'"user": "user"', r)
|
||||||
|
|
||||||
def test_password_prompt(self):
|
def test_password_prompt(self):
|
||||||
|
|
||||||
@ -856,6 +890,31 @@ class AuthTest(BaseTestCase):
|
|||||||
self.assertIn('"authenticated": true', r)
|
self.assertIn('"authenticated": true', r)
|
||||||
self.assertIn('"user": "user"', r)
|
self.assertIn('"user": "user"', r)
|
||||||
|
|
||||||
|
def test_credentials_in_url(self):
|
||||||
|
url = httpbin('/basic-auth/user/password')
|
||||||
|
url = 'http://user:password@' + url.split('http://', 1)[1]
|
||||||
|
r = http(
|
||||||
|
'GET',
|
||||||
|
url
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r)
|
||||||
|
self.assertIn('"authenticated": true', r)
|
||||||
|
self.assertIn('"user": "user"', r)
|
||||||
|
|
||||||
|
def test_credentials_in_url_auth_flag_has_priority(self):
|
||||||
|
"""When credentials are passed in URL and via -a at the same time,
|
||||||
|
then the ones from -a are used."""
|
||||||
|
url = httpbin('/basic-auth/user/password')
|
||||||
|
url = 'http://user:wrong_password@' + url.split('http://', 1)[1]
|
||||||
|
r = http(
|
||||||
|
'--auth=user:password',
|
||||||
|
'GET',
|
||||||
|
url
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r)
|
||||||
|
self.assertIn('"authenticated": true', r)
|
||||||
|
self.assertIn('"user": "user"', r)
|
||||||
|
|
||||||
|
|
||||||
class ExitStatusTest(BaseTestCase):
|
class ExitStatusTest(BaseTestCase):
|
||||||
|
|
||||||
@ -865,7 +924,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
httpbin('/status/200')
|
httpbin('/status/200')
|
||||||
)
|
)
|
||||||
self.assertIn(OK, r)
|
self.assertIn(OK, r)
|
||||||
self.assertEqual(r.exit_status, EXIT.OK)
|
self.assertEqual(r.exit_status, ExitStatus.OK)
|
||||||
|
|
||||||
def test_error_response_exits_0_without_check_status(self):
|
def test_error_response_exits_0_without_check_status(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -873,7 +932,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
httpbin('/status/500')
|
httpbin('/status/500')
|
||||||
)
|
)
|
||||||
self.assertIn('HTTP/1.1 500', r)
|
self.assertIn('HTTP/1.1 500', r)
|
||||||
self.assertEqual(r.exit_status, EXIT.OK)
|
self.assertEqual(r.exit_status, ExitStatus.OK)
|
||||||
self.assertTrue(not r.stderr)
|
self.assertTrue(not r.stderr)
|
||||||
|
|
||||||
def test_timeout_exit_status(self):
|
def test_timeout_exit_status(self):
|
||||||
@ -882,7 +941,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
'GET',
|
'GET',
|
||||||
httpbin('/delay/1')
|
httpbin('/delay/1')
|
||||||
)
|
)
|
||||||
self.assertEqual(r.exit_status, EXIT.ERROR_TIMEOUT)
|
self.assertEqual(r.exit_status, ExitStatus.ERROR_TIMEOUT)
|
||||||
|
|
||||||
def test_3xx_check_status_exits_3_and_stderr_when_stdout_redirected(self):
|
def test_3xx_check_status_exits_3_and_stderr_when_stdout_redirected(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -893,7 +952,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
env=TestEnvironment(stdout_isatty=False,)
|
env=TestEnvironment(stdout_isatty=False,)
|
||||||
)
|
)
|
||||||
self.assertIn('HTTP/1.1 301', r)
|
self.assertIn('HTTP/1.1 301', r)
|
||||||
self.assertEqual(r.exit_status, EXIT.ERROR_HTTP_3XX)
|
self.assertEqual(r.exit_status, ExitStatus.ERROR_HTTP_3XX)
|
||||||
self.assertIn('301 moved permanently', r.stderr.lower())
|
self.assertIn('301 moved permanently', r.stderr.lower())
|
||||||
|
|
||||||
@skipIf(requests_version == '0.13.6',
|
@skipIf(requests_version == '0.13.6',
|
||||||
@ -907,7 +966,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
)
|
)
|
||||||
# The redirect will be followed so 200 is expected.
|
# The redirect will be followed so 200 is expected.
|
||||||
self.assertIn('HTTP/1.1 200 OK', r)
|
self.assertIn('HTTP/1.1 200 OK', r)
|
||||||
self.assertEqual(r.exit_status, EXIT.OK)
|
self.assertEqual(r.exit_status, ExitStatus.OK)
|
||||||
|
|
||||||
def test_4xx_check_status_exits_4(self):
|
def test_4xx_check_status_exits_4(self):
|
||||||
r = http(
|
r = http(
|
||||||
@ -916,7 +975,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
httpbin('/status/401')
|
httpbin('/status/401')
|
||||||
)
|
)
|
||||||
self.assertIn('HTTP/1.1 401', r)
|
self.assertIn('HTTP/1.1 401', r)
|
||||||
self.assertEqual(r.exit_status, EXIT.ERROR_HTTP_4XX)
|
self.assertEqual(r.exit_status, ExitStatus.ERROR_HTTP_4XX)
|
||||||
# Also stderr should be empty since stdout isn't redirected.
|
# Also stderr should be empty since stdout isn't redirected.
|
||||||
self.assertTrue(not r.stderr)
|
self.assertTrue(not r.stderr)
|
||||||
|
|
||||||
@ -927,7 +986,7 @@ class ExitStatusTest(BaseTestCase):
|
|||||||
httpbin('/status/500')
|
httpbin('/status/500')
|
||||||
)
|
)
|
||||||
self.assertIn('HTTP/1.1 500', r)
|
self.assertIn('HTTP/1.1 500', r)
|
||||||
self.assertEqual(r.exit_status, EXIT.ERROR_HTTP_5XX)
|
self.assertEqual(r.exit_status, ExitStatus.ERROR_HTTP_5XX)
|
||||||
|
|
||||||
|
|
||||||
class WindowsOnlyTests(BaseTestCase):
|
class WindowsOnlyTests(BaseTestCase):
|
||||||
@ -980,7 +1039,8 @@ class StreamTest(BaseTestCase):
|
|||||||
#self.assertIn(OK_COLOR, r)
|
#self.assertIn(OK_COLOR, r)
|
||||||
|
|
||||||
def test_encoded_stream(self):
|
def test_encoded_stream(self):
|
||||||
"""Test that --stream works with non-prettified redirected terminal output."""
|
"""Test that --stream works with non-prettified
|
||||||
|
redirected terminal output."""
|
||||||
with open(BIN_FILE_PATH, 'rb') as f:
|
with open(BIN_FILE_PATH, 'rb') as f:
|
||||||
r = http(
|
r = http(
|
||||||
'--pretty=none',
|
'--pretty=none',
|
||||||
@ -998,7 +1058,8 @@ class StreamTest(BaseTestCase):
|
|||||||
#self.assertIn(OK, r)
|
#self.assertIn(OK, r)
|
||||||
|
|
||||||
def test_redirected_stream(self):
|
def test_redirected_stream(self):
|
||||||
"""Test that --stream works with non-prettified redirected terminal output."""
|
"""Test that --stream works with non-prettified
|
||||||
|
redirected terminal output."""
|
||||||
with open(BIN_FILE_PATH, 'rb') as f:
|
with open(BIN_FILE_PATH, 'rb') as f:
|
||||||
r = http(
|
r = http(
|
||||||
'--pretty=none',
|
'--pretty=none',
|
||||||
@ -1022,7 +1083,6 @@ class LineEndingsTest(BaseTestCase):
|
|||||||
as the headers/body separator."""
|
as the headers/body separator."""
|
||||||
|
|
||||||
def _validate_crlf(self, msg):
|
def _validate_crlf(self, msg):
|
||||||
#noinspection PyUnresolvedReferences
|
|
||||||
lines = iter(msg.splitlines(True))
|
lines = iter(msg.splitlines(True))
|
||||||
for header in lines:
|
for header in lines:
|
||||||
if header == CRLF:
|
if header == CRLF:
|
||||||
@ -1057,7 +1117,7 @@ class LineEndingsTest(BaseTestCase):
|
|||||||
'GET',
|
'GET',
|
||||||
httpbin('/get')
|
httpbin('/get')
|
||||||
)
|
)
|
||||||
self.assertEqual(r.exit_status,0)
|
self.assertEqual(r.exit_status, 0)
|
||||||
self._validate_crlf(r)
|
self._validate_crlf(r)
|
||||||
|
|
||||||
def test_CRLF_ugly_request(self):
|
def test_CRLF_ugly_request(self):
|
||||||
@ -1162,78 +1222,108 @@ class ArgumentParserTestCase(unittest.TestCase):
|
|||||||
self.parser = input.Parser()
|
self.parser = input.Parser()
|
||||||
|
|
||||||
def test_guess_when_method_set_and_valid(self):
|
def test_guess_when_method_set_and_valid(self):
|
||||||
args = argparse.Namespace()
|
self.parser.args = argparse.Namespace()
|
||||||
args.method = 'GET'
|
self.parser.args.method = 'GET'
|
||||||
args.url = 'http://example.com/'
|
self.parser.args.url = 'http://example.com/'
|
||||||
args.items = []
|
self.parser.args.items = []
|
||||||
|
|
||||||
self.parser._guess_method(args, TestEnvironment())
|
self.parser.env = TestEnvironment()
|
||||||
|
|
||||||
self.assertEqual(args.method, 'GET')
|
self.parser._guess_method()
|
||||||
self.assertEqual(args.url, 'http://example.com/')
|
|
||||||
self.assertEqual(args.items, [])
|
self.assertEqual(self.parser.args.method, 'GET')
|
||||||
|
self.assertEqual(self.parser.args.url, 'http://example.com/')
|
||||||
|
self.assertEqual(self.parser.args.items, [])
|
||||||
|
|
||||||
def test_guess_when_method_not_set(self):
|
def test_guess_when_method_not_set(self):
|
||||||
args = argparse.Namespace()
|
|
||||||
args.method = None
|
|
||||||
args.url = 'http://example.com/'
|
|
||||||
args.items = []
|
|
||||||
|
|
||||||
self.parser._guess_method(args, TestEnvironment())
|
self.parser.args = argparse.Namespace()
|
||||||
|
self.parser.args.method = None
|
||||||
|
self.parser.args.url = 'http://example.com/'
|
||||||
|
self.parser.args.items = []
|
||||||
|
self.parser.env = TestEnvironment()
|
||||||
|
|
||||||
self.assertEqual(args.method, 'GET')
|
self.parser._guess_method()
|
||||||
self.assertEqual(args.url, 'http://example.com/')
|
|
||||||
self.assertEqual(args.items, [])
|
self.assertEqual(self.parser.args.method, 'GET')
|
||||||
|
self.assertEqual(self.parser.args.url, 'http://example.com/')
|
||||||
|
self.assertEqual(self.parser.args.items, [])
|
||||||
|
|
||||||
def test_guess_when_method_set_but_invalid_and_data_field(self):
|
def test_guess_when_method_set_but_invalid_and_data_field(self):
|
||||||
args = argparse.Namespace()
|
self.parser.args = argparse.Namespace()
|
||||||
args.method = 'http://example.com/'
|
self.parser.args.method = 'http://example.com/'
|
||||||
args.url = 'data=field'
|
self.parser.args.url = 'data=field'
|
||||||
args.items = []
|
self.parser.args.items = []
|
||||||
|
self.parser.env = TestEnvironment()
|
||||||
|
self.parser._guess_method()
|
||||||
|
|
||||||
self.parser._guess_method(args, TestEnvironment())
|
self.assertEqual(self.parser.args.method, 'POST')
|
||||||
|
self.assertEqual(self.parser.args.url, 'http://example.com/')
|
||||||
self.assertEqual(args.method, 'POST')
|
|
||||||
self.assertEqual(args.url, 'http://example.com/')
|
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
args.items,
|
self.parser.args.items,
|
||||||
[input.KeyValue(
|
[input.KeyValue(
|
||||||
key='data', value='field', sep='=', orig='data=field')])
|
key='data', value='field', sep='=', orig='data=field')])
|
||||||
|
|
||||||
def test_guess_when_method_set_but_invalid_and_header_field(self):
|
def test_guess_when_method_set_but_invalid_and_header_field(self):
|
||||||
args = argparse.Namespace()
|
self.parser.args = argparse.Namespace()
|
||||||
args.method = 'http://example.com/'
|
self.parser.args.method = 'http://example.com/'
|
||||||
args.url = 'test:header'
|
self.parser.args.url = 'test:header'
|
||||||
args.items = []
|
self.parser.args.items = []
|
||||||
|
|
||||||
self.parser._guess_method(args, TestEnvironment())
|
self.parser.env = TestEnvironment()
|
||||||
|
|
||||||
self.assertEqual(args.method, 'GET')
|
self.parser._guess_method()
|
||||||
self.assertEqual(args.url, 'http://example.com/')
|
|
||||||
|
self.assertEqual(self.parser.args.method, 'GET')
|
||||||
|
self.assertEqual(self.parser.args.url, 'http://example.com/')
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
args.items,
|
self.parser.args.items,
|
||||||
[input.KeyValue(
|
[input.KeyValue(
|
||||||
key='test', value='header', sep=':', orig='test:header')])
|
key='test', value='header', sep=':', orig='test:header')])
|
||||||
|
|
||||||
def test_guess_when_method_set_but_invalid_and_item_exists(self):
|
def test_guess_when_method_set_but_invalid_and_item_exists(self):
|
||||||
args = argparse.Namespace()
|
self.parser.args = argparse.Namespace()
|
||||||
args.method = 'http://example.com/'
|
self.parser.args.method = 'http://example.com/'
|
||||||
args.url = 'new_item=a'
|
self.parser.args.url = 'new_item=a'
|
||||||
args.items = [
|
self.parser.args.items = [
|
||||||
input.KeyValue(
|
input.KeyValue(
|
||||||
key='old_item', value='b', sep='=', orig='old_item=b')
|
key='old_item', value='b', sep='=', orig='old_item=b')
|
||||||
]
|
]
|
||||||
|
|
||||||
self.parser._guess_method(args, TestEnvironment())
|
self.parser.env = TestEnvironment()
|
||||||
|
|
||||||
self.assertEqual(args.items, [
|
self.parser._guess_method()
|
||||||
|
|
||||||
|
self.assertEqual(self.parser.args.items, [
|
||||||
input.KeyValue(
|
input.KeyValue(
|
||||||
key='new_item', value='a', sep='=', orig='new_item=a'),
|
key='new_item', value='a', sep='=', orig='new_item=a'),
|
||||||
input.KeyValue(key
|
input.KeyValue(
|
||||||
='old_item', value='b', sep='=', orig='old_item=b'),
|
key='old_item', value='b', sep='=', orig='old_item=b'),
|
||||||
])
|
])
|
||||||
|
|
||||||
|
|
||||||
|
class TestNoOptions(BaseTestCase):
|
||||||
|
|
||||||
|
def test_valid_no_options(self):
|
||||||
|
r = http(
|
||||||
|
'--verbose',
|
||||||
|
'--no-verbose',
|
||||||
|
'GET',
|
||||||
|
httpbin('/get')
|
||||||
|
)
|
||||||
|
self.assertNotIn('GET /get HTTP/1.1', r)
|
||||||
|
|
||||||
|
def test_invalid_no_options(self):
|
||||||
|
r = http(
|
||||||
|
'--no-war',
|
||||||
|
'GET',
|
||||||
|
httpbin('/get')
|
||||||
|
)
|
||||||
|
self.assertEqual(r.exit_status, 1)
|
||||||
|
self.assertIn('unrecognized arguments: --no-war', r.stderr)
|
||||||
|
self.assertNotIn('GET /get HTTP/1.1', r)
|
||||||
|
|
||||||
|
|
||||||
class READMETest(BaseTestCase):
|
class READMETest(BaseTestCase):
|
||||||
|
|
||||||
@skipIf(not has_docutils(), 'docutils not installed')
|
@skipIf(not has_docutils(), 'docutils not installed')
|
||||||
@ -1267,7 +1357,7 @@ class SessionTest(BaseTestCase):
|
|||||||
shutil.rmtree(self.config_dir)
|
shutil.rmtree(self.config_dir)
|
||||||
|
|
||||||
def test_session_create(self):
|
def test_session_create(self):
|
||||||
# Verify that the has been created
|
# Verify that the session has been created.
|
||||||
r = http(
|
r = http(
|
||||||
'--session=test',
|
'--session=test',
|
||||||
'GET',
|
'GET',
|
||||||
@ -1316,7 +1406,7 @@ class SessionTest(BaseTestCase):
|
|||||||
self.assertNotEqual(r1.json['headers']['Authorization'],
|
self.assertNotEqual(r1.json['headers']['Authorization'],
|
||||||
r3.json['headers']['Authorization'])
|
r3.json['headers']['Authorization'])
|
||||||
|
|
||||||
def test_session_only(self):
|
def test_session_read_only(self):
|
||||||
# Get a response from the original session.
|
# Get a response from the original session.
|
||||||
r1 = http(
|
r1 = http(
|
||||||
'--session=test',
|
'--session=test',
|
||||||
@ -1348,10 +1438,146 @@ class SessionTest(BaseTestCase):
|
|||||||
)
|
)
|
||||||
self.assertIn(OK, r3)
|
self.assertIn(OK, r3)
|
||||||
|
|
||||||
|
# Origin can differ on Travis.
|
||||||
|
del r1.json['origin'], r3.json['origin']
|
||||||
|
|
||||||
# Should be the same as before r2.
|
# Should be the same as before r2.
|
||||||
self.assertDictEqual(r1.json, r3.json)
|
self.assertDictEqual(r1.json, r3.json)
|
||||||
|
|
||||||
|
|
||||||
|
class DownloadUtilsTest(BaseTestCase):
|
||||||
|
|
||||||
|
def test_Content_Range_parsing(self):
|
||||||
|
|
||||||
|
parse = parse_content_range
|
||||||
|
|
||||||
|
self.assertEqual(parse('bytes 100-199/200', 100), 200)
|
||||||
|
self.assertEqual(parse('bytes 100-199/*', 100), 200)
|
||||||
|
|
||||||
|
# missing
|
||||||
|
self.assertRaises(ContentRangeError, parse, None, 100)
|
||||||
|
|
||||||
|
# syntax error
|
||||||
|
self.assertRaises(ContentRangeError, parse, 'beers 100-199/*', 100)
|
||||||
|
|
||||||
|
# unexpected range
|
||||||
|
self.assertRaises(ContentRangeError, parse, 'bytes 100-199/*', 99)
|
||||||
|
|
||||||
|
# invalid instance-length
|
||||||
|
self.assertRaises(ContentRangeError, parse, 'bytes 100-199/199', 100)
|
||||||
|
|
||||||
|
# invalid byte-range-resp-spec
|
||||||
|
self.assertRaises(ContentRangeError, parse, 'bytes 100-99/199', 100)
|
||||||
|
|
||||||
|
# invalid byte-range-resp-spec
|
||||||
|
self.assertRaises(ContentRangeError, parse, 'bytes 100-100/*', 100)
|
||||||
|
|
||||||
|
def test_Content_Disposition_parsing(self):
|
||||||
|
parse = filename_from_content_disposition
|
||||||
|
self.assertEqual(
|
||||||
|
parse('attachment; filename=hello-WORLD_123.txt'),
|
||||||
|
'hello-WORLD_123.txt'
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
parse('attachment; filename=".hello-WORLD_123.txt"'),
|
||||||
|
'hello-WORLD_123.txt'
|
||||||
|
)
|
||||||
|
self.assertIsNone(parse('attachment; filename='))
|
||||||
|
self.assertIsNone(parse('attachment; filename=/etc/hosts'))
|
||||||
|
self.assertIsNone(parse('attachment; filename=hello@world'))
|
||||||
|
|
||||||
|
def test_filename_from_url(self):
|
||||||
|
self.assertEqual(filename_from_url(
|
||||||
|
url='http://example.org/foo',
|
||||||
|
content_type='text/plain'
|
||||||
|
), 'foo.txt')
|
||||||
|
self.assertEqual(filename_from_url(
|
||||||
|
url='http://example.org/foo',
|
||||||
|
content_type='text/html; charset=utf8'
|
||||||
|
), 'foo.html')
|
||||||
|
self.assertEqual(filename_from_url(
|
||||||
|
url='http://example.org/foo',
|
||||||
|
content_type=None
|
||||||
|
), 'foo')
|
||||||
|
self.assertEqual(filename_from_url(
|
||||||
|
url='http://example.org/foo',
|
||||||
|
content_type='x-foo/bar'
|
||||||
|
), 'foo')
|
||||||
|
|
||||||
|
def test_unique_filename(self):
|
||||||
|
|
||||||
|
def make_exists(unique_on_attempt=0):
|
||||||
|
# noinspection PyUnresolvedReferences,PyUnusedLocal
|
||||||
|
def exists(filename):
|
||||||
|
if exists.attempt == unique_on_attempt:
|
||||||
|
return False
|
||||||
|
exists.attempt += 1
|
||||||
|
return True
|
||||||
|
exists.attempt = 0
|
||||||
|
return exists
|
||||||
|
|
||||||
|
self.assertEqual(
|
||||||
|
get_unique_filename('foo.bar', exists=make_exists()),
|
||||||
|
'foo.bar'
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
get_unique_filename('foo.bar', exists=make_exists(1)),
|
||||||
|
'foo.bar-1'
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
get_unique_filename('foo.bar', exists=make_exists(10)),
|
||||||
|
'foo.bar-10'
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class Response(object):
|
||||||
|
|
||||||
|
# noinspection PyDefaultArgument
|
||||||
|
def __init__(self, url, headers={}, status_code=200):
|
||||||
|
self.url = url
|
||||||
|
self.headers = CaseInsensitiveDict(headers)
|
||||||
|
self.status_code = status_code
|
||||||
|
|
||||||
|
|
||||||
|
# noinspection PyTypeChecker
|
||||||
|
class DownloadTest(BaseTestCase):
|
||||||
|
# TODO: more tests
|
||||||
|
|
||||||
|
def test_actual_download(self):
|
||||||
|
url = httpbin('/robots.txt')
|
||||||
|
body = urlopen(url).read().decode()
|
||||||
|
r = http(
|
||||||
|
'--download',
|
||||||
|
url,
|
||||||
|
env=TestEnvironment(
|
||||||
|
stdin_isatty=True,
|
||||||
|
stdout_isatty=False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.assertIn('Downloading', r.stderr)
|
||||||
|
self.assertIn('[K', r.stderr)
|
||||||
|
self.assertIn('Done', r.stderr)
|
||||||
|
self.assertEqual(body, r)
|
||||||
|
|
||||||
|
def test_download_no_Content_Length(self):
|
||||||
|
download = Download(output_file=open(os.devnull, 'w'))
|
||||||
|
download.start(Response(url=httpbin('/')))
|
||||||
|
download._chunk_downloaded(b'12345')
|
||||||
|
download.finish()
|
||||||
|
self.assertFalse(download.interrupted)
|
||||||
|
|
||||||
|
def test_download_interrupted(self):
|
||||||
|
download = Download(
|
||||||
|
output_file=open(os.devnull, 'w')
|
||||||
|
)
|
||||||
|
download.start(Response(
|
||||||
|
url=httpbin('/'),
|
||||||
|
headers={'Content-Length': 5}
|
||||||
|
))
|
||||||
|
download._chunk_downloaded(b'1234')
|
||||||
|
download.finish()
|
||||||
|
self.assertTrue(download.interrupted)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
#noinspection PyCallingNonCallable
|
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
8
tox.ini
8
tox.ini
@ -4,16 +4,10 @@
|
|||||||
# and then run "tox" from this directory.
|
# and then run "tox" from this directory.
|
||||||
|
|
||||||
[tox]
|
[tox]
|
||||||
envlist = py26, py27, py32, pypy
|
envlist = py26, py27, py33, pypy
|
||||||
|
|
||||||
[testenv]
|
[testenv]
|
||||||
commands = {envpython} setup.py test
|
commands = {envpython} setup.py test
|
||||||
|
|
||||||
[testenv:py26]
|
[testenv:py26]
|
||||||
deps = argparse
|
deps = argparse
|
||||||
|
|
||||||
[testenv:py30]
|
|
||||||
deps = argparse
|
|
||||||
|
|
||||||
[testenv:py31]
|
|
||||||
deps = argparse
|
|
||||||
|
Reference in New Issue
Block a user