在 Python 中，如果你需要读取 YAML 格式的配置文件，通常会使用 `PyYAML` 库。为了提高代码的可维护性和复用性，我们可以封装一个 YAML 配置文件访问类。 --- ### ✅ 封装的 YAML 配置文件访问类 ```python import yaml import os class YamlConfigReader: def __init__(self, config_file='config.yaml'): if not os.path.exists(config_file): raise FileNotFoundError(f"配置文件 {config_file} 不存在") with open(config_file, 'r', encoding='utf-8') as f: self.config = yaml.safe_load(f) # 使用 safe_load 防止执行任意代码 def get(self, *keys): """根据键路径获取配置值""" data = self.config for key in keys: if isinstance(data, dict) and key in data: data = data[key] else: raise KeyError(f"配置路径 {'.'.join(keys)} 不存在") return data def get_int(self, *keys): """获取整型值""" value = self.get(*keys) return int(value) def get_float(self, *keys): """获取浮点值""" value = self.get(*keys) return float(value) def get_boolean(self, *keys): """获取布尔值""" value = self.get(*keys) if isinstance(value, bool): return value elif isinstance(value, str): return value.lower() in ('true', 'yes', '1') else: raise ValueError(f"无法将 {value} 转换为布尔值") def get_dict(self, *keys): """获取字典值""" value = self.get(*keys) if not isinstance(value, dict): raise ValueError(f"配置路径 {'.'.join(keys)} 不是字典类型") return value def get_list(self, *keys): """获取列表值""" value = self.get(*keys) if not isinstance(value, list): raise ValueError(f"配置路径 {'.'.join(keys)} 不是列表类型") return value # 示例使用 if __name__ == '__main__': config = YamlConfigReader('config.yaml') db_host = config.get('database', 'host') db_port = config.get_int('database', 'port') debug_mode = config.get_boolean('app', 'debug') print(f"数据库地址: {db_host}") print(f"数据库端口: {db_port}") print(f"调试模式: {debug_mode}") ``` --- ### 📄 示例 `config.yaml` 文件内容： ```yaml database: host: localhost port: 3306 user: root password: 123456 app: name: MyApp version: 1.0.0 debug: true ``` --- ### 🧠 解释： 1. **`__init__` 方法**：加载 YAML 文件内容到 `self.config` 中，使用 `yaml.safe_load()` 是为了防止执行任意代码（更安全）。 2. **`get(*keys)`**：支持嵌套结构的访问，例如 `get('database', 'host')`。 3. **类型转换方法**： - `get_int()`：将值转换为整数。 - `get_float()`：将值转换为浮点数。 - `get_boolean()`：支持字符串形式的布尔值。 - `get_dict()` 和 `get_list()`：用于确保返回值类型正确。 4. **示例使用**：展示了如何通过封装类访问 YAML 配置项。 ---