操作文件和目录

    操作文件和目录的函数一部分放在 os 模块中,一部分放在 os.path 模块中,这一点要注意一下。查看、创建和删除目录可以这么调用:

    # 查看当前目录的绝对路径:

    >>> os.path.abspath('.')

    'Usersmichael'

    # 在某个目录下创建一个新目录,首先把新目录的完整路径表示出来:

    >>> os.path.join('Usersmichael', 'testdir')

    'Usersmichael/testdir'

    # 然后创建一个目录:

    >>> os.mkdir('Usersmichael/testdir')

    # 删掉一个目录:

    >>> os.rmdir('Usersmichael/testdir')

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

    part-1/part-2

    而Windows下会返回这样的字符串:

    part-1\part-2

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

    >>> os.path.split('Usersmichael/testdir/file.txt')

    ('Usersmichael/testdir', 'file.txt')

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

    >>> os.path.splitext('pathto/file.txt')

    ('pathto/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)]

    ['.lein', '.local', '.m2', '.npm', '.ssh', '.Trash', '.vim', 'Applications', 'Desktop', …]

    要列出所有的 .py 文件,也只需一行代码:

    >>> [x for x in os.listdir('.') if os.path.isfile(x) and os.path.splitext(x)[1]=='.py']

    ['apis.py', 'config.py', 'models.py', 'pymonitor.py', 'test_db.py', 'urls.py', 'wsgiapp.py']

    是不是非常简洁?