操作文件和目录

如果我们要操作文件和目录,可以在命令行下输入系统提供的各种命令来完成。比如dir、cp等命令。

如果要在Python程序中执行这些文件和目录的操作怎么办呢?其实操作系统提供的命令只是简单的调用了操作系统的接口函数,Python内置的os模块也可以直接调用操作系统提供的接口函数。

打开Python交互命令行,我们来看看如何使用os模块的基础命令:
>>>import os

>>>os.name

'nt'

如果是'posix',说明是Linux/Unix/Mac OS X,如果是'nt',则是windows系统。

要获取详细的系统信息,可以调用uname()函数:
uname()函数在windows上不支持,也就是说,os的某些函数跟操作系统是相关的。


环境变量

在操作系统中定义的环境变量,全部包存在os.environ这个变量中,可以直接查看:
>>>os.environ

要获取某个环境变量的值可以调用os.environ.get('key'):
>>>os.environ.get('path')


操作文件和目录

操作文件和目录的函数一部分放在os模块中,一部分放在os.path模块中,这一点要注意下。查看、调用、删除目录可以这样做:
>>>os.path.abspath('.')

'C:\\user\\Administrator'

>>>os.path.join('c:\\user\\Administrator','testdir')

'C:\\user\\Administrator\\testdir'

>>>os.mkdir('C:\\\user\\Administrator\\testdir')

>>>os.rmdir('C:\\user\\Administrator\\testdir')

把两个路径合并成一个,不要直接拼写字符串,而是要通过os.path.join()函数,这样可以正确处理不同操作系统的路径分隔符。在Linux下,os.path.join()返回这样的字符串:
part1/part2

而windows下会返回:
part1\part2

同样的道理,要拆分路径时也不要直接拆分字符串,而要通过os.path.split()函数,这样可以把一个路径拆分为两个部分,后一部分总是最后级别的目录或者文件名:
>>>os.path.split('user\administrator\testdir\file.txt')

('user\administrator\testdir','file.txt')

os.path.splitext()可以让你直接得到文件的扩展名,很多时候非常方便:

>>> os.path.splitext('/path/to/file.txt')
('/path/to/file', '.txt')

这些合并、拆分路径的函数并不要求文件真正存在,它们只对字符串进行操作。

文件操作使用下面的函数。假设当前目录下有一个test.txt的文件:
>>>os.rename('test.txt','test.py')

>>>os.remove('test.py')

但是复制文件在os模块中竟然找不到。原因是复制文件并非由操作系统提供的系统调用。理论上讲,我们通过上节学习到的读写文件可以完成文件的复制,只不过要多谢几行代码。

幸运地是,shutil模块提供了copyfile()函数,你还可以在shutil模块中找到很多实用的函数,它们可以看做是os模块的补充。

最后看看如何通过Python的特性来过滤文件。比如我们要列出当前目录下的所有文件:
>>>[x for x in os.listdir('.') if os.path.isdir(x)]

要列出所有的.py文件也秩序一行代码:
>>>[x for x in os.listdir('.') if os.path.isfile(x) and os.path.splitext(x)[1] =='.py']

是不是非常简洁?


小结:
Python的os模块封装了操作系统的目录和文件操作,要注意这些函数有的在os模块中,有的在os.path中。