Discord.py应用：JSON文件参数批量添加与优化

import json
from discord.ext import commands

# 假设这是在一个Cog内部，并且user变量在实际环境中应该通过循环或参数传入
# 但此示例旨在说明原问题中代码的逻辑缺陷和潜在的效率问题
class ExampleCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.inventory_file = "cogs/inventory.json"

    @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_problematic(self, ctx: commands.Context) -> None:
        try:
            with open(self.inventory_file, "r", encoding="utf-8") as f:
                inventory = json.load(f)

            # 原始问题中的代码逻辑，存在多重问题：
            # 1. 'user' 变量未定义，会导致NameError。
            # 2. 即使 'user' 定义了，此代码也只针对一个用户操作，无法实现“更新所有人”的目的。
            # 3. 最重要的是，如果在循环中（即便这里没有显式循环）反复执行文件写入操作，效率会非常低下。
            #    每次写入都会重新打开文件，清空内容，然后写入整个JSON结构，这是昂贵的I/O操作。
            if "some_user_id" in inventory: # 假设这里有一个固定的用户ID进行演示
                inventory["some_user_id"]["law_tuition"] = 0
                with open(self.inventory_file, "w", encoding="utf-8") as f:
                    json.dump(inventory, f, indent=4, ensure_ascii=False)
                await ctx.send("Done!")
            else:
                await ctx.send("指定用户ID未找到或未执行更新。")

        except FileNotFoundError:
            await ctx.send(f"错误：库存文件 '{self.inventory_file}' 未找到。")
        except json.JSONDecodeError:
            await ctx.send(f"错误：库存文件 '{self.inventory_file}' 格式不正确。")
        except Exception as e:
            await ctx.send(f"更新库存时发生未知错误: {e}")
            print(f"Error updating inventory (problematic): {e}")

import json
from discord.ext import commands
import os # 用于检查文件是否存在

class Economy(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        # 定义JSON文件路径，建议使用os.path.join确保跨平台兼容性
        self.inventory_file = os.path.join("cogs", "inventory.json")

    @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:
        # 确保文件存在，如果不存在则创建一个空JSON对象
        if not os.path.exists(self.inventory_file):
            await ctx.send(f"库存文件 '{self.inventory_file}' 未找到，正在创建新文件...")
            with open(self.inventory_file, "w", encoding="utf-8") as f:
                json.dump({}, f, indent=4, ensure_ascii=False) # 创建一个空的JSON对象
            inventory = {} # 初始化为空字典
        else:
            try:
                # 1. 一次性加载所有数据到内存
                with open(self.inventory_file, "r", encoding="utf-8") as f:
                    inventory = json.load(f)
            except json.JSONDecodeError:
                await ctx.send(f"错误：库存文件 '{self.inventory_file}' 格式不正确。请检查JSON文件内容。")
                return
            except Exception as e:
                await ctx.send(f"读取库存文件时发生未知错误: {e}")
                print(f"Error reading inventory file: {e}")
                return

        # 2. 在内存中更新所有用户数据
        # 遍历所有用户ID及其数据
        # 示例JSON结构: {"[USER ID]": {"small_apartment": 0, "news_station": 0, ...}}
        for user_id_str, user_data in inventory.items():
            # 确保 user_data 是字典类型，以防止数据结构异常导致错误
            if isinstance(user_data, dict):
                # 添加或更新 'law_tuition' 参数，并将其值设为0
                user_data["law_tuition"] = 0
            else:
                # 如果user_data不是预期的字典格式，可以记录日志或跳过
                print(f"警告: 用户 '{user_id_str}' 的数据格式异常，跳过更新。数据: {user_data}")
                # 可以在这里选择初始化其为字典，或跳过
                # inventory[user_id_str] = {"law_tuition": 0} # 如果要强制初始化

        try:
            # 3. 一次性将更新后的数据写回文件
            with open(self.inventory_file, "w", encoding="utf-8") as f:
                # 使用 indent 参数使JSON文件更具可读性
                # ensure_ascii=False 允许直接写入非ASCII字符（如中文），而不是转义
                json.dump(inventory, f, indent=4, ensure_ascii=False)

            await ctx.send("所有用户库存已成功更新！")

        except Exception as e:
            await ctx.send(f"写入更新后的库存文件时发生错误: {e}")
            print(f"Error writing updated inventory file: {e}")