From 91fbc466c5db2d02c839cd7e89fa5b0b11fb74b3 Mon Sep 17 00:00:00 2001 From: Alex Pyrgiotis Date: Mon, 7 Oct 2024 14:45:09 +0300 Subject: [PATCH] Add an import preference for vendored packages Prefer importing packages from ./dangerzone/vendor, if there is one there, instead of using the system ones. --- dangerzone/__init__.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/dangerzone/__init__.py b/dangerzone/__init__.py index f2a9961..f029398 100644 --- a/dangerzone/__init__.py +++ b/dangerzone/__init__.py @@ -1,6 +1,19 @@ +import logging import os import sys +logger = logging.getLogger(__name__) + + +try: + from . import vendor # type: ignore [attr-defined] + + vendor_path: str = vendor.__path__[0] + logger.debug(f"Using vendored PyMuPDF libraries from '{vendor_path}'") + sys.path.insert(0, vendor_path) +except ImportError: + pass + if "DANGERZONE_MODE" in os.environ: mode = os.environ["DANGERZONE_MODE"] else: