从 selenium 加载 cookie 以使用 cookielib 进行机械化

Question

我正在尝试使用 selenium 登录网站，然后将 cookie 转移到机械化。我已成功使用 selenium 登录并将其会话 cookie 保存到变量中。尝试使用 cookielib 加载 cookie 时出现问题。

相关编码：

.
. #loging in to website with selenium
.
cookie = browser.get_cookies()   #save the session cookie from selenium to variable "cookie"
.
. #starting up mechanize
.
cj = cookielib.LWPCookieJar() 
.
.
.
cj.set_cookie(cookie) #load cookie from selenium

使用 cj.set_cookie 函数设置 cookie 时出现问题，我收到以下错误消息

File "..../cookielib.py", line 1627, in set_cookie
if cookie.domain not in c: c[cookie.domain] = {}
AttributeError: 'list' object has no attribute 'domain'

Answer 1

如果您打印 Selenium 收集的 cookie 并将其与 mechanize/cookielib 收集的 cookie 进行比较，您会发现它们使用不同的格式。

要克服这个问题，你可以尝试这样的事情（你可能需要稍微修改一下，以满足你的需要。但你明白了一般的想法）：

cj = cookielib.LWPCookieJar()

for s_cookie in cookie:
    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 = s_cookie['expiry'], discard = False, comment = None, comment_url = None, rest = None, rfc2109 = False))

一个更奇特的解决方案是遍历 selenium cookie 并制作一个带有 name : value 对的字典