高效更新JSON数据：Discord.py应用中的库存管理优化实践

# 假设这里有一个循环，遍历所有用户
# for user in users:
#   if f"{user.id}" in inventory: # 假设inventory已加载
#     inventory[user.id]["law_tuition"] = 0
#     with open("cogs/inventory.json", "w") as f:
#       json.dump(inventory, f)
#     await ctx.send("Done!") # 每次更新都发送消息

import json
import discord
from discord.ext import commands

# 假设这是一个Cog类中的方法
class Economy(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.hybrid_command(name="update_shop", description="An administrative command used to update everyone's inventories when the shop is updated!")
    @commands.has_role("*") # 假设这里有正确的角色检查
    async def update_shop(self, ctx: commands.Context) -> None:
        try:
            with open("cogs/inventory.json", "r", encoding="utf-8") as f:
                inventory = json.load(f)
        except FileNotFoundError:
            await ctx.send("库存文件不存在，请检查路径或创建新文件。")
            return
        except json.JSONDecodeError:
            await ctx.send("库存文件格式错误，无法解析。")
            return
        # ... 后续操作

        # ... (接上文加载代码)

        # 遍历所有用户，更新或添加新参数
        for user_id, user_data in inventory.items():
            # 检查用户数据是否为字典类型，确保安全操作
            if isinstance(user_data, dict):
                user_data["law_tuition"] = 0
            # 如果需要，也可以在此处添加其他条件判断或更新逻辑

        # ... 后续保存操作

        # ... (接上文修改代码)

        try:
            with open("cogs/inventory.json", "w", encoding="utf-8") as f:
                json.dump(inventory, f, indent=4) # 使用indent=4使JSON文件更易读
            await ctx.send("库存已成功更新！")
        except IOError:
            await ctx.send("写入库存文件时发生错误，请检查权限。")

import json
import discord
from discord.ext import commands

class Economy(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.hybrid_command(name="update_shop", description="An administrative command used to update everyone's inventories when the shop is updated!")
    @commands.has_role("Admin") # 请替换为实际的角色名称或ID
    async def update_shop(self, ctx: commands.Context) -> None:
        file_path = "cogs/inventory.json" # 定义文件路径，方便管理

        try:
            # 1. 加载JSON数据
            with open(file_path, "r", encoding="utf-8") as f:
                inventory = json.load(f)
        except FileNotFoundError:
            await ctx.send(f"错误：库存文件 '{file_path}' 不存在。请确认路径或创建新文件。")
            return
        except json.JSONDecodeError:
            await ctx.send(f"错误：库存文件 '{file_path}' 格式错误，无法解析。请检查文件内容。")
            return
        except Exception as e:
            await ctx.send(f"加载库存文件时发生未知错误：{e}")
            return

        # 2. 遍历并修改数据（在内存中）
        updates_made = False
        for user_id, user_data in inventory.items():
            if isinstance(user_data, dict):
                if "law_tuition" not in user_data: # 仅在参数不存在时添加
                    user_data["law_tuition"] = 0
                    updates_made = True
            else:
                print(f"警告：用户 {user_id} 的数据不是字典类型，跳过更新。") # 记录异常数据

        if not updates_made:
            await ctx.send("所有用户库存已包含 'law_tuition' 参数，无需更新。")
            return

        try:
            # 3. 保存修改后的数据（一次性）
            with open(file_path, "w", encoding="utf-8") as f:
                json.dump(inventory, f, indent=4, ensure_ascii=False) # ensure_ascii=False支持中文
            await ctx.send("✅ 所有用户库存已成功更新，并添加了 'law_tuition' 参数！")
        except IOError as e:
            await ctx.send(f"错误：写入库存文件时发生I/O错误：{e}。请检查文件权限。")
        except Exception as e:
            await ctx.send(f"保存库存文件时发生未知错误：{e}")

# 在bot setup中加载Cog
async def setup(bot):
    await bot.add_cog(Economy(bot))