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

feat: Only activate pocketbook extractor optionally

Browse source 
Since we make the dependencies for pocketbook html extraction optional
as an extra, this commit ensures the extractor (and cli option) only
gets loaded when they exist.
This commit is contained in:
Marty Oehme 2024-01-24 12:07:04 +01:00
parent e1e09b6011
commit e7e5258b34
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
papis_extract/extractors/__init__.py
View file

@ -1,8 +1,19 @@
import papis.logging
from papis_extract.extraction import Extractor from papis_extract.extraction import Extractor
from papis_extract.extractors import pdf from papis_extract.extractors import pdf
from papis_extract.extractors.pocketbook import PocketBookExtractor from papis_extract.extractors.pocketbook import PocketBookExtractor
logger = papis.logging.get_logger(__name__)
all_extractors: dict[str, Extractor] = { all_extractors: dict[str, Extractor] = {
"pdf": pdf.PdfExtractor(), "pdf": pdf.PdfExtractor(),
"pocketbook": PocketBookExtractor(),
} }
try:
import bs4
import magic
all_extractors["pocketbook"] = PocketBookExtractor()
except ImportError:
logger.debug("pocketbook extractor not activated.")