diff options
| -rw-r--r-- | libmat2/office.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/libmat2/office.py b/libmat2/office.py index 997a247..372d69a 100644 --- a/libmat2/office.py +++ b/libmat2/office.py | |||
| @@ -217,8 +217,13 @@ class MSOfficeParser(ArchiveBasedAbstractParser): | |||
| 217 | removed_fnames = set() | 217 | removed_fnames = set() |
| 218 | with zipfile.ZipFile(self.filename) as zin: | 218 | with zipfile.ZipFile(self.filename) as zin: |
| 219 | for fname in [item.filename for item in zin.infolist()]: | 219 | for fname in [item.filename for item in zin.infolist()]: |
| 220 | if any(map(lambda r: r.search(fname), self.files_to_omit)): # type: ignore | 220 | for file_to_omit in self.files_to_omit: |
| 221 | removed_fnames.add(fname) | 221 | if file_to_omit.search(fname): |
| 222 | matches = map(lambda r: r.search(fname), self.files_to_keep) | ||
| 223 | if any(matches): # the file is whitelisted | ||
| 224 | continue | ||
| 225 | removed_fnames.add(fname) | ||
| 226 | break | ||
| 222 | 227 | ||
| 223 | root = tree.getroot() | 228 | root = tree.getroot() |
| 224 | for item in root.findall('{%s}Override' % namespace['']): | 229 | for item in root.findall('{%s}Override' % namespace['']): |
