Python中检查文件可写性的方法与最佳实践

import os

def check_file_writable_os_access(file_path):
    """
    使用 os.access 检查文件是否可写。
    """
    if os.path.exists(file_path):
        # 文件存在时，检查是否可写
        if os.access(file_path, os.W_OK):
            print(f"文件 '{file_path}' 存在且可写。")
            return True
        else:
            print(f"文件 '{file_path}' 存在但不可写（权限不足）。")
            return False
    else:
        # 文件不存在时，检查其所在目录是否可写
        # 如果目录可写，则表示可以创建并写入新文件
        parent_dir = os.path.dirname(file_path)
        if not parent_dir: # 如果路径是文件名本身，则默认为当前目录
            parent_dir = os.getcwd()

        if os.access(parent_dir, os.W_OK):
            print(f"文件 '{file_path}' 不存在，但其父目录 '{parent_dir}' 可写，可以创建并写入。")
            return True
        else:
            print(f"文件 '{file_path}' 不存在，且其父目录 '{parent_dir}' 不可写，无法创建。")
            return False

# 示例使用
test_file = "my_test_file.txt"
# 确保文件不存在，以便测试创建新文件的场景
if os.path.exists(test_file):
    os.remove(test_file)

check_file_writable_os_access(test_file)

# 创建一个只读文件用于测试不可写场景 (Windows/Linux权限管理方式不同，此处仅作示意)
# 在Linux/macOS上，可以尝试 os.chmod(test_file_ro, 0o444)
# 在Windows上，权限设置更复杂，此示例可能无法直接模拟
test_file_ro = "read_only_test.txt"
with open(test_file_ro, 'w') as f:
    f.write("This is a test.")
# 假设我们在这里设置了只读权限
# os.chmod(test_file_ro, 0o444) # 仅在类Unix系统有效

# check_file_writable_os_access(test_file_ro) # 再次检查

# 清理
if os.path.exists(test_file_ro):
    os.remove(test_file_ro)

import os

def check_file_writable_robust(file_path):
    """
    尝试打开文件进行写入，并捕获 PermissionError 来判断是否可写。
    这是最健壮的检查方法。
    """
    try:
        # 尝试以写入模式打开文件。
        # 'w' 模式会创建文件（如果不存在）或截断文件（如果存在）。
        # 使用 'x' 模式可以避免覆盖现有文件，如果文件存在则抛出 FileExistsError。
        # 如果只是检查可写性，不希望覆盖，可以考虑 'a' 模式（追加）或 'r+' 模式（读写）。
        # 这里为了演示PermissionError，我们用'w'，但实际应用中可能需要更精细的模式选择。
        with open(file_path, 'w') as fp:
            # 如果能成功打开，说明文件是可写的
            print(f"文件 '{file_path}' 可写。")
            return True
    except PermissionError:
        print(f"文件 '{file_path}' 不可写：权限不足。")
        return False
    except FileNotFoundError:
        # 如果文件路径的某个目录不存在，或者路径本身有问题，可能抛出此错误
        print(f"文件 '{file_path}' 路径无效或目录不存在。")
        return False
    except IOError as e:
        # 捕获其他可能的I/O错误，例如磁盘空间不足等
        print(f"文件 '{file_path}' 写入时发生其他I/O错误: {e}")
        return False
    except Exception as e:
        # 捕获其他未预料的错误
        print(f"文件 '{file_path}' 写入时发生未知错误: {e}")
        return False

# 示例使用
test_file_robust = "robust_test_file.txt"

# 确保文件不存在，以便测试创建新文件的场景
if os.path.exists(test_file_robust):
    os.remove(test_file_robust)

check_file_writable_robust(test_file_robust)

# 尝试在受保护的目录中创建文件 (例如，系统根目录，通常会失败)
# 注意：在某些操作系统或权限配置下，这可能仍然成功，请谨慎测试。
# protected_path = "/root/unwritable_test.txt" # Linux/macOS
# check_file_writable_robust(protected_path)