ó
05Vc           @   s   d  Z  d d l Z d d l Z d d l m Z d d l m Z d d l m Z d e j f d „  ƒ  YZ d	 e f d
 „  ƒ  YZ	 d S(   s/   The implementation of the SocketOptionsAdapter.iÿÿÿÿN(   t   adaptersi   (   t
   connection(   t   poolmanagert   SocketOptionsAdapterc           B   sJ   e  Z d  Z e e j d e j e j d f g ƒ Z	 d „  Z
 e d „ Z RS(   sd  An adapter for requests that allows users to specify socket options.

    Since version 2.4.0 of requests, it is possible to specify a custom list
    of socket options that need to be set before establishing the connection.

    Example usage::

        >>> import socket
        >>> import requests
        >>> from requests_toolbelt.adapters import socket_options
        >>> s = requests.Session()
        >>> opts = [(socket.IPPROTO_TCP, socket.TCP_NODELAY, 0)]
        >>> adapter = socket_options.SocketOptionsAdapter(socket_options=opts)
        >>> s.mount('http://', adapter)

    You can also take advantage of the list of default options on this class
    to keep using the original options in addition to your custom options. In
    that case, ``opts`` might look like::

        >>> opts = socket_options.SocketOptionsAdapter.default_options + opts

    t   default_socket_optionsi   c         K   s2   | j  d |  j ƒ |  _ t t |  ƒ j |   d  S(   Nt   socket_options(   t   popt   default_optionsR   t   superR   t   __init__(   t   selft   kwargs(    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyR	   *   s    	c      	   C   s\   t  j d k r< t j d | d | d | d |  j ƒ |  _ n t t |  ƒ j | | | ƒ d  S(   Ni  t	   num_poolst   maxsizet   blockR   (   t   requestst	   __build__R   t   PoolManagerR   R   R   t   init_poolmanager(   R
   t   connectionsR   R   (    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyR   0   s    	(   t   __name__t
   __module__t   __doc__t   getattrR   t   HTTPConnectiont   sockett   IPPROTO_TCPt   TCP_NODELAYR   R	   t   FalseR   (    (    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyR      s   	t   TCPKeepAliveAdapterc           B   s   e  Z d  Z d „  Z RS(   sJ  An adapter for requests that turns on TCP Keep-Alive by default.

    The adapter sets 4 socket options:

    - ``SOL_SOCKET`` ``SO_KEEPALIVE`` - This turns on TCP Keep-Alive
    - ``IPPROTO_TCP`` ``TCP_KEEPINTVL`` 20 - Sets the keep alive interval
    - ``IPPROTO_TCP`` ``TCP_KEEPCNT`` 5 - Sets the number of keep alive probes
    - ``IPPROTO_TCP`` ``TCP_KEEPIDLE`` 60 - Sets the keep alive time if the
      socket library has the ``TCP_KEEPIDLE`` constant

    The latter three can be overridden by keyword arguments (respectively):

    - ``idle``
    - ``interval``
    - ``count``

    You can use this adapter like so::

       >>> from requests_toolbelt.adapters import socket_options
       >>> tcp = socket_options.TCPKeepAliveAdapter(idle=120, interval=10)
       >>> s = requests.Session()
       >>> s.mount('http://', tcp)

    c         K   sâ   | j  d t j ƒ } | j  d d ƒ } | j  d d ƒ } | j  d d ƒ } | t j t j d f t j t j | f t j t j | f g } t	 t d	 d  ƒ d  k	 rÂ | t j t j | f g 7} n  t t |  ƒ j d | |  d  S(
   NR   t   idlei<   t   intervali   t   counti   i   t   TCP_KEEPIDLE(   R   R   R   R   t
   SOL_SOCKETt   SO_KEEPALIVER   t   TCP_KEEPINTVLt   TCP_KEEPCNTR   t   NoneR!   R   R   R	   (   R
   R   R   R   R   R    (    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyR	   Y   s    	(   R   R   R   R	   (    (    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyR   ?   s   (
   R   R   R   R    t   _compatR   R   t   HTTPAdapterR   R   (    (    (    sM   /usr/lib/python2.7/dist-packages/requests_toolbelt/adapters/socket_options.pyt   <module>   s   3