6.5 shutil———高级文件操做 高级文件操做,如复制和设置文件权限 6.5.1 复制文件 copyfile()将源的内容复制给目标,若是没有权限写目标文件则产生IoError from shutil import * from glob import glob print 'BEFORE:', glob('huanhuan.*') copyfile('huanhuan.txt', 'huanhuan.txt.copy') print 'AFTER:', glob('huanhuan.*') 这个函数会打开输入文件进行读写,而不论其类型,因此某些特殊文件不能够用copyfile()复制为新的特殊文件。 >>> ================================ RESTART ================================ >>> BEFORE: ['huanhuan.txt'] AFTER: ['huanhuan.txt', 'huanhuan.txt.copy'] copyfile()实际是使用了底层函数copyfileobj()。copyfile()的参数是文件名,copyfileobj()的参数是打开的文件句柄。第三个参数可选,用于读入块的缓冲区长度。 from shutil import * import os from StringIO import StringIO import sys class VerboseStringIO(StringIO): def read(self, n=-1): next = StringIO.read(self, n) print 'read(%d) bytes' % n return next lorem_ipsum = '''This makes the dependency explicit, limits the scope to the current file and provides faster access to the bit.* functions, too. It's good programming practice not to rely on the global variable bit being set (assuming some other part of your application has already loaded the module). The require function ensures the module is only loaded once, in any case.''' print 'Defalut:' input = VerboseStringIO(lorem_ipsum) output = StringIO() copyfileobj(input, output) print print 'All at once:' input = VerboseStringIO(lorem_ipsum) output = StringIO() copyfileobj(input, output, -1) print print 'Blocks of 256:' input = VerboseStringIO(lorem_ipsum) output = StringIO() copyfileobj(input, output, 256) 默认行为是使用大数据块读取。使用-1会一次性读取全部输入,或者使用其余正数能够设置特定块的大小。 >>> ================================ RESTART ================================ >>> Defalut: read(16384) bytes read(16384) bytes All at once: read(-1) bytes read(-1) bytes Blocks of 256: read(256) bytes read(256) bytes read(256) bytes 相似于UNIX命令行工具cp,copy()函数会用一样的方式解释输出名。若是指定的目标指示一个目录而不是一个文件,会使用源文件的基名在该目录中建立一个新文件。 from shutil import * import os dir = os.getcwd() if not os.path.exists('%s\\example' % dir): os.mkdir('%s\\example' % dir) print 'BEFORE:', os.listdir('example') copy('huanhuan.txt', 'example') print 'AFTER:', os.listdir('example') >>> ================================ RESTART ================================ >>> BEFORE: [] AFTER: ['huanhuan.txt'] copy2()工做相似copy(),不过复制到新文件的元数据会包含访问和修改时间。 from shutil import * import os import time dir = os.getcwd() if not os.path.exists('%s\\example' % dir): os.mkdir('%s\\example' % dir) def show_file_info(filename): stat_info = os.stat(filename) print '\tMode :', stat_info.st_mode print '\tCreated :', time.ctime(stat_info.st_ctime) print '\tAccessed:', time.ctime(stat_info.st_atime) print '\tModified:', time.ctime(stat_info.st_mtime) print 'SOURCE:' show_file_info('huanhuan.txt') copy2('huanhuan.txt', 'example') print 'DEST:' show_file_info('%s\\example\\huanhuan.txt' % dir) 文件特性和原文件彻底相同。 >>> ================================ RESTART ================================ >>> SOURCE: Mode : 33206 Created : Thu Feb 13 17:42:46 2014 Accessed: Thu Feb 13 17:42:46 2014 Modified: Thu Feb 13 17:42:46 2014 DEST: Mode : 33206 Created : Thu Feb 13 18:29:14 2014 Accessed: Thu Feb 13 17:42:46 2014 Modified: Thu Feb 13 17:42:46 2014 6.5.2 复制文件元数据 在UNIX建立一个新文件,会根据当前用户的umask接授权限。要把权限从一个文件复制到另外一个文件,可使用copymode()。 from shutil import * import os from commands import * with open('file_to_change.txt', 'wt') as f: f.write('i love you') os.chmod('file_to_change.txt', 0444) print 'BEFORE:' print getstatus('file_to_change.txt') copymode('shutil_copymode.py', 'file_to_change.txt') print 'AFTER:' print getstatus('file_to_change.txt') 要复制其余元数据,可使用copystat()。 from shutil import * import os import time def show_file_info(filename): stat_info = os.stat(filename) print '\tMode :', stat_info.st_mode print '\tCreated :', time.ctime(stat_info.st_ctime) print '\tAccessed :', time.ctime(stat_info.st_atime) print '\tModified :', time.ctime(stat_info.st_mtime) with open('file_to_change.txt', 'wt') as f: f.write('i love you') os.chmod('file_to_Change.txt', 0444) print 'BEFORE:' show_file_info('file_to_Change.txt') copystat('shutil_copystat.py', 'file_to_Change.txt') print 'AFTER:' show_file_info('file_to_Change.txt') 使用copystat()只会复制与文件关联的权限和日期。 6.5.3 处理目录树 shutil包含三个函数处理目录树。要把一个目录从一个位置复制到另外一个位置,使用copytree()。这会递归遍历源目录树,将文件复制到目标。 copytree()能够将当前这个实现看成起点,在使用前要让它更健壮,能够增长一些特性,如进度条。 from shutil import * from commands import * print 'BEFORE:' print getoutput('ls -rlast /tmp/example') copytree('../shutil', '/tmp/example') print '\nAFTER:' print getoutput('ls -rlast /tmp/example') symlinks参数控制着符号连接做为连接复制仍是文件复制。默认将内容复制到新文件,若是选项为true,会在目标中建立新的符号连接。 要删除一个目录及其内容,可使用rmtree()。 from shutil import * from commands import * print 'BEFORE:' print getoutput('ls -rlast /tmp/example') rmtree('/tmp/example') print '\nAFTER:' print getoutput('ls -rlast /tmp/example') 将一个文件或目录从一个位置移动到另外一个位置,可使用move()。 from shutil import * from glob import glob with open('example.txt', 'wt') as f: f.write('i love you') print 'BEFORE: ', glob('example*') move('example.txt', 'example.out') print 'AFTER: ',glob('example*') 其语义与UNIX命令mv相似。若是源与目标都在同一个文件系统内,则会重命名源文件。不然,源文件会复制到目标文件,将源文件删除。 >>> ================================ RESTART ================================ >>> BEFORE: ['example', 'example.txt'] AFTER: ['example', 'example.out']