Python总结----OS操作及常见示例（五）

1、遍历目录并统计文件大小

import os

def get_directory_size(directory):
    total_size = 0
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            total_size += os.path.getsize(file_path)
    return total_size

size = get_directory_size(".")
print("目录总大小 (字节):", size)

2、使用 os 模块或 pathlib 模块遍历目录，并将文件和目录分别存储到不同的列表中

import os

def list_files_and_dirs(root_dir):
    files = []
    dirs = []
    
    for root, dir_names, file_names in os.walk(root_dir):
        for dir_name in dir_names:
            dirs.append(os.path.join(root, dir_name))  # 存储完整路径
        for file_name in file_names:
            files.append(os.path.join(root, file_name))  # 存储完整路径
    
    return files, dirs

# 示例用法
root_directory = "."  # 当前目录
files, directories = list_files_and_dirs(root_directory)

print("文件列表:")
for file in files:
    print(file)

print("\n目录列表:")
for dir in directories:
    print(dir)

3、将指定文件下所有文件合并到一个新文件中，同时排除文件名包含生产二字的文件名

import os

def merge_files_in_directory(input_dir, output_file):
    """
    读取指定目录下的所有文件，并将内容合并到一个新文件中（排除文件名包含“生产”的文件）
    
    Args:
        input_dir (str): 要读取的目录路径
        output_file (str): 合并后的输出文件路径
    """
    # 检查目录是否存在
    if not os.path.isdir(input_dir):
        print(f"错误：目录 '{input_dir}' 不存在！")
        return

    # 创建输出文件（如果不存在）
    with open(output_file, 'w', encoding='utf-8') as outfile:
        # 遍历目录下的所有文件
        for filename in os.listdir(input_dir):
            file_path = os.path.join(input_dir, filename)
            
            # 跳过子目录，只处理文件，并排除文件名包含“生产”的文件
            if os.path.isfile(file_path) and "生产" not in filename:
                try:
                    # 读取文件内容并写入输出文件
                    with open(file_path, 'r', encoding='utf-8') as infile:
                        outfile.write(f"=== 文件: {filename} ===\n")  # 可选：添加文件名作为分隔
                        outfile.write(infile.read())
                        outfile.write("\n\n")  # 可选：添加换行分隔
                    print(f"已合并: {filename}")
                except UnicodeDecodeError:
                    print(f"警告: {filename} 不是文本文件，已跳过")
                except Exception as e:
                    print(f"错误: 处理 {filename} 时出错 - {e}")
            else:
                if "生产" in filename:
                    print(f"跳过（包含“生产”）：{filename}")

    print(f"\n所有文件已合并到: {output_file}")

# 示例用法
input_directory = "./input_files"  # 替换为你的目录路径
output_file_path = "./merged_output.txt"  # 合并后的文件路径

merge_files_in_directory(input_directory, output_file_path)

 4、os.path.abspath(__file__)

获取当前脚本文件的绝对路径，无论脚本是如何被运行的。如果脚本路径是 ./project/script.py，且当前工作目录是 /home/user，结果可能是 /home/user/project/script.py。

用途：当脚本需要读取同目录下的配置文件或数据文件时，可以用它构建文件的绝对路径：

import os
config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "config.json")

5、os.path.dirname(os.path.abspath(__file__))

获取当前脚本所在的目录的绝对路径，如果脚本路径是 /home/user/project/script.py，结果会是 /home/user/project。

用途：当脚本需要读取同目录下的其他文件（如配置文件、数据文件）时，可以用它构建文件的绝对路径：

import os

# 获取当前脚本所在目录
script_dir = os.path.dirname(os.path.abspath(__file__))

# 拼接同级目录下的文件路径
config_path = os.path.join(script_dir, "config.json")
data_path = os.path.join(script_dir, "data", "input.csv")

6、os.path.dirname(os.path.dirname(os.path.realpath(__file__)))

获取当前脚本所在目录的父目录的父目录（上两级目录）的绝对路径。如果脚本路径是 /home/user/project/src/script.py，结果会是 /home/user

用途：当需要访问脚本目录的上级或上上级目录中的文件时（如配置文件、共享资源）：

import os

# 获取上两级目录（例如脚本在 /project/src/utils/script.py）
project_root = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))

# 拼接路径（如访问 /project/config/settings.json）
config_path = os.path.join(project_root, "config", "settings.json")