<think>我们有一个目标目录路径，需要确保该目录存在，如果不存在则创建它。 使用os.makedirs可以递归创建目录，避免因中间目录不存在而失败。 同时，我们需要注意目录创建时的权限问题（使用默认权限即可）和可能出现的异常（如权限不足）。 下面是一个健壮的创建目录的函数，以及使用示例。 </think> 以下是安全创建结果目录的完整解决方案，包含错误处理和权限管理： ```python import os import argparse import errno import sys import stat def create_result_directory(result_dir, test_name, mode=0o755): """ 创建结果目录（包括所有必要父目录） 参数: result_dir: 结果根目录 test_name: 测试名称子目录 mode: 目录权限（默认0o755） 返回: 创建的完整路径 异常: OSError: 目录创建失败 """ # 安全拼接路径 target_dir = os.path.join(result_dir, test_name) try: # 递归创建目录（如果不存在） os.makedirs(target_dir, mode=mode, exist_ok=True) # 验证目录可写 if not os.access(target_dir, os.W_OK): # 尝试修复权限 os.chmod(target_dir, mode) if not os.access(target_dir, os.W_OK): raise PermissionError(f"目录不可写: {target_dir}") print(f"目录已创建/验证: {target_dir}") return target_dir except OSError as e: # 处理特定错误情况 if e.errno == errno.EEXIST and not os.path.isdir(target_dir): raise NotADirectoryError(f"路径存在但不是目录: {target_dir}") from e elif e.errno == errno.EACCES: raise PermissionError(f"权限不足: {target_dir}") from e else: raise RuntimeError(f"创建目录失败: {target_dir}") from e # 使用示例 if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument('--result_dir', type=str, default='./results', help='结果根目录') parser.add_argument('--test_name', type=str, required=True, help='测试名称') args = parser.parse_args() try: # 创建目录 result_path = create_result_directory(args.result_dir, args.test_name) # 示例：在目录中创建文件 test_file = os.path.join(result_path, "test.txt") with open(test_file, 'w') as f: f.write("目录创建成功！") print(f"测试文件已创建: {test_file}") except Exception as e: print(f"错误: {e}", file=sys.stderr) sys.exit(1) ``` ### 代码关键点解释： 1. **安全目录创建**： ```python os.makedirs(target_dir, mode=0o755, exist_ok=True) ``` - `exist_ok=True`：目录已存在时不报错 - `mode=0o755`：设置合理权限（用户rwx，组rx，其他rx） 2. **权限验证与修复**： ```python if not os.access(target_dir, os.W_OK): os.chmod(target_dir, mode) # 尝试修复权限 if not os.access(target_dir, os.W_OK): raise PermissionError(...) ``` - 确保目录实际可写，避免后续文件操作失败 - 尝试自动修复权限问题 3. **详细的错误处理**： ```python except OSError as e: if e.errno == errno.EEXIST and not os.path.isdir(target_dir): raise NotADirectoryError(...) elif e.errno == errno.EACCES: raise PermissionError(...) ``` - 区分不同错误类型： - 路径已存在但不是目录 - 权限不足 - 其他创建错误 4. **跨平台兼容性**： - 使用 `os.path.join()` 处理路径分隔符差异 - 权限模式使用八进制表示（0o755）兼容POSIX系统 - Windows系统会忽略权限中的执行位 ### 进阶功能扩展： 1. **添加目录所有权检查**（Linux/macOS）： ```python import pwd # 检查当前用户是否拥有目录 if hasattr(os, 'geteuid'): dir_uid = os.stat(target_dir).st_uid if dir_uid != os.geteuid(): print(f"警告: 目录由用户 {pwd.getpwuid(dir_uid).pw_name} 拥有") ``` 2. **自动清理空目录**： ```python def safe_create_dir(path): if os.path.exists(path) and os.path.isdir(path): if not os.listdir(path): # 目录为空 print(f"清空现有目录: {path}") for f in os.listdir(path): os.remove(os.path.join(path, f)) return create_result_directory(path) ``` 3. **带时间戳的目录创建**： ```python from datetime import datetime timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") result_dir = os.path.join(args.result_dir, f"{test_name}_{timestamp}") ``` ### 最佳实践建议： 1. **权限设置**： - 生产环境推荐权限：0o755（目录）、0o644（文件） - 敏感数据使用更严格权限：0o700 2. **路径安全**： - 防止路径遍历攻击：`os.path.abspath(os.path.realpath(path))` - 验证用户输入：`if not test_name.isidentifier(): raise ValueError(...)` 3. **磁盘空间检查**： ```python import shutil total, used, free = shutil.disk_usage(result_dir) if free < 1024**3: # 小于1GB raise IOError("磁盘空间不足") ```