Python 使用 TinyDB 构建 Cookie 池：简单高效的实现方法

要使用 TinyDB 实现 cookie 池，首先需要安装 TinyDB 模块。可以使用以下命令进行安装：\n\n\npip install tinydb\n\n\n然后，创建一个cookies.json文件作为数据库存储 cookie 信息。\n\n接下来，可以使用以下代码实现 cookie 池的基本功能：\n\npython\nfrom tinydb import TinyDB, Query\n\nclass CookiePool:\n def __init__(self, db_path='cookies.json'):\n self.db = TinyDB(db_path)\n self.table = self.db.table('cookies')\n\n def add_cookie(self, domain, cookie):\n self.table.insert({'domain': domain, 'cookie': cookie})\n\n def get_cookie(self, domain):\n Cookie = Query()\n result = self.table.search(Cookie.domain == domain)\n return result[0]['cookie'] if result else None\n\n def delete_cookie(self, domain):\n Cookie = Query()\n self.table.remove(Cookie.domain == domain)\n\n\n然后，可以使用以下代码进行测试：\n\npython\ncookie_pool = CookiePool()\n\n# 添加 cookie\ncookie_pool.add_cookie('example.com', 'cookie_value')\n\n# 获取 cookie\ncookie = cookie_pool.get_cookie('example.com')\nprint(cookie) # 输出：cookie_value\n\n# 删除 cookie\ncookie_pool.delete_cookie('example.com')\n\n# 再次获取 cookie\ncookie = cookie_pool.get_cookie('example.com')\nprint(cookie) # 输出：None\n\n\n这样，你就可以使用 TinyDB 来实现一个简单的 cookie 池。