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

Add some typing to epub.py

Browse Source
This commit is contained in:
jvoisin 2021-03-07 17:50:17 +01:00
parent 497f5f71fc
commit 626669f95f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libmat2/epub.py
View File

@ -2,6 +2,7 @@ import logging
import re import re
import uuid import uuid
import xml.etree.ElementTree as ET # type: ignore import xml.etree.ElementTree as ET # type: ignore
from typing import Dict, Any
from . import archive, office from . import archive, office
@ -38,7 +39,7 @@ class EPUBParser(archive.ZipParser):
except (TypeError, UnicodeDecodeError): except (TypeError, UnicodeDecodeError):
return {file_path: 'harmful content', } return {file_path: 'harmful content', }
def _specific_cleanup(self, full_path: str): def _specific_cleanup(self, full_path: str) -> bool:
if full_path.endswith('hmh.opf') or full_path.endswith('content.opf'): if full_path.endswith('hmh.opf') or 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'):
@ -47,7 +48,7 @@ class EPUBParser(archive.ZipParser):
return self.__handle_ops_xml(full_path) return self.__handle_ops_xml(full_path)
return True return True
def __handle_ops_xml(self, full_path: str): def __handle_ops_xml(self, full_path: str) -> bool:
try: try:
tree, namespace = office._parse_xml(full_path) tree, namespace = office._parse_xml(full_path)
except ET.ParseError: # pragma: nocover except ET.ParseError: # pragma: nocover
@ -63,7 +64,7 @@ class EPUBParser(archive.ZipParser):
return True return True
def __handle_tocncx(self, full_path: str): def __handle_tocncx(self, full_path: str) -> bool:
try: try:
tree, namespace = office._parse_xml(full_path) tree, namespace = office._parse_xml(full_path)
except ET.ParseError: # pragma: nocover except ET.ParseError: # pragma: nocover
@ -79,7 +80,7 @@ class EPUBParser(archive.ZipParser):
short_empty_elements=False) short_empty_elements=False)
return True return True
def __handle_contentopf(self, full_path: str): def __handle_contentopf(self, full_path: str) -> bool:
try: try:
tree, namespace = office._parse_xml(full_path) tree, namespace = office._parse_xml(full_path)
except ET.ParseError: except ET.ParseError: