endfield-stock-recognizer/main.spec at main · Logical-Byte/endfield-stock-recognizer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import importlib.resources
import urllib.request
from pathlib import Path
from typing import TYPE_CHECKING

from PyInstaller.utils.hooks import (
    collect_data_files,
)

if TYPE_CHECKING:
    from typing import cast

    from PyInstaller.building.api import EXE, PYZ  # noqa: TC004
    from PyInstaller.building.build_main import Analysis  # noqa: TC004
    from PyInstaller.config import CONF

    DISTPATH = cast("str", CONF["distpath"])


# 下载并解压模型
models_dir = Path("src/endfield_stock_recognizer/models")
models_dir.mkdir(parents=True, exist_ok=True)

# 下载识别模型
rec_model_url = "https://www.modelscope.cn/models/RapidAI/RapidOCR/resolve/v3.4.0/onnx/PP-OCRv5/rec/ch_PP-OCRv5_rec_mobile_infer.onnx"
rec_model_path = models_dir / "ch_PP-OCRv5_rec_mobile_infer.onnx"
if not rec_model_path.exists():
    print("Downloading recognition model...")
    urllib.request.urlretrieve(rec_model_url, rec_model_path)
    print("Recognition model downloaded.")


datas = []
binaries = []
datas.extend(collect_data_files("endfield_stock_recognizer"))
datas.extend(
    [
        (importlib.resources.files("rapidocr") / "config.yaml", "rapidocr"),
        (importlib.resources.files("rapidocr") / "default_models.yaml", "rapidocr"),
    ]
)


a = Analysis(
    ["src/endfield_stock_recognizer/__main__.py"],
    pathex=[],
    binaries=binaries,
    datas=datas,
    hiddenimports=[],
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=[],
    noarchive=False,
    optimize=0,
)
pyz = PYZ(a.pure)

# 删除不需要的 opencv_videoio_ffmpeg DLL
ffmpeg_dll_paths = list(
    Path(str(importlib.resources.files("cv2"))).rglob("opencv_videoio_ffmpeg*.dll")
)
modified_binaries = [
    (dest_name, src_name, typecode)
    for dest_name, src_name, typecode in a.binaries
    if Path(src_name) not in ffmpeg_dll_paths
]

exe = EXE(
    pyz,
    a.scripts,
    modified_binaries,
    a.datas,
    [],
    name="endfield-stock-recognizer",
    debug=False,
    bootloader_ignore_signals=False,
    strip=False,
    upx=True,
    upx_exclude=[],
    runtime_tmpdir=None,
    console=True,
    disable_windowed_traceback=False,
    argv_emulation=False,
    target_arch=None,
    codesign_identity=None,
    entitlements_file=None,
)