如何使用Selenium处理Google身份验证器

Question

我需要从sellercentral.amazon.de下载大量excel文件（估计：500 - 1000）。手动下载不是一种选择，因为每次下载都需要多次点击才能弹出excel。

由于亚马逊不能为我提供一个简单的xml结构，我决定自己进行自动化。想到的第一件事是Selenium和Firefox。

问题：

需要登录到sellercentral，以及双因素身份验证（2FA）。因此，如果我登录一次，我可以打开另一个标签，输入sellercentral.amazon.de并立即登录。我甚至可以打开另一个浏览器实例，并立即登录。他们可能正在使用会话cookie。 “刮”的目标网址是https://sellercentral.amazon.de/listing/download?ref=ag_dnldinv_apvu_newapvu。

但是当我使用selenium webdrive打开我的python脚本中的URL时，会启动一个新的浏览器实例，其中我没有登录。即使有同时运行的firefox实例，我也是登录。所以我猜硒发起的实例有些不同。

我尝试过的：

我尝试在第一个.get（）（打开网站）之后设置一个时间延迟，然后我将手动登录，然后重做.get（），这使得脚本永远继续。

from selenium import webdriver
import time


browser = webdriver.Firefox()

# Wait for website to fire onload event
browser.get("https://sellercentral.amazon.de/listing/download?ref=ag_dnldinv_apvu_newapvu")

time.sleep(30000)

browser.get("https://sellercentral.amazon.de/listing/download?ref=ag_dnldinv_apvu_newapvu")

elements = browser.find_elements_by_tag_name("browse-node-component")


print(str(elements))

我在找什么？

需要使用谷歌身份验证器的双因素身份验证令牌的解决方案。

我希望selenium在firefox浏览器的现有实例中作为选项卡打开，我将在此处事先登录。因此，不需要登录（应该），并且可以进行“抓取”和下载。如果没有直接的方法，也许有人想出一个解决方法？

我知道selenium无法下载文件本身，因为弹出窗口不再是浏览器的一部分。当我到达那里时，我会解决这个问题。

重要的附注：Firefox不是给定的！我很乐意接受任何浏览器的解决方案。

Answer 1

以下是将读取Google身份验证器令牌并在登录中使用的代码。使用js打开新选项卡。在运行测试代码之前安装pyotp软件包。

pip安装pyotp

测试代码：

from pyotp import *
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

driver = webdriver.Firefox()
driver.get("https://sellercentral.amazon.de/listing/download?ref=ag_dnldinv_apvu_newapvu")
wait = WebDriverWait(driver,10)
# enter the email
email = wait.until(EC.presence_of_element_located((By.XPATH, "//input[@name='email']")))
email.send_keys("email goes here")

# enter password
driver.find_element_by_xpath("//input[@name='password']").send_keys("password goes here")

# click on signin button
driver.find_element_by_xpath("//input[@id='signInSubmit']").click()

#wait for the 2FA feild to display
authField = wait.until(EC.presence_of_element_located((By.xpath, "xpath goes here")))
# get the token from google authenticator
totp = TOTP("secret goes here")
token = totp.now()
print (token)
# enter the token in the UI
authField.send_keys(token)
# click on the button to complete 2FA
driver.find_element_by_xpath("xpath of the button goes here").click()
# now open new tab
driver.execute_script("""window.open("https://sellercentral.amazon.de/listing/download?ref=ag_dnldinv_apvu_newapvu")""")
# continue with your logic from here