使用 selenium Python 运行 execute_script 的正确方法是什么？

Question

我想使用需要粘贴到浏览器控制台的JS代码自动登录Discord。 我只是在学习 Python 并且对 JS 完全一无所知。 这是粘贴到控制台的代码：

function login(token) {
    setInterval(() => {
      document.body.appendChild(document.createElement `iframe`).contentWindow.localStorage.token = `"${token}"`
    }, 50);
    setTimeout(() => {
      location.reload();
    }, 2500);
  }

login(token);

这是我失败的尝试：

token = 'token'
driver.execute_script('return function login(token) {setInterval(() => {document.body.appendChild(document.createElement `iframe`).contentWindow.localStorage.token = `"${token}"`}, 50);setTimeout(() => {location.reload();}, 2500);}login('+"'"+f'{token}'+"');")

收到错误：

selenium.common.exceptions.JavascriptException: Message: javascript error: Unexpected identifier 'login'

Answer 1

token = 'token'

js = 'function login(token) {setInterval(() => {document.body.appendChild(document.createElement `iframe`).contentWindow.localStorage.token = `"${token}"`}, 50);setTimeout(() => {location.reload();}, 500);}'
time.sleep(3)
driver.execute_script(js + f'login("{token}")')