本章介介绍了shutil，zipfile模块的使用，我们先来认识一下这2个模块吧。
一.shutil模块
shutil模块主要用于对文件或文件夹进行处理，包括：复制，移动，改名和删除文件，在shutil模块中主要以下这么几个函数：
1.复制文件和文件夹
shutil模块提供了2个函数:shutil.copy()和shutil.copytree()
shutil.copy的语法格式：
copy(src, dst)
作用：
将src处的文件复制到dst路径中去，其中src,dst都是字符串形式的路径。如果dst是一个文件名，它将作为被复制文件的新名字，相当于把原路径的文件复制到新路径并改名。
举例：
将/etc/my.cnf 复制到/root/mysql中
In []: import shutil
In []: shutil.copy('/etc/my.cnf','/root/mysql/')
Out[]: '/root/mysql/my.cnf'
In []: ll /root/mysql/
total
-rw-r--r--. root Apr : my.cnf
将/etc/my.cnf复制到/root/mysql并改名为mysql.cnf
In []: shutil.copy('/etc/my.cnf','/root/mysql/mysql.cnf')
Out[]: '/root/mysql/mysql.cnf'
In []: ll /root/mysql/
total
-rw-r--r--. root Apr : my.cnf
-rw-r--r--. root Apr : mysql.cnf
shutil.copytree的语法格式：
copytree(src, dst)
作用：
复制整个文件夹。将src处的文件夹，包括它的所有文件和子文件夹，复制到路径dst处的文件夹。返回一个新复制的文件夹路径的字符串。
举例：
In []: shutil.copytree('/etc/yum.repos.d','/root/repo.back')
Out[]: '/root/repo.back'
In []: ll /root/drwxr-xr-x. root Apr repo.back/
drwxr-xr-x. root Apr repo.bak/
注意：dst必须是一个系统中不存在目录，不然会报错：
In [20]: shutil.copytree('/etc/yum.repos.d','/root/repo.bak')
FileExistsError: [Errno ] File exists: '/root/repo.bak'
2.文件和文件夹的移动和改名
shutil.move()
语法格式：
move(stc,dst)
作用：
将路径stc处的文件夹移动到路径dst，并返回新位置的绝对路径的字符串。
举例：
将/root/目录下的a.txt移动到/root/test/目录中
In []: shutil.move('/root/a.txt','/root/test/')
Out[]: '/root/test/a.txt'
In []: ll /root/test
total
-rw-r--r--. root Apr : a.txt
注意：
如果dst指向一个文件夹，src文件将移动到dst中，并保持原来的文件名，前提是dst必须是系统中已经存在的目录。
如果目标文件中已存在相同名称的文件将被覆盖，需要注意。
3.删除文件和文件夹
在os模块中：
os.remove(path)可以删除一个文件
os.rmdir(path)可以删除一个空文件夹。
在shutil模块中：
shutil.rmtree(path)可以删除一个文件夹及其所有的内容。
语法格式：
os.rmdir(path)
shutil.rmtree(path)
举例：
In []: os.remove('/root/test/a.txt')
In []: ll /root/test/
total
In []: shutil.move('/root/CentOS-Base.repo','/root/test/')
Out[]: '/root/test/CentOS-Base.repo'
In []: ll test
total
-rw-r--r--. root Apr CentOS-Base.repo
In []: shutil.rmtree('/root/test')
In []: ll
total
-rw-------. root Apr : anaconda-ks.cfg
drwxr-xr-x. root Apr download/
drwxr-xr-x. root Apr : mysql/
drwxr-xr-x. root Apr : py34/
drwxr-xr-x. root Apr : python/
drwxr-xr-x. root Apr repo.back/
drwxr-xr-x. root Apr repo.bak/
注意：
以上的删除都是永久的删除。为了安全起见最好使用send2trash第三方模块，它会将删除的文件放入回收站。在python3中已集成了这个模块。
send2trash用法:
import send2trash
send2trash(path)
二.遍历目录树
对文件的处理，尤其是批量操作就不得不对目录进行遍历。在python中os模块中的os.walk()函数就可以做到。
这个函数会递归遍历指定目录及子目录，返回一个3元组信息：当前目录名，子目录名，文件名，不包括 . 和 ..
常见用法：
#!/usr/bin/env python3.4
#coding:utf-8
import os
for foldName,subfolders,filenames in os.walk('/root/'):
print('The current folder is: ' + foldName)
for subfolder in subfolders:
print('subfolder of ' + foldName + ':' + subfolder)
for filename in filenames:
print('file inside ' + foldName + ':' + filename)
print('')
三.实践项目参考答案
#!/usr/bin/env python3.4
# coding:utf-8
import os
import shutil
import send2trash # 9.8.1
# 拷贝指定格式文件到指定目录，下面程序是将/etc目录下所的.conf文件拷贝到/root/test/目录里。
src = '/etc/'
dst = '/root/test/'
ftype = '.conf'
count = 0
for filename in os.listdir(src):
if filename.endswith(ftype):
shutil.copy(src + filename,dst)
count += 1
print('文件 ' + src + filename + '\t被拷贝到---> ' + dst + ' 目录下')
print("该目录下所有的 " + ftype + "文件已被拷贝到" + dst + "目录下")
print('共拷贝了 ' + str(count) + ' 个文件') # 9.8.2
# 搜索指定目录下大于100M的文件,打印出来并删除
# 可以手动创建一个指定大小的空文件做试验
# dd if=/dev/zero of=hello.txt bs=100M count=1
for foldname,subfolders,filenames in os.walk(dst):
for files in filenames:
if os.path.getsize(dst + files) / 1024 /1024 > 100:
print('大于100M的文件有：' + files + ' ' + str(os.path.getsize(dst + files) / 1024 / 1024) +'Mb')
send2trash.send2trash(dst + files)
9.8.3
假设test文件夹下有如下文件，文件是以spam开头加上数字编号，但是编号并不连续有缺失，而且有的并不包含数字，我们需要找出不符合文件名的文件并重新命名成连续编号的文件名。
(py34) [root@master test]# ls
spam002.txt spam004.txt spam006.txt spam008.txt spam999.txt
spam003.txt spam005.txt spam007.txt spam011.txt spamkkdf.txt
参考代码如下：
#!/usr/bin/env python3.4
# coding:utf-8
import re
import os
fdir = '/root/python/py-9/test/'
fdir_list = os.listdir(fdir)
fdir_count = len(fdir_list)
print(fdir_list)
print('该目录下共有 %d 个文件' %fdir_count)
f_pre = 'spam'
f_num = []
f_end = '.txt'
fs_list = []
# 这里只假定文件数量小100的情况
for i in range(1,fdir_count + 1):
if i < 10:
f_name = f_pre + '' + str(i) + f_end
f_num.append('' + str(i))
fs_list.append(f_name)
else:
f_name = f_pre + '' + str(i) + f_end
f_num.append('' + str(i))
fs_list.append(f_name)
max_f_num = max(f_num)
print('该目录下文件最大的编号应该是: %s' %max_f_num)
print('正确的文件名应该是:')
print(fs_list) # 使用正则表达式搜索目录中已有编号的文件并存入列表yf_num中
re_num = '\d{3}'
yf_num = re.findall(f_pre + re_num + f_end,' '.join(fdir_list))
ra_num = re.findall(re_num,' '.join(fdir_list))
print('目录中已有编号文件：\n%s' %yf_num) # fq_list为目录中缺失编号的文件名列表
# fx_list为当前目录中需要修改名称的文件列表
fq_list = []
fx_list = []
# 定位缺失的编号文件并放入列表中
for a in fs_list:
if a not in yf_num:
fq_list.append(a)
print('缺少的文件编号是：\n%s' %fq_list) # 查找目录中没有编号或不正连续的编号文件并放入列表中
for f_rename in fdir_list:
if f_rename not in fs_list:
fx_list.append(f_rename)
print('需要修改的文件名有：\n%s' %fx_list) # 更改文件名
for k in fq_list:
for v in fx_list:
os.rename(fdir + v,fdir + k)
# 每当修完一个文件名应该更新一下这个列表
fx_list.remove(v)
print('改完名后的结果为:')
os.system('ls')
今天回来看看，决定用函数的方式来练习并实现，代码有了一些小的改进，上代码：
#!/usr/bin/env python3.4
# coding:utf-8
import re
import os
fdir = '/root/github/shell/python3/py-9/test/'
fdir_list = os.listdir(fdir)
fdir_f_count = len(fdir_list)
print('当前目录的文件为:\n%s' % fdir_list)
def getFileformat(f_pre,f_num,f_end):
fileformat = f_pre + f_num + f_end
return fileformat
file_list = []
def getTruefile():
for i in range(1,fdir_f_count + 1):
if i < 10:
f_format = getFileformat('spam','' + str(i),'.txt')
file_list.append(f_format)
else:
f_format = getFileformat('spam','' + str(i),'.txt')
file_list.append(f_format)
return file_list
truefilelist = getTruefile()
print('正确的文件编号应该是:\n%s' %truefilelist)
lostnumfilelist = []
def getLostnumfile(fdirlist,truelist):
for lf in truelist:
if lf not in fdirlist:
lostnumfilelist.append(lf)
return lostnumfilelist
lostnumfile = getLostnumfile(fdir_list,file_list)
print('缺失的文件编号为:\n%s' %lostnumfile) renamelist = []
def getrenamefile(fdirlist,func):
for a in fdirlist:
if a not in func:
renamelist.append(a)
return renamelist
renamefilelist = getrenamefile(fdir_list,file_list)
print('需要修改的文件是\n%s' %renamefilelist) def renamefile(func1,func2):
for b in func1:
for c in func2:
os.rename(fdir + c,fdir + b)
func2.remove(c)
rename = renamefile(lostnumfile,renamefilelist)
os.chdir(fdir)
print('修改后的结果为:')
os.system('ls')