import zipfile

import os

from zipfile import ZipFile

class ZipManage(ZipFile):

def check(self, Name):

# 检查 文件指针是否有效

if self.fp == '':

raise RuntimeError('ZIP archive is closed')

# 检查 文件是否用 append 模式打开

if self.mode != 'a':

raise RuntimeError('ZIP archive Requires appended("a") mode to open')

# 检查 文件或文件夹 是否在zip中

is_existed = False

for FileName in self.filelist:

if FileName.filename.startswith(Name):

is_existed = True;

break;

if is_existed == False:

raise RuntimeError('ZIP archive not found ', Name)

# 删除zip包中的文件

def remove(self, szFileName):

self.check(szFileName)

# 获取 要删除文件在zip中的信息

fileinfo = self.getinfo(szFileName)

print(type(fileinfo))

headerOffSet = fileinfo.header_offset

fileBlockLen = len(fileinfo.FileHeader()) + fileinfo.compress_size

# 根据要删除的文件信息， 更新zip包中其他文件的头部偏移值

for info in self.infolist():

if info.header_offset >= (headerOffSet + fileBlockLen):

info.header_offset -= fileBlockLen

# 将文件指针移动到 待删除文件的末尾

self.fp.seek(headerOffSet + fileBlockLen)

# 读取 zip中 待删除文件后 的全部数据

data = self.fp.read()

# 移动文件指针到 待删除文件的开始

self.fp.seek(headerOffSet)

# 覆写数据

self.fp.write(data)

# 截断文件

self.fp.truncate()

# 更新 zip 中的文件索引信息

self._didModify = True

self.filelist.remove(fileinfo)

del self.NameToInfo[fileinfo.filename]

# 删除 zip 中文件夹

def remove_dir(self, szDirName):

self.check(szDirName)

# 获取 要删除文件夹在zip中的信息

dirInfo = self.getinfo(szDirName)

# 获取 头部信息 和 文件块的大小

headerOffSet = dirInfo.header_offset

fileBlockLen = 0

for file in self.infolist():

if file.filename.startswith(szDirName):

fileBlockLen += (len(file.FileHeader()) + file.compress_size)

for file_head_info in self.infolist():

if file_head_info.header_offset >= (headerOffSet + fileBlockLen):

file_head_info.header_offset -= fileBlockLen

# 将文件指针移动到 待删除文件的末尾

self.fp.seek(headerOffSet + fileBlockLen)

# 读取 zip中 待删除文件后 的全部数据

latedata = self.fp.read()

# 移动文件指针到 待删除文件的开始

self.fp.seek(headerOffSet)

# 覆写数据

self.fp.write(latedata)

# 截断文件

self.fp.truncate()

# 更新 zip 中的文件索引信息

self._didModify = True

# 反向迭代删除

for i in range(len(self.infolist())-1, -1, -1):

info = self.infolist()[i]

if info.filename.startswith(szDirName):

self.filelist.remove(info)

del self.NameToInfo[info.filename]

# 替换 zip 文件中的信息

def replace(self, szReplaceFilename, szNewFilePath):

self.check(szReplaceFilename)

if not os.path.exists(szNewFilePath):

print(szNewFilePath,' is not existed')

return;

fileinfo = self.getinfo(szReplaceFilename)

self.remove(szReplaceFilename)

self.write(szNewFilePath,fileinfo.filename,fileinfo.compress_type)

def test_remove():

file = ZipManage('/home/s/Desktop/1.zip', 'a')

file.remove('1/1/2/22/test.txt')

def test_remove_dir():

file = ZipManage('/home/s/Desktop/1.zip', 'a')

file.remove_dir('1/1/2/22/')

def test_replace():

file = ZipManage('/home/s/Desktop/1.zip', 'a')

file.replace('1/1/test.txt', '/home/s/Desktop/test.txt')