在Python批量向MySQL数据库插入数据的场景中，将变量信息存储到YAML配置文件中可以提升代码的可维护性和灵活性。通过YAML文件，可以集中管理数据库连接信息、插入数据的字段定义以及其他配置参数，使代码更加清晰和易于调整。 ### YAML配置文件的结构设计 YAML文件通常以键值对的形式存储配置信息，可以嵌套结构以表示更复杂的数据关系。以下是一个适合存储数据库插入操作相关变量的YAML文件示例： ```yaml database: host: localhost port: 3306 username: user password: secret database: test_db insert_config: table: users fields: - name - age - email ``` 上述配置定义了数据库连接信息和插入操作的目标表及字段。这种结构使得在代码中读取和使用这些配置信息变得简单直观。 ### 使用PyYAML库读取和写入YAML文件 Python中可以通过`PyYAML`库来读取和写入YAML文件。以下是如何将变量保存到YAML配置文件中的步骤： #### 1. 安装PyYAML库 如果尚未安装`PyYAML`库，可以通过以下命令进行安装： ```bash pip install pyyaml ``` #### 2. 写入YAML配置文件 可以将变量信息写入YAML文件，以便后续使用。以下是一个示例代码，演示如何将数据库连接信息和插入配置写入YAML文件： ```python import yaml # 定义需要保存到YAML的变量信息 config = { "database": { "host": "localhost", "port": 3306, "username": "user", "password": "secret", "database": "test_db" }, "insert_config": { "table": "users", "fields": ["name", "age", "email"] } } # 写入YAML文件 with open("config.yaml", "w", encoding="utf-8") as file: yaml.dump(config, file, allow_unicode=True) ``` #### 3. 读取YAML配置文件 在批量插入数据时，可以通过读取YAML文件获取配置信息。以下是一个示例代码，展示如何从YAML文件中读取配置并用于数据库操作： ```python import yaml import mysql.connector # 读取YAML配置文件 with open("config.yaml", "r", encoding="utf-8") as file: config = yaml.safe_load(file) # 提取数据库连接信息 db_config = config["database"] cnx = mysql.connector.connect( host=db_config["host"], port=db_config["port"], user=db_config["username"], password=db_config["password"], database=db_config["database"] ) # 提取插入配置 insert_config = config["insert_config"] table = insert_config["table"] fields = insert_config["fields"] # 示例数据 data = [ {"name": "Alice", "age": 25, "email": "alice@example.com"}, {"name": "Bob", "age": 30, "email": "bob@example.com"} ] # 构建SQL插入语句 cursor = cnx.cursor() columns = ", ".join(fields) placeholders = ", ".join(["%s"] * len(fields)) sql = f"INSERT INTO {table} ({columns}) VALUES ({placeholders})" # 批量插入数据 for row in data: values = [row[field] for field in fields] cursor.execute(sql, values) cnx.commit() cursor.close() cnx.close() ``` ### 4. YAML文件的更新与扩展 YAML文件支持动态更新和扩展。例如，可以通过代码动态添加新的字段或修改现有配置。以下是一个示例，展示如何向YAML文件中添加新的字段： ```python # 读取现有配置 with open("config.yaml", "r", encoding="utf-8") as file: config = yaml.safe_load(file) # 添加新字段 config["insert_config"]["fields"].append("phone") # 写回YAML文件 with open("config.yaml", "w", encoding="utf-8") as file: yaml.dump(config, file, allow_unicode=True) ``` 通过这种方式，可以灵活地管理批量插入数据所需的变量信息，并确保配置的可维护性。 ---