Python中的OS库

在Python中os模块提供了许多与操作系统进行交互的功能，如文件和目录操作、环境变量访问、进程管理等。它是Python标准库的一部分，无需额外安装即可使用

1. 导入os模块

import os

2. 文件和目录操作

2.1 获取当前工作目录

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

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

2.2 更改当前工作目录

使用os.chdir(path)可以更改当前工作目录

os.chdir('/path/to/new/directory')
print("新的工作目录:", os.getcwd())

2.3 列出目录内容

os.listdir(path)函数可以列出指定目录下的所有文件和子目录

contents = os.listdir('.')
print("目录内容:", contents)

2.4 检查路径是否存在

使用os.path.exists(path)可以检查指定路径是否存在

path = '/path/to/check'
if os.path.exists(path):
    print("路径存在")
else:
    print("路径不存在")

2.5 检查是否为文件或目录

os.path.isfile(path)和os.path.isdir(path)分别用于检查路径是否为文件或目录

if os.path.isfile(path):
    print("是文件")
elif os.path.isdir(path):
    print("是目录")
else:
    print("未知类型")

2.6 创建目录

使用os.mkdir(path)创建单级目录，使用os.makedirs(path)创建多级目录

os.mkdir('new_directory')
os.makedirs('parent/child')

2.7 删除目录

使用os.rmdir(path)删除空目录，使用shutil.rmtree(path)删除非空目录

import shutil
os.rmdir('empty_directory')
shutil.rmtree('non_empty_directory')

2.8 重命名文件或目录

使用os.rename(src, dst)重命名文件或目录

os.rename('old_name.txt', 'new_name.txt')

2.9 删除文件

使用os.remove(path)删除文件

os.remove('file_to_delete.txt')

2.10 文件属性

获取文件大小使用os.path.getsize(path)，获取修改时间使用os.path.getmtime(path)

file_size = os.path.getsize('some_file.txt')
modification_time = os.path.getmtime('some_file.txt')
print("文件大小:", file_size, "字节")
print("修改时间:", modification_time)

3. 环境变量

3.1 获取环境变量

使用os.getenv(variable_name)获取指定环境变量的值

value = os.getenv('HOME')
print("HOME环境变量:", value)

3.2 设置环境变量

使用os.environ[‘variable_name’] = value设置环境变量

os.environ['MY_VAR'] = 'my_value'
print("MY_VAR环境变量:", os.getenv('MY_VAR'))

4. 进程管理

4.1 获取当前进程ID

使用os.getpid()获取当前进程的ID

pid = os.getpid()
print("当前进程ID:", pid)

4.2 执行系统命令

使用os.system(command)执行系统命令

os.system('ls -l')

4.3 使用subprocess模块执行命令

推荐使用subprocess模块来执行外部命令，因为它更灵活和安全

import subprocess
result = subprocess.run(['ls', '-l'], capture_output=True, text=True)
print("命令输出:", result.stdout)

5. 操作系统信息

5.1 获取操作系统名称

使用os.name获取操作系统的名称

print("操作系统名称:", os.name)

5.2 获取文件分隔符

使用os.sep获取当前操作系统使用的路径分隔符

print("路径分隔符:", os.sep)

5.3 获取换行符

使用os.linesep获取当前操作系统使用的换行符

print("换行符:", os.linesep)

6. 文件路径操作

6.1 连接路径

使用os.path.join(path, *paths)连接路径部件

path = os.path.join('directory', 'subdirectory', 'file.txt')
print("连接后的路径:", path)

6.2 分割路径

使用os.path.split(path)分割路径为目录和文件名

directory, filename = os.path.split('/path/to/file.txt')
print("目录:", directory)
print("文件名:", filename)

6.3 获取文件扩展名

使用os.path.splitext(path)分割文件名和扩展名

filename, extension = os.path.splitext('file.txt')
print("文件名:", filename)
print("扩展名:", extension)

7. 文件操作示例

import os

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

# 列出目录内容
contents = os.listdir(current_dir)
print("目录内容:", contents)

# 检查路径是否存在
path = 'example_dir'
if not os.path.exists(path):
    os.mkdir(path)
    print("创建目录:", path)
else:
    print("目录已存在:", path)

# 创建文件
file_path = os.path.join(path, 'example.txt')
with open(file_path, 'w') as file:
    file.write("Hello, os module!")

# 检查文件是否存在
if os.path.isfile(file_path):
    print("文件存在:", file_path)
else:
    print("文件不存在:", file_path)

# 获取文件大小
file_size = os.path.getsize(file_path)
print("文件大小:", file_size, "字节")

# 删除文件
os.remove(file_path)
print("删除文件:", file_path)

# 删除目录
os.rmdir(path)
print("删除目录:", path)