Marty/papis-extract
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: Fixed pocketbook extractor trying to read all files

Browse source 
The complete read routine would work before figuring out that it is
a file of xml mimetype. This means that it would try to read to memory
any file as the first thing, pdfs, even binaries. Of course doing
so crashed the program.
This commit is contained in:
Marty Oehme 2024-01-25 21:40:31 +01:00
parent 72ddaaf1bc
commit 163fd63038
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
papis_extract/extractors/pocketbook.py
View file

@ -12,11 +12,11 @@ logger = papis.logging.get_logger(__name__)
class PocketBookExtractor: class PocketBookExtractor:
def can_process(self, filename: Path) -> bool: def can_process(self, filename: Path) -> bool:
content = self._read_file(filename) if not magic.from_file(filename, mime=True) == "text/xml":
if not content:
return False return False
if not magic.from_buffer(content, mime=True) == "text/xml": content = self._read_file(filename)
if not content:
return False return False
html = BeautifulSoup(content, features="xml") html = BeautifulSoup(content, features="xml")