From 626669f95f83f3e920966e8f82b301c269cdb1b3 Mon Sep 17 00:00:00 2001
From: jvoisin <julien.voisin@dustri.org>
Date: Sun, 7 Mar 2021 17:50:17 +0100
Subject: [PATCH] Add some typing to epub.py
---
libmat2/epub.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/libmat2/epub.py b/libmat2/epub.py
index f43667e..83f5728 100644
--- a/libmat2/epub.py
+++ b/libmat2/epub.py
@@ -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:
--
GitLab