python编程快速上手之第9章实践项目参考答案

本章介介绍了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')