Python中-os库的用法超详细介绍

import os

def list_files(startpath):
    for root, dirs, files in os.walk(startpath):
        level = root.replace(startpath, '').count(os.sep)
        indent = ' ' * 4 * level
        print(f"{indent}{os.path.basename(root)}/")
        subindent = ' ' * 4 * (level + 1)
        for f in files:
            print(f"{subindent}{f}")

list_files("/tmp")

os.walk(top, topdown=True, onerror=None, followlinks=False)

root/
├── dir1/
│   ├── file1.txt
│   └── subdir/
└── dir2/
    └── file2.log

# 第一次迭代
('root', ['dir1', 'dir2'], [])  
# 第二次迭代（进入 dir1）
('root/dir1', ['subdir'], ['file1.txt'])  
# 第三次迭代（进入 subdir）
('root/dir1/subdir', [], [])  
# 第四次迭代（进入 dir2）
('root/dir2', [], ['file2.log'])

import os

for root, dirs, files in os.walk("data"):
    for file in files:
        full_path = os.path.join(root, file)  # 拼接完整路径[3,6](@ref)
        print(f"发现文件：{full_path}")

folder = "docs"
for filename in os.listdir(folder):
    if filename.endswith(".txt"):
        new_name = filename.replace(" ", "_")
        os.rename(
            os.path.join(folder, filename),
            os.path.join(folder, new_name)
        )

import tempfile

def create_temp_dir():
    temp_dir = tempfile.mkdtemp()
    try:
        print(f"临时目录已创建: {temp_dir}")
        # 执行操作...
    finally:
        os.rmdir(temp_dir)
        print("临时目录已删除")

# 错误方式
path = "dir" + "/" + "file.txt"  # 在Windows下失效

# 正确方式
path = os.path.join("dir", "file.txt")

try:
    os.remove("non_exist.txt")
except FileNotFoundError as e:
    print(f"文件不存在: {e}")

# 危险！用户输入可能包含恶意命令
user_input = "some; rm -rf /"
os.system(f"echo {user_input}")

# 安全方式
import subprocess
subprocess.run(["echo", user_input], check=True)

if os.name == 'posix':
    print("运行在 Unix-like 系统")
elif os.name == 'nt':
    print("运行在 Windows 系统")