dependencies.py

import sys
import io
import importlib
import platform
import site


def _import_module(module_name, feedback=None):
    try:
        return importlib.import_module(module_name)
    except Exception as e:
        if feedback:
            feedback.pushInfo(f'Could not import "{module_name}": {e}')
        return None
    

def _add_user_site_to_path(feedback=None):
    try:
        user_site = site.getusersitepackages()

        if user_site and user_site not in sys.path:
            sys.path.append(user_site)

            if feedback:
                feedback.pushInfo(f'Added user site-packages to sys.path: {user_site}')

    except Exception as e:
        if feedback:
            feedback.pushInfo(f'Could not add user site-packages: {e}')


def _pip_install(args, feedback=None):
    old_stdout = sys.stdout
    old_stderr = sys.stderr

    fake_out = io.StringIO()
    fake_err = io.StringIO()

    try:
        import pip

        sys.stdout = fake_out
        sys.stderr = fake_err

        result = pip.main(args)

        stdout_text = fake_out.getvalue().strip()
        stderr_text = fake_err.getvalue().strip()

        if stdout_text and feedback:
            feedback.pushInfo(stdout_text)
        if stderr_text and feedback:
            feedback.pushInfo(stderr_text)

        if result not in (0, None):
            msg = stderr_text or stdout_text or f'pip.main returned code {result}'
            if feedback:
                feedback.reportError(msg)
            return False

        return True

    except Exception as e:
        stdout_text = fake_out.getvalue().strip()
        stderr_text = fake_err.getvalue().strip()
        extra = stderr_text or stdout_text

        if feedback:
            if extra:
                feedback.reportError(f'{e} | pip output: {extra}')
            else:
                feedback.reportError(str(e))
        return False

    finally:
        sys.stdout = old_stdout
        sys.stderr = old_stderr


def _install_package(package_name, feedback=None):
    system_os = platform.system()

    if feedback:
        feedback.pushInfo(f'Package "{package_name}" not found. Trying to install...')
        feedback.pushInfo(f'Operating system detected: {system_os}')
        feedback.pushInfo('Installation method: pip.main()')

    strategies = [
        ["install", package_name],
        ["install", "--user", package_name],
        ["install", "--upgrade", package_name],
        ["install", "--user", "--upgrade", package_name],
    ]

    for args in strategies:
        if feedback:
            feedback.pushInfo(f'Trying pip: {" ".join(args)}')

        ok = _pip_install(args, feedback)
        if ok:
            importlib.invalidate_caches()
            if feedback:
                feedback.pushInfo(f'Package "{package_name}" installed successfully.')
            return True

    if feedback:
        feedback.reportError(
            f'Failed to install "{package_name}" automatically. '
            f'Try manual installation in the QGIS Python environment.'
        )
    return False



# ==========================
# PyPDF2
# ==========================

def ensure_pypdf2(feedback=None):
    PyPDF2 = _import_module("PyPDF2", feedback)

    if PyPDF2 is not None and hasattr(PyPDF2, "PdfReader"):
        if feedback:
            try:
                feedback.pushInfo(f'PyPDF2 already available: {PyPDF2.__file__}')
            except Exception:
                feedback.pushInfo('PyPDF2 already available.')
        return PyPDF2.PdfReader

    if not _install_package("PyPDF2", feedback):
        return None

    importlib.invalidate_caches()

    PyPDF2 = _import_module("PyPDF2", feedback)

    if PyPDF2 is None or not hasattr(PyPDF2, "PdfReader"):
        if feedback:
            feedback.reportError(
                'PyPDF2 installation was attempted, but PdfReader is still unavailable.'
            )
        return None

    if feedback:
        try:
            feedback.pushInfo(f'PyPDF2 loaded after installation: {PyPDF2.__file__}')
        except Exception:
            feedback.pushInfo('PyPDF2 loaded after installation.')

    return PyPDF2.PdfReader


# ==========================
# defusedxml
# ==========================

def ensure_defusedxml(feedback=None):
    _add_user_site_to_path(feedback)

    defusedxml = _import_module("defusedxml", feedback)

    if defusedxml is not None:
        if feedback:
            try:
                feedback.pushInfo(f'defusedxml already available: {defusedxml.__file__}')
            except Exception:
                feedback.pushInfo('defusedxml already available.')
        return True

    if not _install_package("defusedxml", feedback):
        return False

    importlib.invalidate_caches()
    _add_user_site_to_path(feedback)

    defusedxml = _import_module("defusedxml", feedback)

    if defusedxml is None:
        if feedback:
            feedback.reportError(
                'defusedxml installation was attempted, but the module is still unavailable. '
                'Restart QGIS and try again.'
            )
        return False

    if feedback:
        try:
            feedback.pushInfo(f'defusedxml loaded after installation: {defusedxml.__file__}')
        except Exception:
            feedback.pushInfo('defusedxml loaded after installation.')

    return True