Python中替代JavaScript localStorage的轻量数据持久化方案咨询

针对你的需求（本地脚本、小量字典/字符串数据、跨脚本保留、类似JS localStorage的用法），逐个解答你的问题：

1. pickle 是否合适？ #

pickle能实现持久化，但不是最优选择：

• 优点：可以序列化几乎所有Python对象，写法简单
• 缺点：
  • 生成的是二进制文件，人类不可读，不方便调试
  • 安全风险极高：加载不可信来源的pickle文件可能执行恶意代码，尤其你的数据包含用户输入时，绝对不推荐
  • 仅支持Python环境读取，不跨语言

对于你这种仅存字典、字符串的简单场景，有更轻量安全的替代方案。

2. 各方案对比与推荐 #

JSON（首推） #

完全匹配你的需求：

• 轻量无依赖，Python标准库自带json模块
• 序列化后是人类可读的文本格式，方便查看和修改
• 安全可控，只支持基本数据类型（字典、字符串、数字等），不会执行任何代码
• 跨语言兼容，其他语言也能读取
• 容易封装成你想要的storage.set/get接口

实现示例：

import json
import os

class SimpleStorage:
    def __init__(self, file_path="storage.json"):
        self.file_path = file_path
        # 初始化空存储文件（如果不存在）
        if not os.path.exists(self.file_path):
            with open(self.file_path, "w", encoding="utf-8") as f:
                json.dump({}, f)

    def set(self, key, value):
        # 读取现有数据
        with open(self.file_path, "r", encoding="utf-8") as f:
            data = json.load(f)
        # 更新并写入
        data[key] = value
        with open(self.file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, indent=2)

    def get(self, key, default=None):
        with open(self.file_path, "r", encoding="utf-8") as f:
            data = json.load(f)
        return data.get(key, default)

# 使用示例
storage = SimpleStorage()
storage.set("user_prefs", {"theme": "dark", "lang": "en"})
prefs = storage.get("user_prefs")
print(prefs)  # 脚本重启后仍能读取到数据

shelve #

• 接口类似Python原生字典，使用方便，支持多键值存储
• 底层依赖pickle序列化，同样存在安全风险，不适合处理用户输入数据
• 适合频繁修改多键值的场景，但不如JSON安全可读

sqlite3 #

• 属于轻量数据库，适合结构化数据、多表关联、复杂查询场景
• 对于你的小量键值数据来说，操作过于繁琐，属于过度设计

diskcache（第三方库） #

• 功能强大，支持缓存过期、批量操作等高级特性
• 但需要额外安装依赖（pip install diskcache），对于简单的本地持久化需求，没必要引入

3. 包含用户输入字符串时的安全选择 #

绝对优先选JSON：

• JSON只会把用户输入当作普通文本序列化/反序列化，不会执行任何代码，完全规避恶意注入风险
• pickle/shelve绝对不能用：恶意用户可构造特殊pickle数据，在加载时执行任意代码
• sqlite3虽然安全，但操作复杂度远高于JSON，没必要为小数据引入

内容的提问来源于stack exchange，提问作者Waqas Ahmed