graphate_entry.py

"""Точка входа для сборки PyInstaller (исполняемый файл graphate)."""

from __future__ import annotations

import importlib.util
import sys
from pathlib import Path


def _bootstrap_rasterio_serde() -> None:
    """В frozen-сборке ``rasterio.serde`` часто не попадает в PYZ: он импортируется только из Cython ``_base``.

    В ``graphate.spec`` кладём ``serde.py`` в ``_MEIPASS/rasterio/`` (datas); здесь
    регистрируем модуль до первого ``import rasterio``.
    """

    if not getattr(sys, "frozen", False):
        return
    base = Path(getattr(sys, "_MEIPASS", "") or "")
    if not base.is_dir():
        return
    serde_path = base / "rasterio" / "serde.py"
    if not serde_path.is_file():
        return
    if "rasterio.serde" in sys.modules:
        return
    spec = importlib.util.spec_from_file_location("rasterio.serde", serde_path)
    if spec is None or spec.loader is None:
        return
    mod = importlib.util.module_from_spec(spec)
    sys.modules["rasterio.serde"] = mod
    spec.loader.exec_module(mod)


def _run() -> int:
    from app.main import main

    return main()


if __name__ == "__main__":
    _bootstrap_rasterio_serde()
    sys.exit(_run())