python - 在 urllib2 中使用来自 selenium 的 session cookie

Question

我正在尝试使用 Selenium 登录网站，然后使用 urllib2 发出 RESTy 请求。不过，为了使其正常工作，我需要 urllib2 才能使用与 Selenium 相同的 session 。

使用 selenium 登录效果很好，我可以调用

self.driver.get_cookies()

我有一个 selenium 知道的所有 cookie 的列表，它最终看起来有点像这样:

[{u'domain': u'my.awesome.web.app.local',
  u'expiry': 1319230106,
  u'name': u'ci_session',
  u'path': u'/',
  u'secure': False,
  u'value': u'9YEz6Qs9rNlONzXbZPZ5i9jm2Nn4HNrbaCJj2c%2B...'
}]

我尝试了几种不同的方式来使用 urllib2 中的 cookie，我认为这个看起来最好:

# self.driver is my selenium driver
all_cookies = self.driver.get_cookies()
cp = urllib2.HTTPCookieProcessor()
cj = cp.cookiejar
for s_cookie in all_cookies:
    cj.set_cookie(
        cookielib.Cookie(
            version=0
            , name=s_cookie['name']
            , value=s_cookie['value']
            , port='80'
            , port_specified=False
            , domain=s_cookie['domain']
            , domain_specified=True
            , domain_initial_dot=False
            , path=s_cookie['path']
            , path_specified=True
            , secure=s_cookie['secure']
            , expires=None
            , discard=False
            , comment=None
            , comment_url=None
            , rest=None
            , rfc2109=False
        )
    )
opener = urllib2.build_opener(cp)
response = opener.open(url_that_requires_a_logged_in_user)
response.geturl()

虽然它不起作用。

最后一次调用 response.geturl() 返回登录页面。

我错过了什么吗？

对于如何寻找问题有什么想法吗？

谢谢。

Answer 1

我能够通过使用 requests 库来克服这个问题。我迭代了来自 selenium 的 cookie，然后将它们传递到一个包含 name:value 对的简单字典中。

all_cookies = self.driver.get_cookies()

cookies = {}  
for s_cookie in all_cookies:
    cookies[s_cookie["name"]]=s_cookie["value"]

r = requests.get(my_url,cookies=cookies)