Python之OS编程

OS模块是Python标准库中的一个重要模块，它提供了与操作系统直接交互的功能。本文将从多个方面对Python之OS编程进行详细阐述。

一、OS模块介绍

OS模块是Python的一个内置模块，它提供了一系列与操作系统交互的函数。

import os

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

# 列出目录下的文件和文件夹
files = os.listdir(current_dir)
print("目录下的文件和文件夹：", files)

# 创建目录
new_dir = os.path.join(current_dir, "new_folder")
os.mkdir(new_dir)
print("创建目录：", new_dir)

# 运行系统命令
os.system("ls -l")

二、文件操作

OS模块提供了丰富的文件操作方法，可以方便地对文件进行读写、复制、移动等操作。

# 检查文件是否存在
file_path = os.path.join(current_dir, "test.txt")
if os.path.exists(file_path):
    print("文件存在")
else:
    print("文件不存在")

# 读取文件内容
with open(file_path, "r") as f:
    content = f.read()
    print("文件内容：", content)

# 写入文件
new_file_path = os.path.join(current_dir, "new_file.txt")
with open(new_file_path, "w") as f:
    f.write(content)
    print("写入文件成功")

# 复制文件
new_file_path2 = os.path.join(current_dir, "new_file2.txt")
os.popen(f"cp {new_file_path} {new_file_path2}")
print("复制文件成功")

三、目录操作

除了文件操作，OS模块也支持对目录进行创建、删除和遍历等操作。

# 创建目录
new_dir2 = os.path.join(current_dir, "new_folder2")
os.mkdir(new_dir2)
print("创建目录：", new_dir2)

# 删除目录
os.rmdir(new_dir2)
print("删除目录：", new_dir2)

# 遍历目录
for root, dirs, files in os.walk(current_dir):
    print(f"目录：{root}")
    print(f"子目录：{dirs}")
    print(f"文件：{files}")

四、环境变量

OS模块还提供了操作环境变量的方法，可以获取、设置和删除环境变量。

# 获取环境变量
path = os.getenv("PATH")
print("环境变量PATH：", path)

# 设置环境变量
os.environ["MY_VAR"] = "Hello World"
print("设置环境变量成功")

# 删除环境变量
del os.environ["MY_VAR"]
print("删除环境变量成功")

通过OS模块，我们可以方便地进行文件操作、目录操作以及环境变量的管理。以上仅是OS模块的一部分功能，更多功能请参考官方文档。