1
0
Fork 0

Add some typing to epub.py

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

View File

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