commit 3f31d64ddf68deedd17ea1d6698f958aa2a61020 · pyrox.dev/nixpkgs

+133

pkgs/development/python-modules/rapidocr-onnxruntime/default.nix

···

       1
       1
       +
       {

     

       2
       2
       +
         lib,

     

       3
       3
       +
         buildPythonPackage,

     

       4
       4
       +
         fetchFromGitHub,

     

       5
       5
       +
       

     

       6
       6
       +
         fetchzip,

     

       7
       7
       +
         replaceVars,

     

       8
       8
       +
       

     

       9
       9
       +
         setuptools,

     

       10
       10
       +
         pyclipper,

     

       11
       11
       +
         opencv-python,

     

       12
       12
       +
         numpy,

     

       13
       13
       +
         six,

     

       14
       14
       +
         shapely,

     

       15
       15
       +
         pyyaml,

     

       16
       16
       +
         pillow,

     

       17
       17
       +
         onnxruntime,

     

       18
       18
       +
         tqdm,

     

       19
       19
       +
       

     

       20
       20
       +
         pytestCheckHook,

     

       21
       21
       +
         requests,

     

       22
       22
       +
       }:

     

       23
       23
       +
       let

     

       24
       24
       +
         version = "1.4.4";

     

       25
       25
       +
       

     

       26
       26
       +
         src = fetchFromGitHub {

     

       27
       27
       +
           owner = "RapidAI";

     

       28
       28
       +
           repo = "RapidOCR";

     

       29
       29
       +
           tag = "v${version}";

     

       30
       30
       +
           hash = "sha256-x0VELDKOffxbV3v0aDFJFuDC4YfsGM548XWgINmRc3M=";

     

       31
       31
       +
         };

     

       32
       32
       +
       

     

       33
       33
       +
         models =

     

       34
       34
       +
           fetchzip {

     

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

     

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

     

       37
       37
       +
             stripRoot = false;

     

       38
       38
       +
           }

     

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

     

       40
       40
       +
       in

     

       41
       41
       +
       buildPythonPackage {

     

       42
       42
       +
         pname = "rapidocr-onnxruntime";

     

       43
       43
       +
         inherit version src;

     

       44
       44
       +
         pyproject = true;

     

       45
       45
       +
       

     

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

     

       47
       47
       +
       

     

       48
       48
       +
         # HACK:

     

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

     

       50
       50
       +
         # existing infrastructure to work with it.

     

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

     

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

     

       53
       53
       +
       

     

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

     

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

     

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

     

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

     

       58
       58
       +
         patches = [

     

       59
       59
       +
           (replaceVars ./setup-py-override-version-checking.patch {

     

       60
       60
       +
             inherit version;

     

       61
       61
       +
           })

     

       62
       62
       +
         ];

     

       63
       63
       +
       

     

       64
       64
       +
         postPatch = ''

     

       65
       65
       +
           mv setup_onnxruntime.py setup.py

     

       66
       66
       +
       

     

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

     

       68
       68
       +
       

     

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

     

       70
       70
       +
         '';

     

       71
       71
       +
       

     

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

     

       73
       73
       +
         # instead of rapidocr_onnxruntime for the wheel to build correctly.

     

       74
       74
       +
         preBuild = ''

     

       75
       75
       +
           mkdir rapidocr_onnxruntime_t

     

       76
       76
       +
           mv rapidocr_onnxruntime rapidocr_onnxruntime_t

     

       77
       77
       +
           mv rapidocr_onnxruntime_t rapidocr_onnxruntime

     

       78
       78
       +
         '';

     

       79
       79
       +
       

     

       80
       80
       +
         # Revert the above hack

     

       81
       81
       +
         postBuild = ''

     

       82
       82
       +
           mv rapidocr_onnxruntime rapidocr_onnxruntime_t

     

       83
       83
       +
           mv rapidocr_onnxruntime_t/* .

     

       84
       84
       +
         '';

     

       85
       85
       +
       

     

       86
       86
       +
         build-system = [ setuptools ];

     

       87
       87
       +
       

     

       88
       88
       +
         dependencies = [

     

       89
       89
       +
           pyclipper

     

       90
       90
       +
           opencv-python

     

       91
       91
       +
           numpy

     

       92
       92
       +
           six

     

       93
       93
       +
           shapely

     

       94
       94
       +
           pyyaml

     

       95
       95
       +
           pillow

     

       96
       96
       +
           onnxruntime

     

       97
       97
       +
           tqdm

     

       98
       98
       +
         ];

     

       99
       99
       +
       

     

       100
       100
       +
         pythonImportsCheck = [ "rapidocr_onnxruntime" ];

     

       101
       101
       +
       

     

       102
       102
       +
         nativeCheckInputs = [

     

       103
       103
       +
           pytestCheckHook

     

       104
       104
       +
           requests

     

       105
       105
       +
         ];

     

       106
       106
       +
       

     

       107
       107
       +
         # These are tests for different backends.

     

       108
       108
       +
         disabledTestPaths = [

     

       109
       109
       +
           "tests/test_vino.py"

     

       110
       110
       +
           "tests/test_paddle.py"

     

       111
       111
       +
         ];

     

       112
       112
       +
       

     

       113
       113
       +
         disabledTests = [

     

       114
       114
       +
           # Needs Internet access

     

       115
       115
       +
           "test_long_img"

     

       116
       116
       +
         ];

     

       117
       117
       +
       

     

       118
       118
       +
         # rapidocr-onnxruntime has been renamed to rapidocr by upstream since 2.0.0. However, some packages like open-webui still requires rapidocr-onnxruntime 1.4.4. Therefore we set no auto update here.

     

       119
       119
       +
         # nixpkgs-update: no auto update

     

       120
       120
       +
         passthru.skipBulkUpdate = true;

     

       121
       121
       +
       

     

       122
       122
       +
         meta = {

     

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

     

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

     

       125
       125
       +
           badPlatforms = [ "aarch64-linux" ];

     

       126
       126
       +
           changelog = "https://github.com/RapidAI/RapidOCR/releases/tag/${src.tag}";

     

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

     

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

     

       129
       129
       +
           license = lib.licenses.asl20;

     

       130
       130
       +
           maintainers = with lib.maintainers; [ wrvsrx ];

     

       131
       131
       +
           mainProgram = "rapidocr_onnxruntime";

     

       132
       132
       +
         };

     

       133
       133
       +
       }

+31

pkgs/development/python-modules/rapidocr-onnxruntime/setup-py-override-version-checking.patch

···

       1
       1
       +
       diff --git i/setup_onnxruntime.py w/setup_onnxruntime.py

     

       2
       2
       +
       index e9572e9..f5f3b32 100644

     

       3
       3
       +
       --- i/setup_onnxruntime.py

     

       4
       4
       +
       +++ w/setup_onnxruntime.py

     

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

     

       6
       6
       +
        from typing import List, Union

     

       7
       7
       +
       

     

       8
       8
       +
        import setuptools

     

       9
       9
       +
       -from get_pypi_latest_version import GetPyPiLatestVersion

     

       10
       10
       +
       

     

       11
       11
       +
       

     

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

     

       13
       13
       +
       @@ -25,17 +24,7 @@ def get_readme():

     

       14
       14
       +
       

     

       15
       15
       +
       

     

       16
       16
       +
        MODULE_NAME = "rapidocr_onnxruntime"

     

       17
       17
       +
       -

     

       18
       18
       +
       -obtainer = GetPyPiLatestVersion()

     

       19
       19
       +
       -latest_version = obtainer(MODULE_NAME)

     

       20
       20
       +
       -VERSION_NUM = obtainer.version_add_one(latest_version, add_patch=True)

     

       21
       21
       +
       -

     

       22
       22
       +
       -if len(sys.argv) > 2:

     

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

     

       24
       24
       +
       -    matched_versions = obtainer.extract_version(match_str)

     

       25
       25
       +
       -    if matched_versions:

     

       26
       26
       +
       -        VERSION_NUM = matched_versions

     

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

     

       28
       28
       +
       +VERSION_NUM = "@version@"

     

       29
       29
       +
       

     

       30
       30
       +
        project_urls = {

     

       31
       31
       +
            "Documentation": "https://rapidai.github.io/RapidOCRDocs",

pkgs/top-level/python-packages.nix

···

       15555
       15555
        
       

     

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

     

       15557
       15557
        
       

     

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

     

       15559
       15559
       +
       

     

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

     

       15559
       15561
        
       

     

       15560
       15562
        
         rarfile = callPackage ../development/python-modules/rarfile { inherit (pkgs) libarchive; };