测试下remove函数的灵活性:
# 建立10个目录tmp_* ,以及10各文件tmp__*:
for i in range(10):
os.mkdir('tmp_'+str(i))
f = open('tmp__'+str(i), 'w'); f.close()
remove('tmp_1') # tmp_1为目录
remove(glob.glob('tmp_[0-9]') + glob.glob('tmp__[0-9]'))
作为上述remove函数实现的一个注记,进行了下列测试:
if not isinstance(files, list):
它实际上是过于严厉。需要的只是一个被遍历的一个文件/目录名序列。实际上,并不关心名称是否存储在一个列表、元组或者数值数组中,所以更好的测试应该像下面这样:
if not operator.isSequenceType(files):
四、文件的复制与重命名
当要复制文件时,可以使用shutil模块:
import shutil
shutil.copy(myfile, tmpfile)
#拷贝最后访问时间和最后修改时间:
shutil.copy2(myfile, tmpfile)
# 拷贝一个目录树:
shutil.copytree(root_of_tree, destination_dir, True)
Copytree的第三个参数规定对符号链接的处理,其中True表示保留符号链接;而False则意味着使用文件的物理副本替代符号链接。
Python语言能够很好地支持路径名的跨平台组成:Os.path.join能使用正确的分界符(在UNIX和Mac OS X操作系统中使用/,在 Windows 上使用\)来联接目录和文件名,变量os.curdir和os.pardir分别表示当前工作目录及其父目录。 像下面的UNIX操作系统命令
cp .
可以使用Python语言提供一个跨平台的实现:
shutil.copy(os.path.join(os.pardir,os.pardir,'f1.c'), os.curdir)
Os模块中的rename函数通常被用于重命名一个文件:
os.rename(myfile, 'tmp.1') # 将myfile重命名为'tmp.1'
这个函数也可用来在相同的文件系统之内移动文件。这里,将myfile移动到目录d下面:
os.rename(myfile, os.path.join(d, myfile))
在跨文件系统移动文件时,可以先使用shutil.copy2来复制文件,然后再删除原来的副本即可,如下:
shutil.copy2(myfile, os.path.join(d, myfile))
os.remove(myfile)
后面这种移动文件的方法是最安全的。
五、分解路径名
假设使用变量fname来存放一个包含完整路径的文件名,例如:
/usr/home/hpl/scripting/python/intro/hw.py
有时候,需要将这样的文件路径拆分为基本名称hw.py和目录名/usr/home/hpl/scripting/python/intro。在Python语言中,可以使用下列代码达到目的:
basename = os.path.basename(fname)
dirname = os.path.dirname(fname)
# 或
dirname, basename = os.path.split(fname)
扩展名是通过os.path.splitext函数提取出来的,
root, extension = os.path.splitext(fname)
这样,fname中的扩展名部分即.py被赋给变量extension,而其余部分则赋给了变量root。()如果想得到不带点号的扩展名的话,只需使用os.path.splitext(fname)[1][1:]即可。
假设一个文件名为f,其扩展名随意,若想将其扩展名改为ext,可以使用下面的代码:
newfile = os.path.splitext(f)[0] + ext
下面是一个具体的示例:
>>> f = '/some/path/case2.data_source'
>>> moviefile = os.path.basename(os.path.splitext(f)[0] + '.mpg')
>>> moviefile
'case2.mpg'
六、目录的创建和移动
Os模块中的函数mkdir可以用来创建目录,而chdir函数则可以移动目录,如下:
origdir = os.getcwd() # 将当前位置记下来
newdir = os.path.join(os.pardir, 'mynewdir')
if not os.path.isdir(newdir):
os.mkdir(newdir) # 或者os.mkdir(newdir,'0755')
os.chdir(newdir)
...
os.chdir(origdir) # 返回原目录
os.chdir(os.environ['HOME']) # 移到主目录
假设想要在自己的主目录下创建一个新目录py/src/test1,但是目前py、src和test1都不存在。如果使用mkdir命令来创建的话,需要使用三次才能建好这个嵌套的目录,但是使用Python语言提供的os.makedirs命令的话,则无需这样麻烦了,该命令可以一次建好整个目录:
os.makedirs(os.path.join(os.environ['HOME'],'py','src','test1'))
七、遍历目录树
下面的函数调用
os.path.walk(root, myfunc, arg)
将遍历root目录树;然后,对每个目录名dirname分别调用myfunc(arg, dirname, files)即可,这里参数files是dir中的文件名列表(可通过调用os.listdir(dirname)来获得);arg是用户从调用代码中传递来的参数。对于UNIX操作系统用户来说,Python语言中跨平台的os.path.walk相当于Unix命令find。
在解释os.path.walk的用法时,人们常使用写出主目录中所有子目录内的文件的名称为例进行说明。当然,也可以在一个交互式的Python命令行中使用下列代码段来体会os.path.walk的使用:
def ls(arg, dirname, files):
print dirname, 'has the files', files
os.path.walk(os.environ['HOME'], ls, None)
本例中,参数arg并非必需,所以在os.path.walk调用中让其取值为None即可。
为了列出主目录中所有大于1Mb的文件,可以使用下面的代码:
def checksize1(arg, dirname, files):
for file in files:
filepath = os.path.join(dirname, file)
if os.path.isfile(filepath):
size = os.path.getsize(filepath)
if size > 1000000:
size_in_Mb = size/1000000.0
arg.append((size_in_Mb, filename))
bigfiles = []
root = os.environ['HOME']
os.path.walk(root, checksize1, bigfiles)
for size, name in bigfiles:
print name, '大小为', size, 'Mb'