如何使用Python和Selenium WebDriver获取localStorage

Question

如何获取LocalStorage，相当于以下代码的Cookies:

driver.get_cookies()

- Francesco Borzi

Answer 1

Python API 无法直接读写本地存储，但可以使用 execute_script 实现。

class LocalStorage:

    def __init__(self, driver) :
        self.driver = driver

    def __len__(self):
        return self.driver.execute_script("return window.localStorage.length;")

    def items(self) :
        return self.driver.execute_script( \
            "var ls = window.localStorage, items = {}; " \
            "for (var i = 0, k; i < ls.length; ++i) " \
            "  items[k = ls.key(i)] = ls.getItem(k); " \
            "return items; ")

    def keys(self) :
        return self.driver.execute_script( \
            "var ls = window.localStorage, keys = []; " \
            "for (var i = 0; i < ls.length; ++i) " \
            "  keys[i] = ls.key(i); " \
            "return keys; ")

    def get(self, key):
        return self.driver.execute_script("return window.localStorage.getItem(arguments[0]);", key)

    def set(self, key, value):
        self.driver.execute_script("window.localStorage.setItem(arguments[0], arguments[1]);", key, value)

    def has(self, key):
        return key in self.keys()

    def remove(self, key):
        self.driver.execute_script("window.localStorage.removeItem(arguments[0]);", key)

    def clear(self):
        self.driver.execute_script("window.localStorage.clear();")

    def __getitem__(self, key) :
        value = self.get(key)
        if value is None :
          raise KeyError(key)
        return value

    def __setitem__(self, key, value):
        self.set(key, value)

    def __contains__(self, key):
        return key in self.keys()

    def __iter__(self):
        return self.items().__iter__()

    def __repr__(self):
        return self.items().__str__()

使用示例：

# get the local storage
storage = LocalStorage(driver)

# set an item
storage["mykey"] = 1234
storage.set("mykey2", 5678)

# get an item
print(storage["mykey"])      # raises a KeyError if the key is missing
print(storage.get("mykey"))  # returns None if the key is missing

# delete an item
storage.remove("mykey")

# iterate items
for key, value in storage.items():
  print("%s: %s" % (key, value))

# delete items
storage.clear()

- Florent B.

---

关闭并重新打开浏览器后，保存的值是否可以被访问？ - West

可能不会，因为每个新窗口都是在隐身模式下。 - vitaliis

酷炫的实现 - Den Silver

我尝试在本地存储上运行此解决方案，该存储具有多个项/条目，通过执行此命令 print(storage.items())，但它返回空字典。 有什么解决方法吗？ - ZA09

有没有想法如何在geckodriver中使storage.clear()正常工作，而不会出现“JavascriptException: SecurityError: The operation is insecure”错误？ - Marcel Wilson

Answer 2

我使用以下代码解决了问题:

driver.execute_script("return window.localStorage;")

编辑：这是一个快速简短的答案。有关更详细的答案，请参见Florent B.的答案。

- Francesco Borzi

Answer 3

请参考这个问题的回答： 在Selenium中获取JavaScript代码的返回值。

基本上，您需要执行以下操作：

from selenium import webdriver

wd = webdriver.Firefox()
wd.get("http://localhost/foo/bar")
wd.execute_script("return localStorage.getItem('foo')")

- amirouche

Answer 4

尝试：

driver.execute_script("window.localStorage.setItem('key','value');")
driver.execute_script("window.localStorage.getItem('key');")

- Maxim