Python os 库 5个 基础代码示例

示例 1：列出目录中的所有文件和文件夹

功能描述：列出指定目录中的所有文件和文件夹。

import os

def list_files_in_directory(directory):
    """列出指定目录中的所有文件和文件夹"""
    try:
        with os.scandir(directory) as entries:
            for entry in entries:
                print(entry.name)
    except FileNotFoundError:
        print(f"目录 {directory} 不存在")
    except PermissionError:
        print(f"没有权限访问目录 {directory}")

# 使用示例
list_files_in_directory(".")

示例 2：创建新目录

功能描述：创建一个新目录。

import os

def create_directory(directory):
    """创建一个新目录"""
    try:
        os.makedirs(directory, exist_ok=True)
        print(f"目录 {directory} 创建成功")
    except OSError as e:
        print(f"创建目录 {directory} 失败: {e}")

# 使用示例
create_directory("new_folder")

示例 3：删除文件

功能描述：删除指定文件。

import os

def delete_file(file_path):
    """删除指定文件"""
    try:
        os.remove(file_path)
        print(f"文件 {file_path} 删除成功")
    except FileNotFoundError:
        print(f"文件 {file_path} 不存在")
    except PermissionError:
        print(f"没有权限删除文件 {file_path}")

# 使用示例
delete_file("example.txt")

示例 4：重命名文件或目录

功能描述：重命名指定的文件或目录。

import os

def rename_file_or_directory(old_name, new_name):
    """重命名指定的文件或目录"""
    try:
        os.rename(old_name, new_name)
        print(f"{old_name} 重命名为 {new_name} 成功")
    except FileNotFoundError:
        print(f"{old_name} 不存在")
    except PermissionError:
        print(f"没有权限重命名 {old_name}")

# 使用示例
rename_file_or_directory("old_name.txt", "new_name.txt")

示例 5：获取文件大小

功能描述：获取指定文件的大小。

import os

def get_file_size(file_path):
    """获取指定文件的大小"""
    try:
        size = os.path.getsize(file_path)
        print(f"文件 {file_path} 的大小为 {size} 字节")
    except FileNotFoundError:
        print(f"文件 {file_path} 不存在")
    except PermissionError:
        print(f"没有权限访问文件 {file_path}")

# 使用示例
get_file_size("example.txt")