Как «поддерживать жизнь» с cookielib и httplib в питоне? - PullRequest
Новая
       29

Как «поддерживать жизнь» с cookielib и httplib в питоне?

1 голос
/ 19 июня 2009

В python я использую httplib, потому что он «поддерживает активность» http-соединения (в отличие от urllib (2)). Теперь я хочу использовать cookielib с httplib, но они, похоже, ненавидят друг друга !! (нет способа соединить их вместе).

Кто-нибудь знает решение этой проблемы?

Ответы [ 2 ]

2 голосов
/ 28 июня 2009

Вам следует рассмотреть возможность использования библиотеки Requests при первой же возможности реорганизовать код. В то же время;

HACK ALERT! :)

Я бы пошел другим путем, но я сделал хак (хотя и по разным причинам), который создает интерфейс между httplib и cookielib .

Я создал фальшивый HTTPRequest с минимально необходимым набором методов, чтобы CookieJar распознавал его и обрабатывал файлы cookie по мере необходимости. Я использовал этот поддельный объект запроса, установив все данные, необходимые для cookielib.

Вот код класса: 

class HTTPRequest( object ):
"""
Data container for HTTP request (used for cookie processing).
"""

    def __init__( self, host, url, headers={}, secure=False ):
        self._host = host
        self._url = url
        self._secure = secure
        self._headers = {}
        for key, value in headers.items():
            self.add_header(key, value)

    def has_header( self, name ):
        return name in self._headers

    def add_header( self, key, val ):
        self._headers[key.capitalize()] = val

    def add_unredirected_header(self, key, val):
        self._headers[key.capitalize()] = val

    def is_unverifiable( self ):
        return True

    def get_type( self ):
        return 'https' if self._secure else 'http'

    def get_full_url( self ):
        port_str = ""
        port = str(self._host[1])
        if self._secure:
            if port != 443:
                port_str = ":"+port
        else:
            if port != 80:
                port_str = ":"+port
        return self.get_type() + '://' + self._host[0] + port_str + self._url

    def get_header( self, header_name, default=None ):
        return self._headers.get( header_name, default )

    def get_host( self ):
        return self._host[0]

    get_origin_req_host = get_host

    def get_headers( self ):
        return self._headers

Обратите внимание, что в классе есть поддержка только протокола HTTPS (все, что мне нужно на данный момент).

Код, который использовал этот класс был (пожалуйста, обратите внимание на другой хак, чтобы сделать ответ совместимым с cookielib): 

cookies = CookieJar()

headers = {
    # headers that you wish to set
}

# construct fake request
fake_request = HTTPRequest( host, request_url, headers )

# add cookies to fake request
cookies.add_cookie_header(fake_request)

# issue an httplib.HTTPConnection based request using cookies and headers from the fake request
http_connection.request(type, request_url, body, fake_request.get_headers())

response = http_connection.getresponse()

if response.status == httplib.OK:
    # HACK: pretend we're urllib2 response
    response.info = lambda : response.msg

    # read and store cookies from response
    cookies.extract_cookies(response, fake_request)

    # process response...
2 голосов
/ 19 июня 2009
...