python(6)-shutil模块

高级的 文件、文件夹、压缩包 处理模块

shutil.copyfileobj(fsrc, fdst[, length])

将文件内容拷贝到另一个文件中：

#源码
def copyfileobj(fsrc, fdst, length=16*1024):
"""copy data from file-like object fsrc to file-like object fdst"""
while 1:
buf = fsrc.read(length)
if not buf:
break
fdst.write(buf)

#复制文件
import shutil

f = open('settings.py', 'rb')
f1 = open('aaa.py', 'wb')

shutil.copyfileobj(f, f1, length=20)

f.close()
f1.close()

shutil.copyfile(src, dst)

拷贝文件

#源码i
def copyfile(src, dst):
"""Copy data from src to dst"""
if _samefile(src, dst):
raise Error("`%s` and `%s` are the same file" % (src, dst))

for fn in [src, dst]:
try:
st = os.stat(fn)
except OSError:
# File most likely does not exist
pass
else:
# XXX What about other special files? (sockets, devices...)
if stat.S_ISFIFO(st.st_mode):
raise SpecialFileError("`%s` is a named pipe" % fn)

with open(src, 'rb') as fsrc:
with open(dst, 'wb') as fdst:
copyfileobj(fsrc, fdst)

shutil.copymode(src, dst)

只拷贝权限，内容、组、用户均不变

#源码
def copymode(src, dst):
"""Copy mode bits from src to dst"""
if hasattr(os, 'chmod'):
st = os.stat(src)
mode = stat.S_IMODE(st.st_mode)
os.chmod(dst, mode)

示例：

#第一步，查看文件权限
# ll
总用量 16
drwxrwxr-x 2 huangxm huangxm 4096  3月 24 14:32 ./
drwxrwxr-x 6 huangxm huangxm 4096  3月 24 14:32 ../
-rw-rw-r-- 1 huangxm huangxm   49  3月 24 14:32 aaa.txt
-rw-r--r-- 1 root     root       29  3月 24 14:32 bbb.txt

#第二步，新建ccc.py，写入以下代码：
#!/usr/bin/env python3

import shutil
#将bbb.txt的文件权限复制给aaa.txt
shutil.copymode('bbb.txt', 'aaa.txt')

#python3 ccc.py执行代码
#第三步：查看文件权限
# ls -l
总用量 12
-rw-r--r-- 1 huangxm huangxm 49  3月 24 14:32 aaa.txt
-rw-r--r-- 1 root     root     29  3月 24 14:32 bbb.txt
-rw-r--r-- 1 root     root     77  3月 24 14:35 ccc.py

shutil.copystat(src, dst)

拷贝状态信息，包括：mode bits, atime, mtime, flags

#源码
def copystat(src, dst):
"""Copy all stat info (mode bits, atime, mtime, flags) from src to dst"""
st = os.stat(src)
mode = stat.S_IMODE(st.st_mode)
if hasattr(os, 'utime'):
os.utime(dst, (st.st_atime, st.st_mtime))
if hasattr(os, 'chmod'):
os.chmod(dst, mode)
if hasattr(os, 'chflags') and hasattr(st, 'st_flags'):
try:
os.chflags(dst, st.st_flags)
except OSError, why:
for err in 'EOPNOTSUPP', 'ENOTSUP':
if hasattr(errno, err) and why.errno == getattr(errno, err):
break
else:
raise

shutil.copy(src, dst)

拷贝文件和权限

#源码
def copy(src, dst):
"""Copy data and mode bits ("cp src dst").

The destination may be a directory.

"""
if os.path.isdir(dst):
dst = os.path.join(dst, os.path.basename(src))
copyfile(src, dst)
copymode(src, dst)

shutil.copy2(src, dst)

拷贝文件和状态信息

#源码
def copy2(src, dst):
"""Copy data and all stat info ("cp -p src dst").

The destination may be a directory.

"""
if os.path.isdir(dst):
dst = os.path.join(dst, os.path.basename(src))
copyfile(src, dst)
copystat(src, dst)

shutil.ignore_patterns(*patterns)

shutil.copytree(src, dst, symlinks=False, ignore=None)

src = 'D:/temp/python/zuoye/day6'
#当前目录的zy_d6文件夹,不需要提前创建
dst = 'zy_d6'
#要排除的文件，可以多个用,分隔 支持通配符比如temp*
ignore_file = shutil.ignore_patterns('*.py')
shutil.copytree(src, dst, ignore=ignore_file)

shutil.rmtree(path)

递归删除文件

shutil.move(src, dst)

递归移动文件

shutil.make_archive(base_name, format, …)

创建压缩包并返回文件路径，例如： zip、tar

base_name: 压缩包的文件名，也可以是压缩包的路径。文件名是保存到当前路径，路径时保存到指定路径

format: 压缩包种类， zip, tar, bztar, gztar

root_dir: 要压缩的文件夹路径（默认当前目录）

owner: 用户， 默认当前用户

group: 组，默认当前组

logger: 用户于记录日志，通常是logging.Logger对象

#将当前目录打包放到当前目录下
ret = shutil.make_archive('dq', 'gztar')

返回值是压缩包名称，上例中ret的值是 dq.tar.gz

#将指定目录打包放到当前目录
root_dir = 'D:/temp/python/zuoye/day6'
ret = shutil.make_archive('dq', 'gztar', root_dir=rootdir)
print(ret)

shutil 对压缩包的处理是调用ZipFile和TarFile两个模块来进行的，详细：

import zipfile

# 压缩
z = zipfile.ZipFile('laxi.zip', 'w')
z.write('a.log')
z.write('data.data')
z.close()

# 解压
z = zipfile.ZipFile('laxi.zip', 'r')
z.extractall()
z.close()

import tarfile

# 压缩
tar = tarfile.open('your.tar','w')
tar.add('/Users/wupeiqi/PycharmProjects/bbs2.zip', arcname='bbs2.zip')
tar.add('/Users/wupeiqi/PycharmProjects/cmdb.zip', arcname='cmdb.zip')
tar.close()

# 解压
tar = tarfile.open('your.tar','r')
tar.extractall()  # 可设置解压地址
tar.close()