Python OS模块使用

import os

import os

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

# 改变当前目录
os.chdir("/path/to/new/directory")
print("新目录:", os.getcwd())

# 返回上级目录
os.chdir("..")
print("返回上级目录:", os.getcwd())

# 列出当前目录所有文件和文件夹
items = os.listdir()
print("当前目录内容:")
for item in items:
    print(f"  {item}")

# 列出指定目录内容
desktop_items = os.listdir("/Users/username/Desktop")
print("桌面文件:", desktop_items)

# 创建单个目录
os.mkdir("new_folder")

# 创建多级目录（如果父目录不存在也会创建）
os.makedirs("parent/child/grandchild")

# 创建目录时设置权限（Linux/macOS）
os.mkdir("private_folder", 0o755)  # 读写执行权限

# 删除空目录
os.rmdir("empty_folder")

# 递归删除目录（包括所有子目录和文件）
import shutil
shutil.rmtree("folder_with_contents")

# 删除多级空目录
os.removedirs("level1/level2/level3")  # 如果都是空目录，会全部删除

# 创建空文件
with open("new_file.txt", "w") as f:
    pass

# 重命名文件
os.rename("old_name.txt", "new_name.txt")

# 重命名目录
os.rename("old_folder", "new_folder")

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

# 检查文件是否存在后再删除
if os.path.exists("temp_file.txt"):
    os.remove("temp_file.txt")
    print("文件已删除")
else:
    print("文件不存在")

path = "some_path"

# 检查是否是文件
if os.path.isfile(path):
    print(f"{path} 是一个文件")

# 检查是否是目录
if os.path.isdir(path):
    print(f"{path} 是一个目录")

# 检查路径是否存在
if os.path.exists(path):
    print(f"{path} 存在")
else:
    print(f"{path} 不存在")

# 检查是否是链接
if os.path.islink(path):
    print(f"{path} 是一个符号链接")

import time

file_path = "example.txt"

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

# 获取文件修改时间
mod_time = os.path.getmtime(file_path)
readable_time = time.ctime(mod_time)
print(f"最后修改: {readable_time}")

# 获取文件创建时间（Windows）
create_time = os.path.getctime(file_path)
print(f"创建时间: {time.ctime(create_time)}")

# 获取文件访问时间
access_time = os.path.getatime(file_path)
print(f"最后访问: {time.ctime(access_time)}")

import os

# 路径拼接（推荐使用，避免手动拼接错误）
full_path = os.path.join("folder", "subfolder", "file.txt")
print("完整路径:", full_path)

# 路径分割
directory, filename = os.path.split("/home/user/document.txt")
print(f"目录: {directory}, 文件名: {filename}")

# 文件名和扩展名分割
name, extension = os.path.splitext("image.jpg")
print(f"文件名: {name}, 扩展名: {extension}")

# 获取绝对路径
abs_path = os.path.abspath("relative/path/file.txt")
print("绝对路径:", abs_path)

# 获取路径的目录部分
dir_name = os.path.dirname("/home/user/file.txt")
print("目录名:", dir_name)

# 获取路径的基本名称（文件名）
base_name = os.path.basename("/home/user/file.txt")
print("基本名称:", base_name)

def list_all_files(start_path):
    """列出目录下所有文件"""
    print(f"扫描目录: {start_path}")
    
    for root, dirs, files in os.walk(start_path):
        # 当前目录层级
        level = root.replace(start_path, '').count(os.sep)
        indent = '  ' * level
        print(f"{indent}[{os.path.basename(root)}/]")
        
        # 列出文件
        sub_indent = '  ' * (level + 1)
        for file in files:
            print(f"{sub_indent}{file}")

# 使用示例
list_all_files(".")

def find_files(extension, search_path="."):
    """查找指定扩展名的文件"""
    found_files = []
    
    for root, dirs, files in os.walk(search_path):
        for file in files:
            if file.endswith(extension):
                full_path = os.path.join(root, file)
                found_files.append(full_path)
    
    return found_files

# 查找所有Python文件
python_files = find_files(".py")
print("找到的Python文件:")
for file in python_files:
    print(f"  {file}")

