From fb7d3d5fa605596d419a5ea3950827c59c53c942 Mon Sep 17 00:00:00 2001 From: jvoisin Date: Sat, 23 Jul 2011 19:16:58 +0200 Subject: Add handling of the "comment" field on a zip archive --- lib/archive.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'lib') diff --git a/lib/archive.py b/lib/archive.py index 2ee817e..33fe6e7 100644 --- a/lib/archive.py +++ b/lib/archive.py @@ -40,6 +40,8 @@ class ZipStripper(GenericArchiveStripper): def is_clean(self): zipin = zipfile.ZipFile(self.filename, 'r') + if zipin.comment != '': + return False for item in zipin.infolist(): if not self.is_file_clean(item): return False @@ -69,6 +71,7 @@ class ZipStripper(GenericArchiveStripper): zipmeta['system'] = field.create_system zipmeta['zip_version'] = field.create_version metadata[field.filename] = zipmeta + metadata["%s comment" % self.filename] = zipin.comment zipin.close() return metadata @@ -94,6 +97,7 @@ class ZipStripper(GenericArchiveStripper): mat.secure_remove(item.filename) else: self.folder_list.insert(0, item.filename) + zipout.comment = '' logging.info('%s treated' % self.filename) self.remove_folder() zipin.close() -- cgit v1.3