python标准库os库的函数介绍

1. Python标准库os库介绍

Python标准库中的os库是一个提供了多种操作系统服务的Python标准库。在写Python程序时，os库可以用来处理与操作系统相关的任务，如文件操作、目录操作、进程管理等。在本文中，我们将对os库中的常用函数进行介绍。

2. 文件操作函数

2.1 os.getcwd()

os.getcwd()函数返回当前工作目录的绝对路径。

import os

current_dir = os.getcwd()
print(f"Current directory: {current_dir}")

输出结果：

Current directory: /Users/user

2.2 os.listdir(path='.')

os.listdir()函数返回指定目录下的文件和目录名，如果未提供路径，则返回当前目录下的内容。

import os

files = os.listdir()
print(f"Content of current directory: {files}")

输出结果：

Content of current directory: ['file1.txt', 'file2.txt', 'dir1', 'dir2']

2.3 os.makedirs(name, mode=0o777, exist_ok=False)

os.makedirs()函数递归创建目录，如果目录已存在则抛出异常。

import os

dir_name = 'new_dir/subdir'
os.makedirs(dir_name)
files = os.listdir()
print(f"Content of current directory: {files}")

输出结果：

Content of current directory: ['file1.txt', 'file2.txt', 'dir1', 'dir2', 'new_dir']

2.4 os.rename(src, dst)

os.rename()函数用于重命名文件或目录，将src修改为dst。

import os

os.rename('file1.txt', 'new_file.txt')
files = os.listdir()
print(f"Content of current directory: {files}")

输出结果：

Content of current directory: ['new_file.txt', 'file2.txt', 'dir1', 'dir2', 'new_dir']

3. 进程操作函数

3.1 os.getpid()

os.getpid()函数返回当前进程的PID(Process ID)。

import os

pid = os.getpid()
print(f"Current PID: {pid}")

输出结果：

Current PID: 12345

3.2 os.kill(pid, signal.SIGTERM)

os.kill()函数向指定PID发送指定的信号(signal)。

import os

import signal
pid = os.getpid()
os.kill(pid, signal.SIGTERM)

该代码会结束当前运行的Python程序。

4. 环境变量操作函数

4.1 os.environ

os.environ返回一个包含系统环境变量的字典。

import os

env_vars = os.environ
print(f"Environment variables: {env_vars}")

输出结果的一部分：

Environment variables: {'HOME': '/home/user', 'PATH': '/usr/local/bin:/usr/bin:/bin', ...}

4.2 os.getenv(key, default=None)

os.getenv()函数根据指定的键(key)获取环境变量的值(value)，如果键不存在，返回默认值(default)。

import os

path = os.getenv('PATH')
print(f"Environment variable PATH: {path}")

输出结果：

Environment variable PATH: /usr/local/bin:/usr/bin:/bin

5. 测试文件是否存在和判断文件类型函数

5.1 os.path.exists(path)

os.path.exists()函数用于测试指定的路径(path)是否存在。

import os

exists = os.path.exists('file1.txt')
if exists:
    print("file1.txt exists")

输出结果：

file1.txt exists

5.2 os.path.isfile(path)

os.path.isfile()函数用于测试指定的路径(path)是否为一个文件。

import os

is_file = os.path.isfile('file1.txt')
if is_file:
    print("file1.txt is a file")

输出结果：

file1.txt is a file

5.3 os.path.isdir(path)

os.path.isdir()函数用于测试指定的路径(path)是否为一个目录。

import os

is_dir = os.path.isdir('dir1')
if is_dir:
    print("dir1 is a directory")

输出结果：

dir1 is a directory

6. 总结

以上介绍了os库中的常用函数，包括文件操作、进程操作、环境变量操作、测试文件是否存在和判断文件类型等。通过使用这些函数，可以方便地处理Python程序中与操作系统相关的任务。