mirror of
https://github.com/httpie/cli.git
synced 2025-08-12 13:47:51 +02:00
Compare commits
128 Commits
Author | SHA1 | Date | |
---|---|---|---|
9d2ac5d8ad | |||
3e4e1c72a4 | |||
29f6b6a2a9 | |||
26b2d408e7 | |||
b5f180a5ee | |||
354aaa94bd | |||
2ad4059f92 | |||
5a6b65ecc6 | |||
2acb303552 | |||
f7b703b4bf | |||
00de49f4c3 | |||
67496162fa | |||
8378ad3624 | |||
f87884dd8d | |||
b671ee35e7 | |||
69247066dc | |||
383dba524a | |||
60f09776a5 | |||
48719aa70e | |||
809a461a26 | |||
c3d550e930 | |||
172df162b3 | |||
1bad62ab0e | |||
8d302f91f9 | |||
63b61bc811 | |||
5af88756a6 | |||
7f624e61b5 | |||
6e848b3203 | |||
8e112a6948 | |||
87c59ae561 | |||
76eebeac2a | |||
5b9cbcb530 | |||
8ad33d5f6a | |||
86ac4cdb7b | |||
e09b74021c | |||
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 | |||
2e57c080fd | |||
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>`_
|
||||||
@ -27,3 +28,4 @@ Patches and ideas
|
|||||||
* `Tomek Wójcik <https://github.com/tomekwojcik>`_
|
* `Tomek Wójcik <https://github.com/tomekwojcik>`_
|
||||||
* `Davey Shafik <https://github.com/dshafik>`_
|
* `Davey Shafik <https://github.com/dshafik>`_
|
||||||
* `cido <https://github.com/cido>`_
|
* `cido <https://github.com/cido>`_
|
||||||
|
* `Justin Bonnar <https://github.com/jargonjustin>`_
|
||||||
|
355
README.rst
355
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,34 @@ 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
|
||||||
|
|
||||||
|
|
||||||
|
Set a custom ``Host`` header to work around missing DNS records:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http localhost:8000 Host:example.com
|
||||||
|
|
||||||
|
..
|
||||||
|
|
||||||
--------
|
--------
|
||||||
|
|
||||||
*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 +266,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 +275,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 +306,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 +324,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 +345,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 +369,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 +389,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
|
||||||
@ -391,7 +425,7 @@ To set custom headers you can use the ``Header:Value`` notation:
|
|||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
$ http example.org User-Agent:Bacon/1.0 Cookie:valued-visitor=yes X-Foo:Bar Referer:http://httpie.org/
|
$ http example.org User-Agent:Bacon/1.0 'Cookie:valued-visitor=yes;foo=bar' X-Foo:Bar Referer:http://httpie.org/
|
||||||
|
|
||||||
|
|
||||||
.. code-block:: http
|
.. code-block:: http
|
||||||
@ -399,7 +433,7 @@ To set custom headers you can use the ``Header:Value`` notation:
|
|||||||
GET / HTTP/1.1
|
GET / HTTP/1.1
|
||||||
Accept: */*
|
Accept: */*
|
||||||
Accept-Encoding: identity, deflate, compress, gzip
|
Accept-Encoding: identity, deflate, compress, gzip
|
||||||
Cookie: valued-visitor=yes
|
Cookie: valued-visitor=yes;foo=bar
|
||||||
Host: example.org
|
Host: example.org
|
||||||
Referer: http://httpie.org/
|
Referer: http://httpie.org/
|
||||||
User-Agent: Bacon/1.0
|
User-Agent: Bacon/1.0
|
||||||
@ -424,8 +458,8 @@ Any of the default headers can be overwritten.
|
|||||||
Authentication
|
Authentication
|
||||||
==============
|
==============
|
||||||
|
|
||||||
The currently supported authentication schemes are Basic and Digest (more to
|
The currently supported authentication schemes are Basic and Digest
|
||||||
come). There are two flags that control authentication:
|
(see `auth plugins`_ for more). There are two flags that control authentication:
|
||||||
|
|
||||||
=================== ======================================================
|
=================== ======================================================
|
||||||
``--auth, -a`` Pass a ``username:password`` pair as
|
``--auth, -a`` Pass a ``username:password`` pair as
|
||||||
@ -433,13 +467,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 +501,45 @@ 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
|
||||||
|
[...]
|
||||||
|
|
||||||
|
|
||||||
|
------------
|
||||||
|
Auth Plugins
|
||||||
|
------------
|
||||||
|
|
||||||
|
* `httpie-ntlm <https://github.com/jkbr/httpie-ntlm>`_
|
||||||
|
* `httpie-oauth <https://github.com/jkbr/httpie-oauth>`_
|
||||||
|
|
||||||
|
|
||||||
=======
|
=======
|
||||||
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 +611,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:
|
||||||
|
|
||||||
@ -658,7 +718,16 @@ On OS X, you can send the contents of the clipboard with ``pbpaste``:
|
|||||||
|
|
||||||
|
|
||||||
Passing data through ``stdin`` cannot be combined with data fields specified
|
Passing data through ``stdin`` cannot be combined with data fields specified
|
||||||
on the command line.
|
on the command line:
|
||||||
|
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ echo 'data' | http POST example.org more=data # This is invalid
|
||||||
|
|
||||||
|
|
||||||
|
To prevent HTTPie from reading ``stdin`` data you can use the
|
||||||
|
``--ignore-stdin`` option.
|
||||||
|
|
||||||
|
|
||||||
-------------------------
|
-------------------------
|
||||||
@ -700,6 +769,7 @@ Also, the following formatting is applied:
|
|||||||
* HTTP headers are sorted by name.
|
* HTTP headers are sorted by name.
|
||||||
* JSON data is indented, sorted by keys, and unicode escapes are converted
|
* JSON data is indented, sorted by keys, and unicode escapes are converted
|
||||||
to the characters they represent.
|
to the characters they represent.
|
||||||
|
* XML data is indented for better readability.
|
||||||
|
|
||||||
One of these options can be used to control output processing:
|
One of these options can be used to control output processing:
|
||||||
|
|
||||||
@ -723,7 +793,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 +851,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 --print=hb "$@" | 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 +959,80 @@ 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 any previous ones.
|
||||||
and cookies (manually specified or sent by the server) persist between requests:
|
HTTPie also supports persistent sessions, where custom headers (except for the
|
||||||
|
ones starting with ``Content-`` or ``If-``), authorization, and cookies
|
||||||
|
(manually specified or sent by the server) persist between requests
|
||||||
|
to the same host.
|
||||||
|
|
||||||
|
--------------
|
||||||
|
Named Sessions
|
||||||
|
--------------
|
||||||
|
|
||||||
|
Create a new session named ``user1`` for ``example.org``:
|
||||||
|
|
||||||
.. 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
|
Named sessions' data is 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).
|
||||||
|
|
||||||
See also `config`_.
|
------------------
|
||||||
|
Anonymous Sessions
|
||||||
|
------------------
|
||||||
|
|
||||||
|
Instead of a name, you can also directly specify a path to a session file. This
|
||||||
|
allows for sessions to be re-used across multiple hosts:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
$ http --session=/tmp/session.json example.org
|
||||||
|
$ http --session=/tmp/session.json admin.example.org
|
||||||
|
$ http --session=~/.httpie/sessions/another.example.org/test.json example.org
|
||||||
|
$ http --session-read-only=/tmp/session.json example.org
|
||||||
|
|
||||||
|
|
||||||
|
**Warning:** All session data, including credentials, cookie data,
|
||||||
|
and custom headers are stored in plain text.
|
||||||
|
|
||||||
|
Note that session files can also be created and edited manually in a text
|
||||||
|
editor; they are plain JSON.
|
||||||
|
|
||||||
|
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 +1042,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.
|
||||||
@ -893,14 +1073,18 @@ When using HTTPie from **shell scripts**, it can be handy to set the
|
|||||||
``--check-status`` flag. It instructs HTTPie to exit with an error if the
|
``--check-status`` flag. It instructs HTTPie to exit with an error if the
|
||||||
HTTP status is one of ``3xx``, ``4xx``, or ``5xx``. The exit status will
|
HTTP status is one of ``3xx``, ``4xx``, or ``5xx``. The exit status will
|
||||||
be ``3`` (unless ``--follow`` is set), ``4``, or ``5``,
|
be ``3`` (unless ``--follow`` is set), ``4``, or ``5``,
|
||||||
respectively. Also, the ``--timeout`` option allows to overwrite the default
|
respectively.
|
||||||
30s timeout:
|
|
||||||
|
The ``--ignore-stdin`` option prevents HTTPie from reading data from ``stdin``,
|
||||||
|
which is usually not desirable during non-interactive invocations.
|
||||||
|
|
||||||
|
Also, the ``--timeout`` option allows to overwrite the default 30s timeout:
|
||||||
|
|
||||||
.. code-block:: bash
|
.. code-block:: bash
|
||||||
|
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
if http --timeout=2.5 --check-status HEAD example.org/health &> /dev/null; then
|
if http --check-status --ignore-stdin --timeout=2.5 HEAD example.org/health &> /dev/null; then
|
||||||
echo 'OK!'
|
echo 'OK!'
|
||||||
else
|
else
|
||||||
case $? in
|
case $? in
|
||||||
@ -1003,6 +1187,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 +1200,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 +1213,33 @@ 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.8.0-dev`_
|
||||||
|
* `0.7.0`_ (2013-09-24)
|
||||||
|
* Added ``--ignore-stdin``.
|
||||||
|
* Added support for auth plugins.
|
||||||
|
* Improved ``Content-Disposition`` parsing for ``--download`` mode.
|
||||||
|
* `0.6.0`_ (2013-06-03)
|
||||||
|
* XML data is now formatted.
|
||||||
|
* ``--session`` and ``--session-read-only`` now also accept paths to
|
||||||
|
session files (eg. ``http --session=/tmp/session.json example.org``).
|
||||||
|
* `0.5.1`_ (2013-05-13)
|
||||||
|
* ``Content-*`` and ``If-*`` request headers are not stored in sessions
|
||||||
|
anymore as they are request-specific.
|
||||||
|
* `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 +1252,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 +1322,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 +1330,13 @@ 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.5.1: https://github.com/jkbr/httpie/compare/0.5.0...0.5.1
|
||||||
|
.. _0.6.0: https://github.com/jkbr/httpie/compare/0.5.1...0.6.0
|
||||||
|
.. _0.7.0: https://github.com/jkbr/httpie/compare/0.6.0...0.7.0
|
||||||
|
.. _0.8.0-dev: https://github.com/jkbr/httpie/compare/0.7.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.7.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
|
||||||
|
620
httpie/cli.py
620
httpie/cli.py
@ -3,207 +3,279 @@
|
|||||||
NOTE: the CLI interface may change before reaching v1.0.
|
NOTE: the CLI interface may change before reaching v1.0.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from argparse import FileType, OPTIONAL, ZERO_OR_MORE, SUPPRESS
|
from textwrap import dedent, wrap
|
||||||
|
#noinspection PyCompatibility
|
||||||
from requests.compat import is_windows
|
from argparse import (RawDescriptionHelpFormatter, FileType,
|
||||||
|
OPTIONAL, ZERO_OR_MORE, SUPPRESS)
|
||||||
|
|
||||||
from . import __doc__
|
from . import __doc__
|
||||||
from . import __version__
|
from . import __version__
|
||||||
from .config import DEFAULT_CONFIG_DIR
|
from .plugins.builtin import BuiltinAuthPlugin
|
||||||
|
from .plugins import plugin_manager
|
||||||
|
from .sessions import DEFAULT_SESSIONS_DIR
|
||||||
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, OUTPUT_OPTIONS_DEFAULT,
|
||||||
PRETTY_MAP, PRETTY_STDOUT_TTY_ONLY)
|
PRETTY_MAP, PRETTY_STDOUT_TTY_ONLY, SessionNameValidator)
|
||||||
|
|
||||||
|
|
||||||
def _(text):
|
class HTTPieHelpFormatter(RawDescriptionHelpFormatter):
|
||||||
"""Normalize whitespace."""
|
"""A nicer help formatter.
|
||||||
return ' '.join(text.strip().split())
|
|
||||||
|
|
||||||
|
Help for arguments can be indented and contain new lines.
|
||||||
|
It will be de-dented and arguments in the help
|
||||||
|
will be separated by a blank line for better readability.
|
||||||
|
|
||||||
|
|
||||||
|
"""
|
||||||
|
def __init__(self, max_help_position=6, *args, **kwargs):
|
||||||
|
# A smaller indent for args help.
|
||||||
|
kwargs['max_help_position'] = max_help_position
|
||||||
|
super(HTTPieHelpFormatter, self).__init__(*args, **kwargs)
|
||||||
|
|
||||||
|
def _split_lines(self, text, width):
|
||||||
|
text = dedent(text).strip() + '\n\n'
|
||||||
|
return text.splitlines()
|
||||||
|
|
||||||
parser = Parser(
|
parser = Parser(
|
||||||
|
formatter_class=HTTPieHelpFormatter,
|
||||||
description='%s <http://httpie.org>' % __doc__.strip(),
|
description='%s <http://httpie.org>' % __doc__.strip(),
|
||||||
epilog=_('''
|
epilog=dedent("""
|
||||||
Suggestions and bug reports are greatly appreciated:
|
For every --OPTION there is also a --no-OPTION that reverts OPTION
|
||||||
|
to its default value.
|
||||||
|
|
||||||
|
Suggestions and bug reports are greatly appreciated:
|
||||||
|
|
||||||
https://github.com/jkbr/httpie/issues
|
https://github.com/jkbr/httpie/issues
|
||||||
''')
|
|
||||||
|
""")
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
###############################################################################
|
|
||||||
# Positional arguments.
|
# Positional arguments.
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
positional = parser.add_argument_group(
|
positional = parser.add_argument_group(
|
||||||
title='Positional arguments',
|
title='Positional arguments',
|
||||||
description=_('''
|
description=dedent("""
|
||||||
These arguments come after any flags and in the
|
These arguments come after any flags and in the order they are listed here.
|
||||||
order they are listed here. Only URL is required.'''
|
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="""
|
||||||
The HTTP method to be used for the request
|
The HTTP method to be used for the request (GET, POST, PUT, DELETE, ...).
|
||||||
(GET, POST, PUT, DELETE, PATCH, ...).
|
|
||||||
If this argument is omitted, then HTTPie
|
This argument can be omitted in which case HTTPie will use POST if there
|
||||||
will guess the HTTP method. If there is some
|
is some data to be sent, otherwise GET:
|
||||||
data to be sent, then it will be POST, otherwise GET.
|
|
||||||
''')
|
$ http example.org # => GET
|
||||||
|
$ http example.org hello=world # => POST
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
positional.add_argument(
|
positional.add_argument(
|
||||||
'url', metavar='URL',
|
'url',
|
||||||
help=_('''
|
metavar='URL',
|
||||||
The protocol defaults to http:// if the
|
help="""
|
||||||
URL does not include one.
|
The scheme defaults to 'http://' if the 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=r"""
|
||||||
A key-value pair whose type is defined by the
|
Optional key-value pairs to be included in the request. The separator used
|
||||||
separator used. It can be an HTTP header (header:value),
|
determines the type:
|
||||||
a data field to be used in the request body (field_name=value),
|
|
||||||
a raw JSON data field (field_name:=value),
|
':' HTTP headers:
|
||||||
a query parameter (name==value),
|
|
||||||
or a file field (field_name@/path/to/file).
|
Referer:http://httpie.org Cookie:foo=bar User-Agent:bacon/1.0
|
||||||
You can use a backslash to escape a colliding
|
|
||||||
separator in the field name.
|
'==' URL parameters to be appended to the request URI:
|
||||||
''')
|
|
||||||
|
search==httpie
|
||||||
|
|
||||||
|
'=' Data fields to be serialized into a JSON object (with --json, -j)
|
||||||
|
or form data (with --form, -f):
|
||||||
|
|
||||||
|
name=HTTPie language=Python description='CLI HTTP client'
|
||||||
|
|
||||||
|
'@' Form file fields (only with --form, -f):
|
||||||
|
|
||||||
|
cs@~/Documents/CV.pdf
|
||||||
|
|
||||||
|
':=' Non-string JSON data fields (only with --json, -j):
|
||||||
|
|
||||||
|
awesome:=true amount:=42 colors:='["red", "green", "blue"]'
|
||||||
|
|
||||||
|
You can use a backslash to escape a colliding separator in the field name:
|
||||||
|
|
||||||
|
field-name-with\:colon=value
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
#######################################################################
|
||||||
# Content type.
|
# Content type.
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
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',
|
||||||
help=_('''
|
action='store_true',
|
||||||
(default) Data items from the command
|
help="""
|
||||||
line are serialized as a JSON object.
|
(default) Data items from the command line are serialized as a JSON object.
|
||||||
The Content-Type and Accept headers
|
The Content-Type and Accept headers are set to application/json
|
||||||
are set to application/json (if not specified).
|
(if not specified).
|
||||||
''')
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
content_type.add_argument(
|
content_type.add_argument(
|
||||||
'--form', '-f', action='store_true',
|
'--form', '-f',
|
||||||
help=_('''
|
action='store_true',
|
||||||
Data items from the command line are serialized as form fields.
|
help="""
|
||||||
The Content-Type is set to application/x-www-form-urlencoded
|
Data items from the command line are serialized as form fields.
|
||||||
(if not specified).
|
|
||||||
The presence of any file fields results
|
The Content-Type is set to application/x-www-form-urlencoded (if not
|
||||||
in a multipart/form-data request.
|
specified). The presence of any file fields results in a
|
||||||
''')
|
multipart/form-data request.
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
#######################################################################
|
||||||
# Output processing
|
# Output processing
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
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,
|
||||||
'''
|
choices=sorted(PRETTY_MAP.keys()),
|
||||||
Save output to FILE.
|
help="""
|
||||||
This option is a replacement for piping output to FILE,
|
Controls output processing. The value can be "none" to not prettify
|
||||||
which would on Windows result in corrupted data
|
the output (default for redirected output), "all" to apply both colors
|
||||||
being saved.
|
and formatting (default for terminal output), "colors", or "format".
|
||||||
|
|
||||||
'''
|
"""
|
||||||
|
)
|
||||||
|
output_processing.add_argument(
|
||||||
|
'--style', '-s',
|
||||||
|
dest='style',
|
||||||
|
metavar='STYLE',
|
||||||
|
default=DEFAULT_STYLE,
|
||||||
|
choices=AVAILABLE_STYLES,
|
||||||
|
help="""
|
||||||
|
Output coloring style (default is "{default}"). On of:
|
||||||
|
|
||||||
|
{available}
|
||||||
|
|
||||||
|
For this option to work properly, please make sure that the $TERM
|
||||||
|
environment variable is set to "xterm-256color" or similar
|
||||||
|
(e.g., via `export TERM=xterm-256color' in your ~/.bashrc).
|
||||||
|
|
||||||
|
"""
|
||||||
|
.format(
|
||||||
|
default=DEFAULT_STYLE,
|
||||||
|
available='\n'.join(
|
||||||
|
'{0: >20}'.format(line.strip())
|
||||||
|
for line in
|
||||||
|
wrap(' '.join(sorted(AVAILABLE_STYLES)), 60)
|
||||||
|
),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
output_processing.add_argument(
|
|
||||||
'--pretty', dest='prettify', default=PRETTY_STDOUT_TTY_ONLY,
|
|
||||||
choices=sorted(PRETTY_MAP.keys()),
|
|
||||||
help=_('''
|
|
||||||
Controls output processing. The value can be "none" to not prettify
|
|
||||||
the output (default for redirected output), "all" to apply both colors
|
|
||||||
and formatting
|
|
||||||
(default for terminal output), "colors", or "format".
|
|
||||||
''')
|
|
||||||
)
|
|
||||||
output_processing.add_argument(
|
|
||||||
'--style', '-s', dest='style', default=DEFAULT_STYLE, metavar='STYLE',
|
|
||||||
choices=AVAILABLE_STYLES,
|
|
||||||
help=_('''
|
|
||||||
Output coloring style. One of %s. Defaults to "%s".
|
|
||||||
For this option to work properly, please make sure that the
|
|
||||||
$TERM environment variable is set to "xterm-256color" or similar
|
|
||||||
(e.g., via `export TERM=xterm-256color' in your ~/.bashrc).
|
|
||||||
''') % (', '.join(sorted(AVAILABLE_STYLES)), DEFAULT_STYLE)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
###############################################################################
|
|
||||||
# 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:
|
||||||
"{request_headers}" stands for the request headers, and
|
|
||||||
"{request_body}" for the request body.
|
'{req_head}' request headers
|
||||||
"{response_headers}" stands for the response headers and
|
'{req_body}' request body
|
||||||
"{response_body}" for response the body.
|
'{res_head}' response headers
|
||||||
The default behaviour is "hb" (i.e., the response
|
'{res_body}' response body
|
||||||
headers and body is printed), if standard output is not redirected.
|
|
||||||
If the output is piped to another program or to a file,
|
The default behaviour is '{default}' (i.e., the response headers and body
|
||||||
then only the body is printed by default.
|
is printed), if standard output is not redirected. If the output is piped
|
||||||
'''.format(
|
to another program or to a file, then only the response body is printed
|
||||||
request_headers=OUT_REQ_HEAD,
|
by default.
|
||||||
request_body=OUT_REQ_BODY,
|
|
||||||
response_headers=OUT_RESP_HEAD,
|
"""
|
||||||
response_body=OUT_RESP_BODY,
|
.format(
|
||||||
))
|
req_head=OUT_REQ_HEAD,
|
||||||
|
req_body=OUT_REQ_BODY,
|
||||||
|
res_head=OUT_RESP_HEAD,
|
||||||
|
res_body=OUT_RESP_BODY,
|
||||||
|
default=OUTPUT_OPTIONS_DEFAULT,
|
||||||
|
)
|
||||||
)
|
)
|
||||||
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',
|
||||||
help=_('''
|
action='store_const',
|
||||||
Print the whole request as well as the response.
|
const=''.join(OUTPUT_OPTIONS),
|
||||||
Shortcut for --print={0}.
|
help="""
|
||||||
'''.format(''.join(OUTPUT_OPTIONS)))
|
Print the whole request as well as the response. Shortcut for --print={0}.
|
||||||
|
|
||||||
|
"""
|
||||||
|
.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',
|
||||||
help=_('''
|
action='store_const',
|
||||||
Print only the response headers.
|
const=OUT_RESP_HEAD,
|
||||||
Shortcut for --print={0}.
|
help="""
|
||||||
'''.format(OUT_RESP_HEAD))
|
Print only the response headers. Shortcut for --print={0}.
|
||||||
|
|
||||||
|
"""
|
||||||
|
.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',
|
||||||
help=_('''
|
action='store_const',
|
||||||
Print only the response body.
|
const=OUT_RESP_BODY,
|
||||||
Shortcut for --print={0}.
|
help="""
|
||||||
'''.format(OUT_RESP_BODY))
|
Print only the response body. Shortcut for --print={0}.
|
||||||
|
|
||||||
|
"""
|
||||||
|
.format(OUT_RESP_BODY)
|
||||||
)
|
)
|
||||||
|
|
||||||
output_options.add_argument('--stream', '-S', action='store_true', default=False,
|
output_options.add_argument(
|
||||||
help=_('''
|
'--stream', '-S',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
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'.
|
||||||
|
|
||||||
Without --stream and with --pretty (either set or implied),
|
Without --stream and with --pretty (either set or implied),
|
||||||
@ -215,138 +287,248 @@ 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_options.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. Note that the --output option needs to be
|
||||||
|
specified as well.
|
||||||
|
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
#######################################################################
|
||||||
# Sessions
|
# Sessions
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
sessions = parser.add_argument_group(title='Sessions')\
|
sessions = parser.add_argument_group(title='Sessions')\
|
||||||
.add_mutually_exclusive_group(required=False)
|
.add_mutually_exclusive_group(required=False)
|
||||||
|
|
||||||
|
session_name_validator = SessionNameValidator(
|
||||||
|
'Session name contains invalid characters.'
|
||||||
|
)
|
||||||
|
|
||||||
sessions.add_argument(
|
sessions.add_argument(
|
||||||
'--session', metavar='SESSION_NAME',
|
'--session',
|
||||||
help=_('''
|
metavar='SESSION_NAME_OR_PATH',
|
||||||
Create, or reuse and update a session.
|
type=session_name_validator,
|
||||||
Withing a session, custom headers, auth credential, as well as any
|
help="""
|
||||||
cookies sent by the server persist between requests.
|
Create, or reuse and update a session. Within a session, custom headers,
|
||||||
You can use the `httpie' management command to manipulate
|
auth credential, as well as any cookies sent by the server persist between
|
||||||
and inspect existing sessions. See `httpie --help'.
|
requests.
|
||||||
''')
|
|
||||||
|
Session files are stored in:
|
||||||
|
|
||||||
|
{session_dir}/<HOST>/<SESSION_NAME>.json.
|
||||||
|
|
||||||
|
"""
|
||||||
|
.format(session_dir=DEFAULT_SESSIONS_DIR)
|
||||||
)
|
)
|
||||||
sessions.add_argument(
|
sessions.add_argument(
|
||||||
'--session-read-only', metavar='SESSION_NAME',
|
'--session-read-only',
|
||||||
help=_('''
|
metavar='SESSION_NAME_OR_PATH',
|
||||||
Create or reuse a session, but do not update it once saved.
|
type=session_name_validator,
|
||||||
''')
|
help="""
|
||||||
|
Create or read a session without updating it form the request/response
|
||||||
|
exchange.
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
#######################################################################
|
||||||
# Authentication
|
# Authentication
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
# ``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), HTTPie will prompt
|
||||||
HTTPie will prompt for the password.
|
for the password.
|
||||||
'''),
|
|
||||||
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
_auth_plugins = plugin_manager.get_auth_plugins()
|
||||||
auth.add_argument(
|
auth.add_argument(
|
||||||
'--auth-type', choices=['basic', 'digest'], default='basic',
|
'--auth-type',
|
||||||
help=_('''
|
choices=[plugin.auth_type for plugin in _auth_plugins],
|
||||||
The authentication mechanism to be used.
|
default=_auth_plugins[0].auth_type,
|
||||||
Defaults to "basic".
|
help="""
|
||||||
''')
|
The authentication mechanism to be used. Defaults to "{default}".
|
||||||
|
|
||||||
|
{types}
|
||||||
|
|
||||||
|
"""
|
||||||
|
.format(default=_auth_plugins[0].auth_type, types='\n '.join(
|
||||||
|
'"{type}": {name}{package}{description}'.format(
|
||||||
|
type=plugin.auth_type,
|
||||||
|
name=plugin.name,
|
||||||
|
package=(
|
||||||
|
'' if issubclass(plugin, BuiltinAuthPlugin)
|
||||||
|
else ' (provided by %s)' % plugin.package_name
|
||||||
|
),
|
||||||
|
description=(
|
||||||
|
'' if not plugin.description else
|
||||||
|
'\n ' + ('\n '.join(wrap(plugin.description)))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
for plugin in _auth_plugins
|
||||||
|
)),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
# 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 (e.g. http:foo.bar:3128).
|
||||||
(e.g. http:foo.bar:3128). You can specify multiple
|
You can specify multiple proxies with different protocols.
|
||||||
proxies with different protocols.
|
|
||||||
''')
|
"""
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--follow', default=False, action='store_true',
|
'--follow',
|
||||||
help=_('''
|
default=False,
|
||||||
Set this flag if full redirects are allowed
|
action='store_true',
|
||||||
(e.g. re-POST-ing of data at new ``Location``)
|
help="""
|
||||||
''')
|
Set this flag if full redirects are allowed (e.g. re-POST-ing of data at
|
||||||
|
new Location).
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--verify', default='yes',
|
'--verify',
|
||||||
help=_('''
|
default='yes',
|
||||||
Set to "no" to skip checking the host\'s SSL certificate.
|
help="""
|
||||||
You can also pass the path to a CA_BUNDLE
|
Set to "no" to skip checking the host's SSL certificate. You can also pass
|
||||||
file for private certs. You can also set
|
the path to a CA_BUNDLE file for private certs. You can also set the
|
||||||
the REQUESTS_CA_BUNDLE environment variable.
|
REQUESTS_CA_BUNDLE environment variable. Defaults to "yes".
|
||||||
Defaults to "yes".
|
|
||||||
''')
|
"""
|
||||||
)
|
)
|
||||||
|
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--timeout', type=float, default=30, metavar='SECONDS',
|
'--timeout',
|
||||||
help=_('''
|
type=float,
|
||||||
The connection timeout of the request in seconds.
|
default=30,
|
||||||
The default value is 30 seconds.
|
metavar='SECONDS',
|
||||||
''')
|
help="""
|
||||||
|
The connection timeout of the request in seconds. The default value is
|
||||||
|
30 seconds.
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
network.add_argument(
|
network.add_argument(
|
||||||
'--check-status', default=False, action='store_true',
|
'--check-status',
|
||||||
help=_('''
|
default=False,
|
||||||
By default, HTTPie exits with 0 when no network or other fatal
|
action='store_true',
|
||||||
errors occur.
|
help="""
|
||||||
|
By default, HTTPie exits with 0 when no network or other fatal errors
|
||||||
|
occur. This flag instructs HTTPie to also check the HTTP status code and
|
||||||
|
exit with an error if the status indicates one.
|
||||||
|
|
||||||
This flag instructs HTTPie to also check the HTTP status code and
|
When the server replies with a 4xx (Client Error) or 5xx (Server Error)
|
||||||
exit with an error if the status indicates one.
|
status code, HTTPie exits with 4 or 5 respectively. If the response is a
|
||||||
|
3xx (Redirect) and --follow hasn't been set, then the exit status is 3.
|
||||||
|
Also an error message is written to stderr if stdout is redirected.
|
||||||
|
|
||||||
When the server replies with a 4xx (Client Error) or 5xx
|
"""
|
||||||
(Server Error) status code, HTTPie exits with 4 or 5 respectively.
|
|
||||||
If the response is a 3xx (Redirect) and --follow
|
|
||||||
hasn't been set, then the exit status is 3.
|
|
||||||
|
|
||||||
Also an error message is written to stderr if stdout is redirected.
|
|
||||||
|
|
||||||
''')
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
#######################################################################
|
||||||
# Troubleshooting
|
# Troubleshooting
|
||||||
###############################################################################
|
#######################################################################
|
||||||
|
|
||||||
troubleshooting = parser.add_argument_group(title='Troubleshooting')
|
troubleshooting = parser.add_argument_group(title='Troubleshooting')
|
||||||
|
|
||||||
|
troubleshooting.add_argument(
|
||||||
|
'--ignore-stdin',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help="""
|
||||||
|
Do not attempt to read stdin.
|
||||||
|
|
||||||
|
"""
|
||||||
|
)
|
||||||
troubleshooting.add_argument(
|
troubleshooting.add_argument(
|
||||||
'--help',
|
'--help',
|
||||||
action='help', default=SUPPRESS,
|
action='help',
|
||||||
help='Show this help message and exit'
|
default=SUPPRESS,
|
||||||
)
|
help="""
|
||||||
troubleshooting.add_argument('--version', action='version', version=__version__)
|
Show this help message and exit.
|
||||||
troubleshooting.add_argument(
|
|
||||||
'--traceback', action='store_true', default=False,
|
"""
|
||||||
help='Prints exception traceback should one occur.'
|
|
||||||
)
|
)
|
||||||
troubleshooting.add_argument(
|
troubleshooting.add_argument(
|
||||||
'--debug', action='store_true', default=False,
|
'--version',
|
||||||
help=_('''
|
action='version',
|
||||||
Prints exception traceback should one occur, and also other
|
version=__version__,
|
||||||
information that is useful for debugging HTTPie itself and
|
help="""
|
||||||
for bug reports.
|
Show version and exit.
|
||||||
''')
|
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
troubleshooting.add_argument(
|
||||||
|
'--traceback',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help="""
|
||||||
|
Prints exception traceback should one occur.
|
||||||
|
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
troubleshooting.add_argument(
|
||||||
|
'--debug',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help="""
|
||||||
|
Prints exception traceback should one occur, and also other information
|
||||||
|
that is useful for debugging HTTPie itself and for reporting bugs.
|
||||||
|
|
||||||
|
"""
|
||||||
)
|
)
|
||||||
|
@ -4,10 +4,10 @@ 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__
|
||||||
|
from .plugins import plugin_manager
|
||||||
|
|
||||||
|
|
||||||
FORM = 'application/x-www-form-urlencoded; charset=utf-8'
|
FORM = 'application/x-www-form-urlencoded; charset=utf-8'
|
||||||
@ -15,57 +15,73 @@ JSON = 'application/json; charset=utf-8'
|
|||||||
DEFAULT_UA = 'HTTPie/%s' % __version__
|
DEFAULT_UA = 'HTTPie/%s' % __version__
|
||||||
|
|
||||||
|
|
||||||
|
class HTTPie(object):
|
||||||
|
|
||||||
|
def __init__(self, env, plugin_manager):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
def get_response(args, config_dir):
|
def get_response(args, config_dir):
|
||||||
"""Send the request and return a `request.Response`."""
|
"""Send the request and return a `request.Response`."""
|
||||||
|
|
||||||
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(
|
||||||
|
args=args,
|
||||||
config_dir=config_dir,
|
config_dir=config_dir,
|
||||||
name=args.session or args.session_read_only,
|
session_name=args.session or args.session_read_only,
|
||||||
request_kwargs=requests_kwargs,
|
requests_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:
|
||||||
credentials = {
|
auth_plugin = plugin_manager.get_auth_plugin(args.auth_type)()
|
||||||
'basic': requests.auth.HTTPBasicAuth,
|
credentials = auth_plugin.get_auth(args.auth.key, args.auth.value)
|
||||||
'digest': requests.auth.HTTPDigestAuth,
|
|
||||||
}[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 +96,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
|
||||||
|
19
httpie/compat.py
Normal file
19
httpie/compat.py
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
"""
|
||||||
|
Python 2/3 compatibility.
|
||||||
|
|
||||||
|
"""
|
||||||
|
#noinspection PyUnresolvedReferences
|
||||||
|
from requests.compat import (
|
||||||
|
is_windows,
|
||||||
|
bytes,
|
||||||
|
str,
|
||||||
|
is_py3,
|
||||||
|
is_py26,
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
#noinspection PyUnresolvedReferences,PyCompatibility
|
||||||
|
from urllib.parse import urlsplit
|
||||||
|
except ImportError:
|
||||||
|
#noinspection PyUnresolvedReferences,PyCompatibility
|
||||||
|
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(
|
||||||
@ -16,8 +16,9 @@ DEFAULT_CONFIG_DIR = os.environ.get(
|
|||||||
class BaseConfigDict(dict):
|
class BaseConfigDict(dict):
|
||||||
|
|
||||||
name = None
|
name = None
|
||||||
help = None
|
helpurl = 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)
|
||||||
@ -27,18 +28,24 @@ class BaseConfigDict(dict):
|
|||||||
def __getattr__(self, item):
|
def __getattr__(self, item):
|
||||||
return self[item]
|
return self[item]
|
||||||
|
|
||||||
@property
|
def _get_path(self):
|
||||||
def path(self):
|
"""Return the config file path without side-effects."""
|
||||||
try:
|
|
||||||
os.makedirs(self.directory, mode=0o700)
|
|
||||||
except OSError as e:
|
|
||||||
if e.errno != errno.EEXIST:
|
|
||||||
raise
|
|
||||||
return os.path.join(self.directory, self.name + '.json')
|
return os.path.join(self.directory, self.name + '.json')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def path(self):
|
||||||
|
"""Return the config file path creating basedir, if needed."""
|
||||||
|
path = self._get_path()
|
||||||
|
try:
|
||||||
|
os.makedirs(os.path.dirname(path), mode=0o700)
|
||||||
|
except OSError as e:
|
||||||
|
if e.errno != errno.EEXIST:
|
||||||
|
raise
|
||||||
|
return path
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def is_new(self):
|
def is_new(self):
|
||||||
return not os.path.exists(self.path)
|
return not os.path.exists(self._get_path())
|
||||||
|
|
||||||
def load(self):
|
def load(self):
|
||||||
try:
|
try:
|
||||||
@ -56,7 +63,15 @@ class BaseConfigDict(dict):
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
def save(self):
|
def save(self):
|
||||||
self['__version__'] = __version__
|
self['__meta__'] = {
|
||||||
|
'httpie': __version__
|
||||||
|
}
|
||||||
|
if self.helpurl:
|
||||||
|
self['__meta__']['help'] = self.helpurl
|
||||||
|
|
||||||
|
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 +87,8 @@ class BaseConfigDict(dict):
|
|||||||
class Config(BaseConfigDict):
|
class Config(BaseConfigDict):
|
||||||
|
|
||||||
name = 'config'
|
name = 'config'
|
||||||
|
helpurl = 'https://github.com/jkbr/httpie#config'
|
||||||
|
about = 'HTTPie configuration file'
|
||||||
|
|
||||||
DEFAULTS = {
|
DEFAULTS = {
|
||||||
'implicit_content_type': 'json',
|
'implicit_content_type': 'json',
|
||||||
|
114
httpie/core.py
114
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 .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
|
||||||
|
from .plugins import plugin_manager
|
||||||
|
|
||||||
|
|
||||||
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,117 @@ 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.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
plugin_manager.load_installed_plugins()
|
||||||
|
from .cli import parser
|
||||||
|
|
||||||
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
|
||||||
|
error('Incomplete download: size=%d; downloaded=%d' % (
|
||||||
|
download.status.total_size,
|
||||||
|
download.status.downloaded
|
||||||
|
))
|
||||||
|
|
||||||
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
|
||||||
|
427
httpie/downloads.py
Normal file
427
httpie/downloads.py
Normal file
@ -0,0 +1,427 @@
|
|||||||
|
# coding=utf-8
|
||||||
|
"""
|
||||||
|
Download mode implementation.
|
||||||
|
|
||||||
|
"""
|
||||||
|
from __future__ import division
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
import mimetypes
|
||||||
|
import threading
|
||||||
|
from time import sleep, time
|
||||||
|
from mailbox import Message
|
||||||
|
|
||||||
|
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
|
||||||
|
|
||||||
|
msg = Message('Content-Disposition: %s' % content_disposition)
|
||||||
|
filename = msg.get_filename()
|
||||||
|
if filename:
|
||||||
|
# Basic sanitation.
|
||||||
|
filename = os.path.basename(filename).lstrip('.').strip()
|
||||||
|
if filename:
|
||||||
|
return filename
|
||||||
|
|
||||||
|
|
||||||
|
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 = ProgressReporterThread(
|
||||||
|
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.start()
|
||||||
|
|
||||||
|
return stream, self._output_file
|
||||||
|
|
||||||
|
def finish(self):
|
||||||
|
assert not self.finished
|
||||||
|
self.finished = True
|
||||||
|
self.status.finished()
|
||||||
|
|
||||||
|
def failed(self):
|
||||||
|
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 ProgressReporterThread(threading.Thread):
|
||||||
|
"""
|
||||||
|
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
|
||||||
|
"""
|
||||||
|
super(ProgressReporterThread, self).__init__()
|
||||||
|
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._should_stop = threading.Event()
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
"""Stop reporting on next tick."""
|
||||||
|
self._should_stop.set()
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
while not self._should_stop.is_set():
|
||||||
|
if self.status.has_finished:
|
||||||
|
self.sum_up()
|
||||||
|
break
|
||||||
|
|
||||||
|
self.report_speed()
|
||||||
|
sleep(self._tick)
|
||||||
|
|
||||||
|
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}:{1:0>2}:{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()
|
299
httpie/input.py
299
httpie/input.py
@ -6,17 +6,22 @@ import sys
|
|||||||
import re
|
import re
|
||||||
import json
|
import json
|
||||||
import mimetypes
|
import mimetypes
|
||||||
import getpass
|
from getpass import getpass
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
from argparse import ArgumentParser, ArgumentTypeError
|
#noinspection PyCompatibility
|
||||||
|
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
|
||||||
|
from .sessions import VALID_SESSION_NAME_PATTERN
|
||||||
|
|
||||||
|
|
||||||
HTTP_POST = 'POST'
|
HTTP_POST = 'POST'
|
||||||
@ -95,37 +100,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':
|
# Arguments processing and environment setup.
|
||||||
args.form = True
|
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 self.args.ignore_stdin and not env.stdin_isatty:
|
||||||
|
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 args.debug:
|
return self.args
|
||||||
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)
|
|
||||||
|
|
||||||
if not env.stdin_isatty:
|
|
||||||
self._body_from_file(args, env.stdin)
|
|
||||||
|
|
||||||
if not (args.url.startswith(HTTP) or args.url.startswith(HTTPS)):
|
|
||||||
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 +136,232 @@ 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.
|
||||||
|
if self.args.ignore_stdin:
|
||||||
|
self.error('Unable to prompt for passwords because'
|
||||||
|
' --ignore-stdin is set.')
|
||||||
|
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.args.ignore_stdin and 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 = (
|
||||||
item.sep in SEP_GROUP_DATA_ITEMS for item in args.items)
|
(not self.args.ignore_stdin and
|
||||||
args.method = HTTP_POST if has_data else HTTP_GET
|
not self.env.stdin_isatty) or any(
|
||||||
|
item.sep in SEP_GROUP_DATA_ITEMS
|
||||||
|
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 +381,19 @@ class KeyValue(object):
|
|||||||
return self.__dict__ == other.__dict__
|
return self.__dict__ == other.__dict__
|
||||||
|
|
||||||
|
|
||||||
|
class SessionNameValidator(object):
|
||||||
|
|
||||||
|
def __init__(self, error_message):
|
||||||
|
self.error_message = error_message
|
||||||
|
|
||||||
|
def __call__(self, value):
|
||||||
|
# Session name can be a path or just a name.
|
||||||
|
if (os.path.sep not in value
|
||||||
|
and not VALID_SESSION_NAME_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`.
|
||||||
|
|
||||||
@ -292,8 +424,8 @@ class KeyValueArgType(object):
|
|||||||
"""Tokenize `s`. There are only two token types - strings
|
"""Tokenize `s`. There are only two token types - strings
|
||||||
and escaped characters:
|
and escaped characters:
|
||||||
|
|
||||||
>>> tokenize(r'foo\=bar\\baz')
|
tokenize(r'foo\=bar\\baz')
|
||||||
['foo', Escaped('='), 'bar', Escaped('\\'), 'baz']
|
=> ['foo', Escaped('='), 'bar', Escaped('\\'), 'baz']
|
||||||
|
|
||||||
"""
|
"""
|
||||||
tokens = ['']
|
tokens = ['']
|
||||||
@ -403,9 +535,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
|
||||||
@ -100,6 +109,7 @@ class HTTPResponse(HTTPMessage):
|
|||||||
def iter_lines(self, chunk_size):
|
def iter_lines(self, chunk_size):
|
||||||
return ((line, b'\n') for line in self._orig.iter_lines(chunk_size))
|
return ((line, b'\n') for line in self._orig.iter_lines(chunk_size))
|
||||||
|
|
||||||
|
#noinspection PyProtectedMember
|
||||||
@property
|
@property
|
||||||
def headers(self):
|
def headers(self):
|
||||||
original = self._orig.raw._original_response
|
original = self._orig.raw._original_response
|
||||||
@ -143,33 +153,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 +179,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
|
|
||||||
|
@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
import json
|
import json
|
||||||
|
import xml.dom.minidom
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from itertools import chain
|
from itertools import chain
|
||||||
|
|
||||||
@ -12,14 +13,17 @@ 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,
|
||||||
OUT_RESP_HEAD, OUT_RESP_BODY)
|
OUT_RESP_HEAD, OUT_RESP_BODY)
|
||||||
|
|
||||||
|
|
||||||
|
# The default number of spaces to indent when pretty printing
|
||||||
|
DEFAULT_INDENT = 4
|
||||||
|
|
||||||
# Colors on Windows via colorama don't look that
|
# Colors on Windows via colorama don't look that
|
||||||
# great and fruity seems to give the best result there.
|
# great and fruity seems to give the best result there.
|
||||||
AVAILABLE_STYLES = set(STYLE_MAP.keys())
|
AVAILABLE_STYLES = set(STYLE_MAP.keys())
|
||||||
@ -61,7 +65,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 +82,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 +122,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 +149,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 +195,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 +213,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 +229,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 +249,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()
|
||||||
@ -261,8 +268,9 @@ class PrettyStream(EncodedStream):
|
|||||||
def _process_body(self, chunk):
|
def _process_body(self, chunk):
|
||||||
return (self.processor
|
return (self.processor
|
||||||
.process_body(
|
.process_body(
|
||||||
chunk.decode(self.msg.encoding, 'replace'),
|
content=chunk.decode(self.msg.encoding, 'replace'),
|
||||||
self.msg.content_type)
|
content_type=self.msg.content_type,
|
||||||
|
encoding=self.msg.encoding)
|
||||||
.encode(self.output_encoding, 'replace'))
|
.encode(self.output_encoding, 'replace'))
|
||||||
|
|
||||||
|
|
||||||
@ -276,9 +284,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()
|
||||||
@ -365,12 +372,13 @@ class BaseProcessor(object):
|
|||||||
"""
|
"""
|
||||||
return headers
|
return headers
|
||||||
|
|
||||||
def process_body(self, content, content_type, subtype):
|
def process_body(self, content, content_type, subtype, encoding):
|
||||||
"""Return processed `content`.
|
"""Return processed `content`.
|
||||||
|
|
||||||
:param content: The body content as text
|
:param content: The body content as text
|
||||||
:param content_type: Full content type, e.g., 'application/atom+xml'.
|
:param content_type: Full content type, e.g., 'application/atom+xml'.
|
||||||
:param subtype: E.g. 'xml'.
|
:param subtype: E.g. 'xml'.
|
||||||
|
:param encoding: The original content encoding.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
return content
|
return content
|
||||||
@ -379,7 +387,7 @@ class BaseProcessor(object):
|
|||||||
class JSONProcessor(BaseProcessor):
|
class JSONProcessor(BaseProcessor):
|
||||||
"""JSON body processor."""
|
"""JSON body processor."""
|
||||||
|
|
||||||
def process_body(self, content, content_type, subtype):
|
def process_body(self, content, content_type, subtype, encoding):
|
||||||
if subtype == 'json':
|
if subtype == 'json':
|
||||||
try:
|
try:
|
||||||
# Indent the JSON data, sort keys by name, and
|
# Indent the JSON data, sort keys by name, and
|
||||||
@ -387,13 +395,29 @@ class JSONProcessor(BaseProcessor):
|
|||||||
content = json.dumps(json.loads(content),
|
content = json.dumps(json.loads(content),
|
||||||
sort_keys=True,
|
sort_keys=True,
|
||||||
ensure_ascii=False,
|
ensure_ascii=False,
|
||||||
indent=4)
|
indent=DEFAULT_INDENT)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
# Invalid JSON but we don't care.
|
# Invalid JSON but we don't care.
|
||||||
pass
|
pass
|
||||||
return content
|
return content
|
||||||
|
|
||||||
|
|
||||||
|
class XMLProcessor(BaseProcessor):
|
||||||
|
"""XML body processor."""
|
||||||
|
# TODO: tests
|
||||||
|
|
||||||
|
def process_body(self, content, content_type, subtype, encoding):
|
||||||
|
if subtype == 'xml':
|
||||||
|
try:
|
||||||
|
# Pretty print the XML
|
||||||
|
doc = xml.dom.minidom.parseString(content.encode(encoding))
|
||||||
|
content = doc.toprettyxml(indent=' ' * DEFAULT_INDENT)
|
||||||
|
except xml.parsers.expat.ExpatError:
|
||||||
|
# Ignore invalid XML errors (skips attempting to pretty print)
|
||||||
|
pass
|
||||||
|
return content
|
||||||
|
|
||||||
|
|
||||||
class PygmentsProcessor(BaseProcessor):
|
class PygmentsProcessor(BaseProcessor):
|
||||||
"""A processor that applies syntax-highlighting using Pygments
|
"""A processor that applies syntax-highlighting using Pygments
|
||||||
to the headers, and to the body as well if its content type is recognized.
|
to the headers, and to the body as well if its content type is recognized.
|
||||||
@ -425,7 +449,7 @@ class PygmentsProcessor(BaseProcessor):
|
|||||||
return pygments.highlight(
|
return pygments.highlight(
|
||||||
headers, HTTPLexer(), self.formatter).strip()
|
headers, HTTPLexer(), self.formatter).strip()
|
||||||
|
|
||||||
def process_body(self, content, content_type, subtype):
|
def process_body(self, content, content_type, subtype, encoding):
|
||||||
try:
|
try:
|
||||||
lexer = self.lexers_by_type.get(content_type)
|
lexer = self.lexers_by_type.get(content_type)
|
||||||
if not lexer:
|
if not lexer:
|
||||||
@ -458,7 +482,8 @@ class OutputProcessor(object):
|
|||||||
installed_processors = {
|
installed_processors = {
|
||||||
'format': [
|
'format': [
|
||||||
HeadersProcessor,
|
HeadersProcessor,
|
||||||
JSONProcessor
|
JSONProcessor,
|
||||||
|
XMLProcessor
|
||||||
],
|
],
|
||||||
'colors': [
|
'colors': [
|
||||||
PygmentsProcessor
|
PygmentsProcessor
|
||||||
@ -484,13 +509,18 @@ class OutputProcessor(object):
|
|||||||
headers = processor.process_headers(headers)
|
headers = processor.process_headers(headers)
|
||||||
return headers
|
return headers
|
||||||
|
|
||||||
def process_body(self, content, content_type):
|
def process_body(self, content, content_type, encoding):
|
||||||
# e.g., 'application/atom+xml'
|
# e.g., 'application/atom+xml'
|
||||||
content_type = content_type.split(';')[0]
|
content_type = content_type.split(';')[0]
|
||||||
# e.g., 'xml'
|
# e.g., 'xml'
|
||||||
subtype = content_type.split('/')[-1].split('+')[-1]
|
subtype = content_type.split('/')[-1].split('+')[-1]
|
||||||
|
|
||||||
for processor in self.processors:
|
for processor in self.processors:
|
||||||
content = processor.process_body(content, content_type, subtype)
|
content = processor.process_body(
|
||||||
|
content,
|
||||||
|
content_type,
|
||||||
|
subtype,
|
||||||
|
encoding
|
||||||
|
)
|
||||||
|
|
||||||
return content
|
return content
|
||||||
|
9
httpie/plugins/__init__.py
Normal file
9
httpie/plugins/__init__.py
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
from .base import AuthPlugin
|
||||||
|
from .manager import PluginManager
|
||||||
|
from .builtin import BasicAuthPlugin, DigestAuthPlugin
|
||||||
|
|
||||||
|
|
||||||
|
plugin_manager = PluginManager()
|
||||||
|
plugin_manager.register(BasicAuthPlugin)
|
||||||
|
plugin_manager.register(DigestAuthPlugin)
|
||||||
|
|
28
httpie/plugins/base.py
Normal file
28
httpie/plugins/base.py
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
class AuthPlugin(object):
|
||||||
|
"""
|
||||||
|
Base auth plugin class.
|
||||||
|
|
||||||
|
See <https://github.com/jkbr/httpie-ntlm> for an example auth plugin.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
# The value that should be passed to --auth-type
|
||||||
|
# to use this auth plugin. Eg. "my-auth"
|
||||||
|
auth_type = None
|
||||||
|
|
||||||
|
# The name of the plugin, eg. "My auth".
|
||||||
|
name = None
|
||||||
|
|
||||||
|
# Optional short description. Will be be shown in the help
|
||||||
|
# under --auth-type.
|
||||||
|
description = None
|
||||||
|
|
||||||
|
# This be set automatically once the plugin has been loaded.
|
||||||
|
package_name = None
|
||||||
|
|
||||||
|
def get_auth(self, username, password):
|
||||||
|
"""
|
||||||
|
Return a ``requests.auth.AuthBase`` subclass instance.
|
||||||
|
|
||||||
|
"""
|
||||||
|
raise NotImplementedError()
|
26
httpie/plugins/builtin.py
Normal file
26
httpie/plugins/builtin.py
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
import requests.auth
|
||||||
|
|
||||||
|
from .base import AuthPlugin
|
||||||
|
|
||||||
|
|
||||||
|
class BuiltinAuthPlugin(AuthPlugin):
|
||||||
|
|
||||||
|
package_name = '(builtin)'
|
||||||
|
|
||||||
|
|
||||||
|
class BasicAuthPlugin(BuiltinAuthPlugin):
|
||||||
|
|
||||||
|
name = 'Basic HTTP auth'
|
||||||
|
auth_type = 'basic'
|
||||||
|
|
||||||
|
def get_auth(self, username, password):
|
||||||
|
return requests.auth.HTTPBasicAuth(username, password)
|
||||||
|
|
||||||
|
|
||||||
|
class DigestAuthPlugin(BuiltinAuthPlugin):
|
||||||
|
|
||||||
|
name = 'Digest HTTP auth'
|
||||||
|
auth_type = 'digest'
|
||||||
|
|
||||||
|
def get_auth(self, username, password):
|
||||||
|
return requests.auth.HTTPDigestAuth(username, password)
|
35
httpie/plugins/manager.py
Normal file
35
httpie/plugins/manager.py
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
from pkg_resources import iter_entry_points
|
||||||
|
|
||||||
|
|
||||||
|
ENTRY_POINT_NAMES = [
|
||||||
|
'httpie.plugins.auth.v1'
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class PluginManager(object):
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self._plugins = []
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return iter(self._plugins)
|
||||||
|
|
||||||
|
def register(self, plugin):
|
||||||
|
self._plugins.append(plugin)
|
||||||
|
|
||||||
|
def get_auth_plugins(self):
|
||||||
|
return list(self._plugins)
|
||||||
|
|
||||||
|
def get_auth_plugin_mapping(self):
|
||||||
|
return dict((plugin.auth_type, plugin) for plugin in self)
|
||||||
|
|
||||||
|
def get_auth_plugin(self, auth_type):
|
||||||
|
return self.get_auth_plugin_mapping()[auth_type]
|
||||||
|
|
||||||
|
def load_installed_plugins(self):
|
||||||
|
|
||||||
|
for entry_point_name in ENTRY_POINT_NAMES:
|
||||||
|
for entry_point in iter_entry_points(entry_point_name):
|
||||||
|
plugin = entry_point.load()
|
||||||
|
plugin.package_name = entry_point.dist.key
|
||||||
|
self.register(entry_point.load())
|
@ -1,115 +1,120 @@
|
|||||||
"""Persistent, JSON-serialized sessions.
|
"""Persistent, JSON-serialized sessions.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
import re
|
||||||
import os
|
import os
|
||||||
import sys
|
|
||||||
import glob
|
|
||||||
import errno
|
|
||||||
import codecs
|
|
||||||
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 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
|
from httpie.plugins import plugin_manager
|
||||||
|
|
||||||
|
|
||||||
SESSIONS_DIR_NAME = 'sessions'
|
SESSIONS_DIR_NAME = 'sessions'
|
||||||
|
DEFAULT_SESSIONS_DIR = os.path.join(DEFAULT_CONFIG_DIR, SESSIONS_DIR_NAME)
|
||||||
|
VALID_SESSION_NAME_PATTERN = re.compile('^[a-zA-Z0-9_.-]+$')
|
||||||
|
# Request headers starting with these prefixes won't be stored in sessions.
|
||||||
|
# They are specific to each request.
|
||||||
|
# http://en.wikipedia.org/wiki/List_of_HTTP_header_fields#Requests
|
||||||
|
SESSION_IGNORED_HEADER_PREFIXES = ['Content-', 'If-']
|
||||||
|
|
||||||
|
|
||||||
def get_response(name, request_kwargs, config_dir, read_only=False):
|
def get_response(session_name, requests_kwargs, config_dir, args,
|
||||||
|
read_only=False):
|
||||||
"""Like `client.get_response`, but applies permanent
|
"""Like `client.get_response`, but applies permanent
|
||||||
aspects of the session to the request.
|
aspects of the session to the request.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
sessions_dir = os.path.join(config_dir, SESSIONS_DIR_NAME)
|
if os.path.sep in session_name:
|
||||||
host = Host(
|
path = os.path.expanduser(session_name)
|
||||||
root_dir=sessions_dir,
|
else:
|
||||||
name=request_kwargs['headers'].get('Host', None)
|
hostname = (
|
||||||
or urlparse(request_kwargs['url']).netloc.split('@')[-1]
|
requests_kwargs['headers'].get('Host', None)
|
||||||
)
|
or urlsplit(requests_kwargs['url']).netloc.split('@')[-1]
|
||||||
|
)
|
||||||
|
|
||||||
session = Session(host, name)
|
assert re.match('^[a-zA-Z0-9_.:-]+$', hostname)
|
||||||
|
|
||||||
|
# host:port => host_port
|
||||||
|
hostname = hostname.replace(':', '_')
|
||||||
|
path = os.path.join(config_dir,
|
||||||
|
SESSIONS_DIR_NAME,
|
||||||
|
hostname,
|
||||||
|
session_name + '.json')
|
||||||
|
|
||||||
|
session = Session(path)
|
||||||
session.load()
|
session.load()
|
||||||
|
|
||||||
# Update session headers with the request headers.
|
request_headers = requests_kwargs.get('headers', {})
|
||||||
session['headers'].update(request_kwargs.get('headers', {}))
|
requests_kwargs['headers'] = dict(session.headers, **request_headers)
|
||||||
# Use the merged headers for the request
|
session.update_headers(request_headers)
|
||||||
request_kwargs['headers'] = session['headers']
|
|
||||||
|
|
||||||
auth = request_kwargs.get('auth', None)
|
if args.auth:
|
||||||
if auth:
|
session.auth = {
|
||||||
session.auth = auth
|
'type': args.auth_type,
|
||||||
|
'username': args.auth.key,
|
||||||
|
'password': args.auth.value,
|
||||||
|
}
|
||||||
elif session.auth:
|
elif session.auth:
|
||||||
request_kwargs['auth'] = session.auth
|
requests_kwargs['auth'] = session.auth
|
||||||
|
|
||||||
|
requests_session = requests.Session()
|
||||||
|
requests_session.cookies = session.cookies
|
||||||
|
|
||||||
rsession = requests.Session(cookies=session.cookies)
|
|
||||||
try:
|
try:
|
||||||
response = rsession.request(**request_kwargs)
|
response = requests_session.request(**requests_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
|
||||||
|
|
||||||
|
|
||||||
class Host(object):
|
|
||||||
"""A host is a per-host directory on the disk containing sessions files."""
|
|
||||||
|
|
||||||
def __init__(self, name, root_dir=DEFAULT_CONFIG_DIR):
|
|
||||||
self.name = name
|
|
||||||
self.root_dir = root_dir
|
|
||||||
|
|
||||||
def __iter__(self):
|
|
||||||
"""Return a iterator yielding `(session_name, session_path)`."""
|
|
||||||
for fn in sorted(glob.glob1(self.path, '*.json')):
|
|
||||||
yield os.path.splitext(fn)[0], os.path.join(self.path, fn)
|
|
||||||
|
|
||||||
def delete(self):
|
|
||||||
shutil.rmtree(self.path)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def path(self):
|
|
||||||
# Name will include ':' if a port is specified, which is invalid
|
|
||||||
# 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:
|
|
||||||
os.makedirs(path, mode=0o700)
|
|
||||||
except OSError as e:
|
|
||||||
if e.errno != errno.EEXIST:
|
|
||||||
raise
|
|
||||||
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):
|
||||||
""""""
|
helpurl = 'https://github.com/jkbr/httpie#sessions'
|
||||||
|
about = 'HTTPie session file'
|
||||||
|
|
||||||
def __init__(self, host, name, *args, **kwargs):
|
def __init__(self, path, *args, **kwargs):
|
||||||
super(Session, self).__init__(*args, **kwargs)
|
super(Session, self).__init__(*args, **kwargs)
|
||||||
self.host = host
|
self._path = path
|
||||||
self.name = name
|
|
||||||
self['headers'] = {}
|
self['headers'] = {}
|
||||||
self['cookies'] = {}
|
self['cookies'] = {}
|
||||||
|
self['auth'] = {
|
||||||
|
'type': None,
|
||||||
|
'username': None,
|
||||||
|
'password': None
|
||||||
|
}
|
||||||
|
|
||||||
|
def _get_path(self):
|
||||||
|
return self._path
|
||||||
|
|
||||||
|
def update_headers(self, request_headers):
|
||||||
|
"""
|
||||||
|
Update the session headers with the request ones while ignoring
|
||||||
|
certain name prefixes.
|
||||||
|
|
||||||
|
:type request_headers: dict
|
||||||
|
|
||||||
|
"""
|
||||||
|
for name, value in request_headers.items():
|
||||||
|
|
||||||
|
if name == 'User-Agent' and value.startswith('HTTPie/'):
|
||||||
|
continue
|
||||||
|
|
||||||
|
for prefix in SESSION_IGNORED_HEADER_PREFIXES:
|
||||||
|
if name.lower().startswith(prefix.lower()):
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self['headers'][name] = value
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def directory(self):
|
def headers(self):
|
||||||
return self.host.path
|
return self['headers']
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def cookies(self):
|
def cookies(self):
|
||||||
@ -122,108 +127,27 @@ class Session(BaseConfigDict):
|
|||||||
|
|
||||||
@cookies.setter
|
@cookies.setter
|
||||||
def cookies(self, jar):
|
def cookies(self, jar):
|
||||||
excluded = [
|
"""
|
||||||
'_rest', 'name', 'port_specified',
|
:type jar: CookieJar
|
||||||
'domain_specified', 'domain_initial_dot',
|
"""
|
||||||
'path_specified', 'comment', 'comment_url'
|
# http://docs.python.org/2/library/cookielib.html#cookie-objects
|
||||||
]
|
stored_attrs = ['value', 'path', 'secure', 'expires']
|
||||||
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_plugin = plugin_manager.get_auth_plugin(auth['type'])()
|
||||||
'digest': HTTPDigestAuth}[auth['type']]
|
return auth_plugin.get_auth(auth['username'], auth['password'])
|
||||||
return Auth(auth['username'], auth['password'])
|
|
||||||
|
|
||||||
@auth.setter
|
@auth.setter
|
||||||
def auth(self, cred):
|
def auth(self, auth):
|
||||||
self['auth'] = {
|
assert set(['type', 'username', 'password']) == set(auth.keys())
|
||||||
'type': {HTTPBasicAuth: 'basic',
|
self['auth'] = auth
|
||||||
HTTPDigestAuth: 'digest'}[type(cred)],
|
|
||||||
'username': cred.username,
|
|
||||||
'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
|
||||||
|
git+git://github.com/kennethreitz/httpbin.git@7c96875e87a448f08fb1981e85eb79e77d592d98
|
||||||
|
docutils
|
20
setup.py
20
setup.py
@ -1,6 +1,5 @@
|
|||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import re
|
|
||||||
import codecs
|
import codecs
|
||||||
from setuptools import setup
|
from setuptools import setup
|
||||||
import httpie
|
import httpie
|
||||||
@ -12,25 +11,23 @@ if sys.argv[-1] == 'test':
|
|||||||
|
|
||||||
|
|
||||||
requirements = [
|
requirements = [
|
||||||
# Debian has only requests==0.10.1 and httpie.deb depends on that.
|
'requests>=1.2.3',
|
||||||
'requests>=0.10.1',
|
|
||||||
'Pygments>=1.5'
|
'Pygments>=1.5'
|
||||||
]
|
]
|
||||||
if sys.version_info[:2] in ((2, 6), (3, 1)):
|
try:
|
||||||
# argparse has been added in Python 3.2 / 2.7
|
#noinspection PyUnresolvedReferences
|
||||||
|
import argparse
|
||||||
|
except ImportError:
|
||||||
requirements.append('argparse>=1.2.1')
|
requirements.append('argparse>=1.2.1')
|
||||||
|
|
||||||
if 'win32' in str(sys.platform).lower():
|
if 'win32' in str(sys.platform).lower():
|
||||||
# Terminal colors for Windows
|
# Terminal colors for Windows
|
||||||
requirements.append('colorama>=0.2.4')
|
requirements.append('colorama>=0.2.4')
|
||||||
|
|
||||||
|
|
||||||
def long_description():
|
def long_description():
|
||||||
"""Pre-process the README so that PyPi can render it properly."""
|
|
||||||
with codecs.open('README.rst', encoding='utf8') as f:
|
with codecs.open('README.rst', encoding='utf8') as f:
|
||||||
rst = f.read()
|
return f.read()
|
||||||
code_block = '(:\n\n)?\.\. code-block::.*'
|
|
||||||
rst = re.sub(code_block, '::', rst)
|
|
||||||
return rst
|
|
||||||
|
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
@ -43,11 +40,10 @@ setup(
|
|||||||
author=httpie.__author__,
|
author=httpie.__author__,
|
||||||
author_email='jakub@roztocil.name',
|
author_email='jakub@roztocil.name',
|
||||||
license=httpie.__licence__,
|
license=httpie.__licence__,
|
||||||
packages=['httpie'],
|
packages=['httpie', 'httpie.plugins'],
|
||||||
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,
|
||||||
|
473
tests/tests.py
473
tests/tests.py
@ -23,19 +23,27 @@ import subprocess
|
|||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
|
#noinspection PyCompatibility
|
||||||
import argparse
|
import argparse
|
||||||
import tempfile
|
import tempfile
|
||||||
import unittest
|
import unittest
|
||||||
import shutil
|
import shutil
|
||||||
|
import time
|
||||||
|
from requests.structures import CaseInsensitiveDict
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
#noinspection PyCompatibility
|
||||||
from urllib.request import urlopen
|
from urllib.request import urlopen
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
# noinspection PyUnresolvedReferences
|
||||||
|
#noinspection PyCompatibility
|
||||||
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 +52,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 +62,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'
|
||||||
@ -97,7 +113,8 @@ with open(BIN_FILE_PATH, 'rb') as f:
|
|||||||
|
|
||||||
|
|
||||||
def httpbin(path):
|
def httpbin(path):
|
||||||
return HTTPBIN_URL + path
|
url = HTTPBIN_URL + path
|
||||||
|
return url
|
||||||
|
|
||||||
|
|
||||||
def mk_config_dir():
|
def mk_config_dir():
|
||||||
@ -130,6 +147,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 +156,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 +171,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 +196,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 +236,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 +261,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 +293,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 +302,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 +461,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 +546,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 +685,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,7 +809,10 @@ 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(
|
||||||
|
'--verbose',
|
||||||
'POST',
|
'POST',
|
||||||
httpbin('/post'),
|
httpbin('/post'),
|
||||||
'@' + FILE_PATH_ARG
|
'@' + FILE_PATH_ARG
|
||||||
@ -785,6 +822,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 +877,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 +895,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 +929,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 +937,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 +946,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 +957,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 +971,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 +980,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 +991,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 +1044,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 +1063,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',
|
||||||
@ -1017,12 +1083,35 @@ class StreamTest(BaseTestCase):
|
|||||||
self.assertIn(BIN_FILE_CONTENT, r)
|
self.assertIn(BIN_FILE_CONTENT, r)
|
||||||
|
|
||||||
|
|
||||||
|
class IgnoreStdinTest(BaseTestCase):
|
||||||
|
|
||||||
|
def test_ignore_stdin(self):
|
||||||
|
with open(FILE_PATH) as f:
|
||||||
|
r = http(
|
||||||
|
'--ignore-stdin',
|
||||||
|
'--verbose',
|
||||||
|
httpbin('/get'),
|
||||||
|
env=TestEnvironment(stdin=f, stdin_isatty=False)
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r)
|
||||||
|
self.assertIn('GET /get HTTP', r) # Don't default to POST.
|
||||||
|
self.assertNotIn(FILE_CONTENT, r) # Don't send stdin data.
|
||||||
|
|
||||||
|
def test_ignore_stdin_cannot_prompt_password(self):
|
||||||
|
r = http(
|
||||||
|
'--ignore-stdin',
|
||||||
|
'--auth=username-without-password',
|
||||||
|
httpbin('/get'),
|
||||||
|
)
|
||||||
|
self.assertEqual(r.exit_status, ExitStatus.ERROR)
|
||||||
|
self.assertIn('because --ignore-stdin', r.stderr)
|
||||||
|
|
||||||
|
|
||||||
class LineEndingsTest(BaseTestCase):
|
class LineEndingsTest(BaseTestCase):
|
||||||
"""Test that CRLF is properly used in headers and
|
"""Test that CRLF is properly used in headers and
|
||||||
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 +1146,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):
|
||||||
@ -1110,6 +1199,8 @@ class ItemParsingTest(BaseTestCase):
|
|||||||
# files
|
# files
|
||||||
self.key_value_type('bar\\@baz@%s' % FILE_PATH_ARG)
|
self.key_value_type('bar\\@baz@%s' % FILE_PATH_ARG)
|
||||||
])
|
])
|
||||||
|
# `requests.structures.CaseInsensitiveDict` => `dict`
|
||||||
|
headers = dict(headers._store.values())
|
||||||
self.assertDictEqual(headers, {
|
self.assertDictEqual(headers, {
|
||||||
'foo:bar': 'baz',
|
'foo:bar': 'baz',
|
||||||
'jack@jill': 'hill',
|
'jack@jill': 'hill',
|
||||||
@ -1139,6 +1230,8 @@ class ItemParsingTest(BaseTestCase):
|
|||||||
self.key_value_type('test-file@%s' % FILE_PATH_ARG),
|
self.key_value_type('test-file@%s' % FILE_PATH_ARG),
|
||||||
self.key_value_type('query==value'),
|
self.key_value_type('query==value'),
|
||||||
])
|
])
|
||||||
|
# `requests.structures.CaseInsensitiveDict` => `dict`
|
||||||
|
headers = dict(headers._store.values())
|
||||||
self.assertDictEqual(headers, {
|
self.assertDictEqual(headers, {
|
||||||
'header': 'value',
|
'header': 'value',
|
||||||
'eh': ''
|
'eh': ''
|
||||||
@ -1162,78 +1255,113 @@ 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.args.ignore_stdin = False
|
||||||
|
|
||||||
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.args.ignore_stdin = False
|
||||||
|
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.args.ignore_stdin = False
|
||||||
|
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.args.ignore_stdin = False
|
||||||
|
|
||||||
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.args.ignore_stdin = False
|
||||||
|
|
||||||
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 +1395,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',
|
||||||
@ -1280,6 +1408,26 @@ class SessionTest(BaseTestCase):
|
|||||||
self.assertEqual(r.json['headers']['Cookie'], 'hello=world')
|
self.assertEqual(r.json['headers']['Cookie'], 'hello=world')
|
||||||
self.assertIn('Basic ', r.json['headers']['Authorization'])
|
self.assertIn('Basic ', r.json['headers']['Authorization'])
|
||||||
|
|
||||||
|
def test_session_ignored_header_prefixes(self):
|
||||||
|
r = http(
|
||||||
|
'--session=test',
|
||||||
|
'GET',
|
||||||
|
httpbin('/get'),
|
||||||
|
'Content-Type: text/plain',
|
||||||
|
'If-Unmodified-Since: Sat, 29 Oct 1994 19:43:31 GMT',
|
||||||
|
env=self.env
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r)
|
||||||
|
|
||||||
|
r2 = http(
|
||||||
|
'--session=test',
|
||||||
|
'GET',
|
||||||
|
httpbin('/get')
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r2)
|
||||||
|
self.assertNotIn('Content-Type', r2.json['headers'])
|
||||||
|
self.assertNotIn('If-Unmodified-Since', r2.json['headers'])
|
||||||
|
|
||||||
def test_session_update(self):
|
def test_session_update(self):
|
||||||
# Get a response to a request from the original session.
|
# Get a response to a request from the original session.
|
||||||
r1 = http(
|
r1 = http(
|
||||||
@ -1316,7 +1464,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 +1496,187 @@ 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)
|
||||||
|
|
||||||
|
def test_session_by_path(self):
|
||||||
|
session_path = os.path.join(self.config_dir, 'session-by-path.json')
|
||||||
|
|
||||||
|
r1 = http(
|
||||||
|
'--session=' + session_path,
|
||||||
|
'GET',
|
||||||
|
httpbin('/get'),
|
||||||
|
'Foo:Bar',
|
||||||
|
env=self.env
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r1)
|
||||||
|
|
||||||
|
r2 = http(
|
||||||
|
'--session=' + session_path,
|
||||||
|
'GET',
|
||||||
|
httpbin('/get'),
|
||||||
|
env=self.env
|
||||||
|
)
|
||||||
|
self.assertIn(OK, r2)
|
||||||
|
|
||||||
|
self.assertEqual(r2.json['headers']['Foo'], 'Bar')
|
||||||
|
|
||||||
|
|
||||||
|
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.assertEqual(
|
||||||
|
parse('attachment; filename="white space.txt"'),
|
||||||
|
'white space.txt'
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
parse(r'attachment; filename="\"quotes\".txt"'),
|
||||||
|
'"quotes".txt'
|
||||||
|
)
|
||||||
|
self.assertEqual(parse('attachment; filename=/etc/hosts'), 'hosts')
|
||||||
|
self.assertIsNone(parse('attachment; filename='))
|
||||||
|
|
||||||
|
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_with_Content_Length(self):
|
||||||
|
download = Download(output_file=open(os.devnull, 'w'))
|
||||||
|
download.start(Response(
|
||||||
|
url=httpbin('/'),
|
||||||
|
headers={'Content-Length': 10}
|
||||||
|
))
|
||||||
|
time.sleep(1.1)
|
||||||
|
download.chunk_downloaded(b'12345')
|
||||||
|
time.sleep(1.1)
|
||||||
|
download.chunk_downloaded(b'12345')
|
||||||
|
download.finish()
|
||||||
|
self.assertFalse(download.interrupted)
|
||||||
|
|
||||||
|
def test_download_no_Content_Length(self):
|
||||||
|
download = Download(output_file=open(os.devnull, 'w'))
|
||||||
|
download.start(Response(url=httpbin('/')))
|
||||||
|
time.sleep(1.1)
|
||||||
|
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