Merge pull request #319053 from pluiedev/init/rapidocr_onnxruntime

{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,

  fetchzip,

  substitute,

  pythonRelaxDepsHook,

  pytestCheckHook,

  setuptools,

  pyclipper,

  opencv4,

  numpy,

  six,

  shapely,

  pyyaml,

  pillow,

  onnxruntime,

}:

let

  version = "1.3.22";

  src = fetchFromGitHub {

    owner = "RapidAI";

    repo = "RapidOCR";

    rev = "v${version}";

    hash = "sha256-8h4DFhnI9imr+bYQZdlrl8UKUdpwnGK+SGxLXSMmcag=";

  };

  models = fetchzip {

    url = "https://github.com/RapidAI/RapidOCR/releases/download/v1.1.0/required_for_whl_v1.3.0.zip";

    hash = "sha256-j/0nzyvu/HfNTt5EZ+2Phe5dkyPOdQw/OZTz0yS63aA=";

    stripRoot = false;

  } + "/required_for_whl_v1.3.0/resources/models";

in

buildPythonPackage {

  pname = "rapidocr-onnxruntime";

  inherit version src;

  pyproject = true;

  sourceRoot = "${src.name}/python";

  # HACK:

  # Upstream uses a very unconventional structure to organize the packages, and we have to coax the

  # existing infrastructure to work with it.

  # See https://github.com/RapidAI/RapidOCR/blob/02829ef986bc2a5c4f33e9c45c9267bcf2d07a1d/.github/workflows/gen_whl_to_pypi_rapidocr_ort.yml#L80-L92

  # for the "intended" way of building this package.

  # The setup.py supplied by upstream tries to determine the current version by

  # fetching the latest version of the package from PyPI, and then bumping the version number.

  # This is not allowed in the Nix build environment as we do not have internet access,

  # hence we patch that out and get the version from the build environment directly.

  patches = [

    (substitute {

      src = ./setup-py-override-version-checking.patch;

      substitutions = [

        "--subst-var-by"

        "version"

        version

      ];

    })

  ];

  postPatch = ''

    mv setup_onnxruntime.py setup.py

    mkdir -p rapidocr_onnxruntime/models

    ln -s ${models}/* rapidocr_onnxruntime/models

    # Magic patch from upstream - what does this even do??

    echo "from .rapidocr_onnxruntime.main import RapidOCR, VisRes" > __init__.py

  '';

  nativeBuildInputs = [ pythonRelaxDepsHook ];

  # Upstream expects the source files to be under rapidocr_onnxruntime/rapidocr_onnxruntime

  # instead of rapidocr_onnxruntime for the wheel to build correctly.

  preBuild = ''

    mkdir rapidocr_onnxruntime_t

    mv rapidocr_onnxruntime rapidocr_onnxruntime_t

    mv rapidocr_onnxruntime_t rapidocr_onnxruntime

  '';

  # Revert the above hack

  postBuild = ''

    mv rapidocr_onnxruntime rapidocr_onnxruntime_t

    mv rapidocr_onnxruntime_t/* .

  '';

  build-system = [ setuptools ];

  dependencies = [

    pyclipper

    opencv4

    numpy

    six

    shapely

    pyyaml

    pillow

    onnxruntime

  ];

  # Remove because we have adopted the `opencv4` as an attribute name.

  pythonRemoveDeps = [ "opencv-python" ];

  pythonImportsCheck = [ "rapidocr_onnxruntime" ];

  nativeCheckInputs = [ pytestCheckHook ];

  # These are tests for different backends.

  disabledTestPaths = [

    "tests/test_vino.py"

    "tests/test_paddle.py"

  ];

  meta = {

    # This seems to be related to https://github.com/microsoft/onnxruntime/issues/10038

    # Also some related issue: https://github.com/NixOS/nixpkgs/pull/319053#issuecomment-2167713362

    broken = (stdenv.isLinux && stdenv.isAarch64);

    changelog = "https://github.com/RapidAI/RapidOCR/releases/tag/v${version}";

    description = "Cross platform OCR Library based on OnnxRuntime";

    homepage = "https://github.com/RapidAI/RapidOCR";

    license = with lib.licenses; [ asl20 ];

    maintainers = with lib.maintainers; [ pluiedev ];

    mainProgram = "rapidocr_onnxruntime";

  };

}

diff --git a/setup_onnxruntime.py b/setup_onnxruntime.py

index cd698d1..c6e8a33 100644

--- a/setup_onnxruntime.py

+++ b/setup_onnxruntime.py

@@ -6,7 +6,6 @@ from pathlib import Path

 from typing import List, Union

 import setuptools

-from get_pypi_latest_version import GetPyPiLatestVersion

 def read_txt(txt_path: Union[Path, str]) -> List[str]:

@@ -26,16 +25,7 @@ def get_readme():

 MODULE_NAME = "rapidocr_onnxruntime"

-obtainer = GetPyPiLatestVersion()

-latest_version = obtainer(MODULE_NAME)

-VERSION_NUM = obtainer.version_add_one(latest_version)

-

-if len(sys.argv) > 2:

-    match_str = " ".join(sys.argv[2:])

-    matched_versions = obtainer.extract_version(match_str)

-    if matched_versions:

-        VERSION_NUM = matched_versions

-sys.argv = sys.argv[:2]

+VERSION_NUM = "@version@"

 project_urls = {

     "Documentation": "https://rapidai.github.io/RapidOCRDocs/docs/install_usage/rapidocr/usage/",

  rapidgzip = callPackage ../development/python-modules/rapidgzip { inherit (pkgs) nasm; };

  rapidocr-onnxruntime = callPackage ../development/python-modules/rapidocr-onnxruntime { };

  rapt-ble = callPackage ../development/python-modules/rapt-ble { };

  rarfile = callPackage ../development/python-modules/rarfile {