security-and-privacy/mat2
security-and-privacy/mat2
1
0
Fork 0
mirror of synced 2024-11-22 09:14:23 +01:00
Code Releases Activity

Improve epub compatibility

Browse Source
This commit is contained in:
jvoisin 2021-01-30 16:24:42 +01:00
parent 88fa71fbde
commit f8111547ae
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libmat2/epub.py
View File

@ -15,11 +15,12 @@ class EPUBParser(archive.ZipParser):
'META-INF/container.xml', 'META-INF/container.xml',
'mimetype', 'mimetype',
'OEBPS/content.opf', 'OEBPS/content.opf',
'content.opf',
})) }))
self.uniqid = uuid.uuid4() self.uniqid = uuid.uuid4()
def _specific_get_meta(self, full_path, file_path): def _specific_get_meta(self, full_path, file_path):
if file_path != 'OEBPS/content.opf': if not file_path.endswith('content.opf'):
return {} return {}
with open(full_path, encoding='utf-8') as f: with open(full_path, encoding='utf-8') as f:
@ -31,7 +32,7 @@ class EPUBParser(archive.ZipParser):
return {file_path: 'harmful content', } return {file_path: 'harmful content', }
def _specific_cleanup(self, full_path: str): def _specific_cleanup(self, full_path: str):
if full_path.endswith('OEBPS/content.opf'): if full_path.endswith('content.opf'):
return self.__handle_contentopf(full_path) return self.__handle_contentopf(full_path)
elif full_path.endswith('OEBPS/toc.ncx'): elif full_path.endswith('OEBPS/toc.ncx'):
return self.__handle_tocncx(full_path) return self.__handle_tocncx(full_path)