Python使用os模块操作文件与目录的完整指南

import os

# 获取当前工作目录
current_dir = os.getcwd()
print(f"当前工作目录: {current_dir}")

# 改变工作目录
os.chdir('/path/to/new/directory')
print(f"新工作目录: {os.getcwd()}")

# 列出目录中的所有文件和子目录
contents = os.listdir('.')
print(f"当前目录内容: {contents}")

# 使用 os.scandir() 获取更详细的信息（推荐）
with os.scandir('.') as entries:
    for entry in entries:
        print(f"{entry.name} - 是目录: {entry.is_dir()}")

# 创建单个目录
os.mkdir('new_directory')

# 创建多级目录
os.makedirs('parent/child/grandchild', exist_ok=True)

# 删除空目录
os.rmdir('new_directory')

# 删除多级目录（谨慎使用）
import shutil
shutil.rmtree('parent')  # 需要导入 shutil 模块

# 重命名文件或目录
os.rename('old_name.txt', 'new_name.txt')

# 删除文件
os.remove('file_to_delete.txt')

# 检查文件或目录是否存在
if os.path.exists('some_file.txt'):
    print("文件存在")
else:
    print("文件不存在")

import os.path

# 路径拼接（跨平台安全）
full_path = os.path.join('folder', 'subfolder', 'file.txt')
print(f"完整路径: {full_path}")

# 分解路径
dirname = os.path.dirname(full_path)  # 'folder/subfolder'
basename = os.path.basename(full_path)  # 'file.txt'
split_result = os.path.split(full_path)  # ('folder/subfolder', 'file.txt')

print(f"目录部分: {dirname}")
print(f"文件名部分: {basename}")
print(f"拆分结果: {split_result}")

# 获取绝对路径
abs_path = os.path.abspath('relative/path/file.txt')

# 获取文件大小（字节）
file_size = os.path.getsize('file.txt')
print(f"文件大小: {file_size} 字节")

# 获取文件最后修改时间
mod_time = os.path.getmtime('file.txt')
from datetime import datetime
print(f"最后修改: {datetime.fromtimestamp(mod_time)}")

# 检查路径类型
print(f"是文件: {os.path.isfile('file.txt')}")
print(f"是目录: {os.path.isdir('folder')}")
print(f"是链接: {os.path.islink('link.txt')}")

# 规范化路径（处理 . 和 ..）
norm_path = os.path.normpath('folder/../folder/./file.txt')
print(f"规范化路径: {norm_path}")

# 获取真实路径（解析符号链接）
real_path = os.path.realpath('link_to_file.txt')

# 获取所有环境变量
env_vars = os.environ
print(f"环境变量数量: {len(env_vars)}")

# 获取特定环境变量
home_dir = os.environ.get('HOME')  # Linux/macOS
user_profile = os.environ.get('USERPROFILE')  # Windows
path_var = os.environ.get('PATH')

print(f"家目录: {home_dir or user_profile}")
print(f"PATH变量: {path_var}")

# 设置环境变量（当前进程有效）
os.environ['MY_VARIABLE'] = 'my_value'

# 检查设置是否成功
print(f"MY_VARIABLE: {os.environ.get('MY_VARIABLE')}")

# 执行系统命令（返回退出状态码）
exit_code = os.system('ls -la' if os.name != 'nt' else 'dir')
print(f"命令退出码: {exit_code}")

import subprocess  # 更推荐使用 subprocess 模块

# 获取命令输出
result = subprocess.run(['ls', '-la'], capture_output=True, text=True)
print(f"命令输出:\n{result.stdout}")

# 获取操作系统名称
print(f"操作系统: {os.name}")  # 'posix', 'nt', 'java'

# 更精确的平台检测
import platform
print(f"系统平台: {platform.system()}")  # 'Linux', 'Windows', 'Darwin'
print(f"平台版本: {platform.release()}")

# 路径分隔符常量
print(f"路径分隔符: {os.path.sep}")
print(f"路径分隔符列表: {os.pathsep}")
print(f"行结束符: {repr(os.linesep)}")

# 使用 os.path.join 自动处理分隔符
path = os.path.join('usr', 'local', 'bin')
print(f"跨平台路径: {path}")

# 获取当前进程ID
pid = os.getpid()
print(f"当前进程ID: {pid}")

# 获取父进程ID
ppid = os.getppid()
print(f"父进程ID: {ppid}")

# 获取当前用户ID
uid = os.getuid() if hasattr(os, 'getuid') else None
print(f"用户ID: {uid}")

# 创建子进程（Unix-like 系统）
if hasattr(os, 'fork'):
    pid = os.fork()
    if pid == 0:
        print("这是子进程")
        os._exit(0)
    else:
        print(f"这是父进程，子进程ID: {pid}")

def list_files_recursively(directory):
    """递归列出目录中的所有文件"""
    file_list = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            full_path = os.path.join(root, file)
            file_list.append(full_path)
    return file_list

# 使用示例
all_files = list_files_recursively('.')
print(f"找到 {len(all_files)} 个文件")

def batch_rename_files(directory, old_ext, new_ext):
    """批量修改文件扩展名"""
    for filename in os.listdir(directory):
        if filename.endswith(old_ext):
            old_path = os.path.join(directory, filename)
            new_name = filename.replace(old_ext, new_ext)
            new_path = os.path.join(directory, new_name)
            os.rename(old_path, new_path)
            print(f"重命名: {filename} -> {new_name}")

# 使用示例
# batch_rename_files('./docs', '.txt', '.md')

def create_project_structure(project_name):
    """创建标准的项目目录结构"""
    directories = [
        project_name,
        os.path.join(project_name, 'src'),
        os.path.join(project_name, 'tests'),
        os.path.join(project_name, 'docs'),
        os.path.join(project_name, 'data'),
        os.path.join(project_name, 'logs'),
    ]
    
    for directory in directories:
        os.makedirs(directory, exist_ok=True)
        print(f"创建目录: {directory}")
    
    # 创建基础文件
    files = [
        os.path.join(project_name, 'README.md'),
        os.path.join(project_name, 'requirements.txt'),
        os.path.join(project_name, '.gitignore'),
    ]
    
    for file_path in files:
        with open(file_path, 'w') as f:
            f.write('# ' + os.path.basename(file_path) + '\n')
        print(f"创建文件: {file_path}")

# 使用示例
# create_project_structure('my_project')

import os
import errno

def safe_remove(filepath):
    """安全删除文件，避免异常"""
    try:
        os.remove(filepath)
        print(f"成功删除: {filepath}")
    except OSError as e:
        if e.errno != errno.ENOENT:  # 文件不存在不是错误
            print(f"删除失败: {e}")
        else:
            print(f"文件不存在: {filepath}")

def is_safe_path(base_path, target_path):
    """检查目标路径是否在基础路径内（防止路径遍历攻击）"""
    base = os.path.abspath(base_path)
    target = os.path.abspath(target_path)
    return os.path.commonpath([base, target]) == base

# 使用示例
if is_safe_path('/safe/directory', '/safe/directory/sub/file.txt'):
    print("路径安全")
else:
    print("路径不安全！")