Python os模块的强大功能与应用

大家好，我是涛哥，本文内容来自涛哥聊Python ，转载请标原创。

在Python中，os模块提供了与操作系统进行交互的功能，允许我们执行各种操作系统任务，如文件和目录操作、环境变量访问、进程管理等。os模块是标准库的一部分，无需额外安装。本文将详细介绍os模块的功能，并提供相应的示例代码，帮助全面掌握这一强大工具。

os 模块概述

os模块提供了一组函数和变量，使用Python代码执行操作系统级别的操作。

通过os模块，可以完成以下任务：

文件和目录操作（创建、删除、移动、重命名等）
环境变量访问
执行系统命令
进程管理
获取系统信息

文件和目录操作

获取当前工作目录

使用os.getcwd()函数可以获取当前工作目录。

import os

current_directory = os.getcwd()
print(f"当前工作目录: {current_directory}")

改变工作目录

使用os.chdir()函数可以改变当前工作目录。

import os

new_directory = "/path/to/new/directory"
os.chdir(new_directory)
print(f"已更改工作目录: {os.getcwd()}")

创建目录

使用os.mkdir()函数可以创建一个新目录。

import os

new_directory = "new_folder"
os.mkdir(new_directory)
print(f"已创建目录: {new_directory}")

使用os.makedirs()可以创建多级目录。

import os

new_directories = "new_folder/sub_folder"
os.makedirs(new_directories)
print(f"已创建多级目录: {new_directories}")

删除目录

使用os.rmdir()可以删除一个空目录。

import os

directory_to_remove = "new_folder/sub_folder"
os.rmdir(directory_to_remove)
print(f"已删除目录: {directory_to_remove}")

使用os.removedirs()可以删除多级空目录。

import os

directories_to_remove = "new_folder/sub_folder"
os.removedirs(directories_to_remove)
print(f"已删除多级目录: {directories_to_remove}")

列出目录内容

使用os.listdir()可以列出指定目录中的内容。

import os

directory = "."
contents = os.listdir(directory)
print(f"{directory} 目录内容:")
for item in contents:
    print(item)

检查路径

使用os.path模块可以检查路径是否存在以及路径的类型（文件或目录）。

import os

path = "example.txt"

if os.path.exists(path):
    if os.path.isfile(path):
        print(f"{path} 是一个文件")
    elif os.path.isdir(path):
        print(f"{path} 是一个目录")
else:
    print(f"{path} 不存在")

重命名和移动文件

使用os.rename()可以重命名或移动文件和目录。

import os

source = "example.txt"
destination = "renamed_example.txt"
os.rename(source, destination)
print(f"已将 {source} 重命名为 {destination}")

删除文件

使用os.remove()可以删除文件。

import os

file_to_remove = "renamed_example.txt"
os.remove(file_to_remove)
print(f"已删除文件: {file_to_remove}")

环境变量访问

获取环境变量

使用os.environ可以访问系统环境变量。

import os

path = os.environ.get('PATH')
print(f"系统 PATH 环境变量: {path}")

设置环境变量

使用os.environ可以设置系统环境变量。

import os

os.environ['MY_VAR'] = '12345'
print(f"已设置环境变量 MY_VAR: {os.environ.get('MY_VAR')}")

执行系统命令

使用os.system()可以执行系统命令。

import os

os.system('echo Hello, World!')

进程管理

获取当前进程ID

使用os.getpid()可以获取当前进程ID。

import os

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

获取父进程ID

使用os.getppid()可以获取父进程ID。

import os

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

创建子进程

使用os.fork()可以创建子进程（仅适用于类Unix系统）。

import os

pid = os.fork()
if pid == 0:
    print("这是子进程")
else:
    print(f"这是父进程，子进程ID: {pid}")

获取系统信息

获取操作系统名称

使用os.name可以获取操作系统名称。

import os

os_name = os.name
print(f"操作系统名称: {os_name}")

获取平台名称

使用os.uname()可以获取系统的详细信息（仅适用于类Unix系统）。

import os

system_info = os.uname()
print(f"系统信息: {system_info}")

综合示例

以下是一个综合示例，展示如何使用os模块进行一系列操作。

import os

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

# 创建新目录
new_directory = "test_folder"
os.mkdir(new_directory)
print(f"已创建目录: {new_directory}")

# 改变工作目录
os.chdir(new_directory)
print(f"已更改工作目录: {os.getcwd()}")

# 创建新文件并写入内容
with open("test_file.txt", "w") as file:
    file.write("Hello, World!")
print("已创建文件: test_file.txt 并写入内容")

# 列出当前目录内容
contents = os.listdir(".")
print(f"当前目录内容: {contents}")

# 重命名文件
os.rename("test_file.txt", "renamed_test_file.txt")
print("已将文件重命名为: renamed_test_file.txt")

# 删除文件
os.remove("renamed_test_file.txt")
print("已删除文件: renamed_test_file.txt")

# 返回上一级目录并删除目录
os.chdir("..")
os.rmdir(new_directory)
print(f"已删除目录: {new_directory}")

总结

本文详细介绍了Python的os模块的各种功能，包括文件和目录操作、环境变量访问、执行系统命令、进程管理以及获取系统信息。通过具体的示例代码，展示了如何使用这些功能来进行操作系统级别的任务。无论是获取当前工作目录、创建和删除目录、重命名和移动文件，还是访问环境变量、执行系统命令、管理进程，os模块都提供了简洁而强大的接口。掌握os模块的使用，可以大大提高在Python编程中的效率和灵活性，在处理系统相关任务时游刃有余。