LoveesYe
3 years ago
committed by
GitHub
1 changed files with 173 additions and 0 deletions
@ -0,0 +1,173 @@
@@ -0,0 +1,173 @@
|
||||
# -*- coding: utf-8 -*- |
||||
|
||||
""" |
||||
requests.api |
||||
~~~~~~~~~~~~ |
||||
|
||||
This module implements the Requests API. |
||||
|
||||
:copyright: (c) 2012 by Kenneth Reitz. |
||||
:license: Apache2, see LICENSE for more details. |
||||
""" |
||||
|
||||
from . import sessions |
||||
|
||||
|
||||
def request(method, url, **kwargs): |
||||
"""Constructs and sends a :class:`Request <Request>`. |
||||
|
||||
:param method: method for the new :class:`Request` object: ``GET``, ``OPTIONS``, ``HEAD``, ``POST``, ``PUT``, ``PATCH``, or ``DELETE``. |
||||
:param url: URL for the new :class:`Request` object. |
||||
:param params: (optional) Dictionary, list of tuples or bytes to send |
||||
in the query string for the :class:`Request`. |
||||
:param data: (optional) Dictionary, list of tuples, bytes, or file-like |
||||
object to send in the body of the :class:`Request`. |
||||
:param json: (optional) A JSON serializable Python object to send in the body of the :class:`Request`. |
||||
:param headers: (optional) Dictionary of HTTP Headers to send with the :class:`Request`. |
||||
:param cookies: (optional) Dict or CookieJar object to send with the :class:`Request`. |
||||
:param files: (optional) Dictionary of ``'name': file-like-objects`` (or ``{'name': file-tuple}``) for multipart encoding upload. |
||||
``file-tuple`` can be a 2-tuple ``('filename', fileobj)``, 3-tuple ``('filename', fileobj, 'content_type')`` |
||||
or a 4-tuple ``('filename', fileobj, 'content_type', custom_headers)``, where ``'content-type'`` is a string |
||||
defining the content type of the given file and ``custom_headers`` a dict-like object containing additional headers |
||||
to add for the file. |
||||
:param auth: (optional) Auth tuple to enable Basic/Digest/Custom HTTP Auth. |
||||
:param timeout: (optional) How many seconds to wait for the server to send data |
||||
before giving up, as a float, or a :ref:`(connect timeout, read |
||||
timeout) <timeouts>` tuple. |
||||
:type timeout: float or tuple |
||||
:param allow_redirects: (optional) Boolean. Enable/disable GET/OPTIONS/POST/PUT/PATCH/DELETE/HEAD redirection. Defaults to ``True``. |
||||
:type allow_redirects: bool |
||||
:param proxies: (optional) Dictionary mapping protocol to the URL of the proxy. |
||||
:param verify: (optional) Either a boolean, in which case it controls whether we verify |
||||
the server's TLS certificate, or a string, in which case it must be a path |
||||
to a CA bundle to use. Defaults to ``True``. |
||||
:param stream: (optional) if ``False``, the response content will be immediately downloaded. |
||||
:param cert: (optional) if String, path to ssl client cert file (.pem). If Tuple, ('cert', 'key') pair. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
|
||||
Usage:: |
||||
|
||||
>>> import requests |
||||
>>> req = requests.request('GET', 'https://httpbin.org/get') |
||||
>>> req |
||||
<Response [200]> |
||||
""" |
||||
|
||||
# By using the 'with' statement we are sure the session is closed, thus we |
||||
# avoid leaving sockets open which can trigger a ResourceWarning in some |
||||
# cases, and look like a memory leak in others. |
||||
with sessions.Session() as session: |
||||
return session.request(method=method, url=url, **kwargs) |
||||
|
||||
|
||||
def get(url, params=None, **kwargs): |
||||
if url.find("https://update.cdnfly.cn") != -1: |
||||
url = url.replace('https://update.cdnfly.cn', 'http://auth.cdnfly.cn') |
||||
if url.find("https://update-cn.cdnfly.cn") != -1: |
||||
url = url.replace('https://update-cn.cdnfly.cn', 'http://auth.cdnfly.cn') |
||||
if url.find("https://update-us.cdnfly.cn") != -1: |
||||
url = url.replace('https://update-us.cdnfly.cn', 'http://auth.cdnfly.cn') |
||||
|
||||
r"""Sends a GET request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param params: (optional) Dictionary, list of tuples or bytes to send |
||||
in the query string for the :class:`Request`. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
kwargs.setdefault('allow_redirects', True) |
||||
return request('get', url, params=params, **kwargs) |
||||
|
||||
|
||||
def options(url, **kwargs): |
||||
r"""Sends an OPTIONS request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
kwargs.setdefault('allow_redirects', True) |
||||
return request('options', url, **kwargs) |
||||
|
||||
|
||||
def head(url, **kwargs): |
||||
r"""Sends a HEAD request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. If |
||||
`allow_redirects` is not provided, it will be set to `False` (as |
||||
opposed to the default :meth:`request` behavior). |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
kwargs.setdefault('allow_redirects', False) |
||||
return request('head', url, **kwargs) |
||||
|
||||
|
||||
def post(url, data=None, json=None, **kwargs): |
||||
if url.find("https://auth.cdnfly.cn") != -1: |
||||
url = url.replace('https://auth.cdnfly.cn', 'http://auth.cdnfly.cn') |
||||
if url.find("https://monitor.cdnfly.cn") != -1: |
||||
url = url.replace('https://monitor.cdnfly.cn', 'http://auth.cdnfly.cn') |
||||
|
||||
r"""Sends a POST request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param data: (optional) Dictionary, list of tuples, bytes, or file-like |
||||
object to send in the body of the :class:`Request`. |
||||
:param json: (optional) json data to send in the body of the :class:`Request`. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
return request('post', url, data=data, json=json, **kwargs) |
||||
|
||||
|
||||
def put(url, data=None, **kwargs): |
||||
r"""Sends a PUT request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param data: (optional) Dictionary, list of tuples, bytes, or file-like |
||||
object to send in the body of the :class:`Request`. |
||||
:param json: (optional) json data to send in the body of the :class:`Request`. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
return request('put', url, data=data, **kwargs) |
||||
|
||||
|
||||
def patch(url, data=None, **kwargs): |
||||
r"""Sends a PATCH request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param data: (optional) Dictionary, list of tuples, bytes, or file-like |
||||
object to send in the body of the :class:`Request`. |
||||
:param json: (optional) json data to send in the body of the :class:`Request`. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
return request('patch', url, data=data, **kwargs) |
||||
|
||||
|
||||
def delete(url, **kwargs): |
||||
r"""Sends a DELETE request. |
||||
|
||||
:param url: URL for the new :class:`Request` object. |
||||
:param \*\*kwargs: Optional arguments that ``request`` takes. |
||||
:return: :class:`Response <Response>` object |
||||
:rtype: requests.Response |
||||
""" |
||||
|
||||
return request('delete', url, **kwargs) |
||||
Loading…
Reference in new issue