# 查找所有图片文件
image_files = find_files((".jpg", ".png", ".gif"))
print("找到的图片文件:")
for file in image_files:
    print(f"  {file}")

import os

# 获取环境变量
home_dir = os.getenv("HOME")  # Linux/macOS
user_profile = os.getenv("USERPROFILE")  # Windows
python_path = os.getenv("PYTHONPATH")

print(f"Home目录: {home_dir}")
print(f"用户配置: {user_profile}")
print(f"Python路径: {python_path}")

# 获取所有环境变量
print("所有环境变量:")
for key, value in os.environ.items():
    print(f"  {key} = {value}")

# 设置环境变量（当前进程有效）
os.environ["MY_VARIABLE"] = "my_value"
print(f"自定义变量: {os.getenv('MY_VARIABLE')}")

import os
import shutil
from datetime import datetime

def backup_files(source_dir, backup_dir):
    """备份目录中的文件"""
    if not os.path.exists(source_dir):
        print("源目录不存在")
        return
    
    # 创建备份目录
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    backup_path = os.path.join(backup_dir, f"backup_{timestamp}")
    os.makedirs(backup_path, exist_ok=True)
    
    # 复制文件
    copied_count = 0
    for item in os.listdir(source_dir):
        source_path = os.path.join(source_dir, item)
        backup_item_path = os.path.join(backup_path, item)
        
        if os.path.isfile(source_path):
            shutil.copy2(source_path, backup_item_path)
            copied_count += 1
            print(f"已备份: {item}")
    
    print(f"备份完成！共备份 {copied_count} 个文件到 {backup_path}")

# 使用示例
backup_files("./documents", "./backups")

def organize_files(directory):
    """按扩展名整理文件"""
    if not os.path.exists(directory):
        print("目录不存在")
        return
    
    # 创建分类文件夹
    categories = {
        'Images': ['.jpg', '.jpeg', '.png', '.gif', '.bmp'],
        'Documents': ['.pdf', '.doc', '.docx', '.txt', '.xlsx'],
        'Code': ['.py', '.js', '.html', '.css', '.java'],
        'Archives': ['.zip', '.rar', '.7z', '.tar']
    }
    
    for category in categories:
        category_path = os.path.join(directory, category)
        os.makedirs(category_path, exist_ok=True)
    
    # 移动文件
    moved_count = 0
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        
        if os.path.isfile(file_path):
            _, ext = os.path.splitext(filename)
            ext = ext.lower()
            
            for category, extensions in categories.items():
                if ext in extensions:
                    dest_path = os.path.join(directory, category, filename)
                    os.rename(file_path, dest_path)
                    moved_count += 1
                    print(f"已移动: {filename} -> {category}/")
                    break
    
    print(f"整理完成！共移动 {moved_count} 个文件")

# 使用示例
organize_files("./downloads")

def check_disk_usage(path="."):
    """检查磁盘使用情况"""
    import shutil
    
    total, used, free = shutil.disk_usage(path)
    
    print(f"路径: {os.path.abspath(path)}")
    print(f"总空间: {total // (2**30)} GB")
    print(f"已使用: {used // (2**30)} GB")
    print(f"剩余空间: {free // (2**30)} GB")
    print(f"使用率: {used/total*100:.1f}%")

def find_large_files(directory, min_size_mb=10):
    """查找大文件"""
    large_files = []
    
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            try:
                file_size = os.path.getsize(file_path)
                if file_size > min_size_mb * 1024 * 1024:
                    large_files.append((file_path, file_size))
            except OSError:
                continue  # 跳过无法访问的文件
    
    # 按文件大小排序
    large_files.sort(key=lambda x: x[1], reverse=True)
    
    print(f"大于 {min_size_mb}MB 的文件:")
    for file_path, size in large_files[:10]:  # 显示前10个
        size_mb = size / (1024 * 1024)
        print(f"  {size_mb:.1f}MB - {file_path}")

# 使用示例
check_disk_usage()
find_large_files(".", 5)

# 安全的文件操作
try:
    os.remove("important_file.txt")
except FileNotFoundError:
    print("文件不存在")
except PermissionError:
    print("没有删除权限")
except Exception as e:
    print(f"发生错误: {e}")