mirror of
https://github.com/NaiboWang/EasySpider.git
synced 2026-08-28 23:20:41 +08:00
Sora
48 lines
1.8 KiB
Bash
Executable File
48 lines
1.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
|
REPO_ROOT="$(cd -- "$SCRIPT_DIR/.." && pwd)"
|
|
PYTHON_BIN="${PYTHON_BIN:-python3}"
|
|
|
|
if [[ "$(uname -s)" != "Linux" || "$(uname -m)" != "x86_64" ]]; then
|
|
echo "This script must run on a native Linux x86_64 build host." >&2
|
|
exit 1
|
|
fi
|
|
|
|
PYTHON_VERSION="$($PYTHON_BIN -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')"
|
|
PYTHON_ARCH="$($PYTHON_BIN -c 'import platform; print(platform.machine())')"
|
|
if [[ "$PYTHON_VERSION" != "3.11" || "$PYTHON_ARCH" != "x86_64" ]]; then
|
|
echo "Python 3.11 x86_64 is required; found Python $PYTHON_VERSION $PYTHON_ARCH." >&2
|
|
exit 1
|
|
fi
|
|
|
|
DDDDOCR_MODEL="$($PYTHON_BIN -c 'import ddddocr, pathlib; print(pathlib.Path(ddddocr.__file__).resolve().parent / "common_old.onnx")')"
|
|
ONNX_STATE="$($PYTHON_BIN -c 'import onnxruntime, pathlib; capi = pathlib.Path(onnxruntime.__file__).resolve().parent / "capi"; matches = sorted(capi.glob("onnxruntime_pybind11_state*.so")); print(matches[0] if matches else "")')"
|
|
|
|
if [[ ! -f "$DDDDOCR_MODEL" ]]; then
|
|
echo "Could not find ddddocr/common_old.onnx in the active Python environment." >&2
|
|
exit 1
|
|
fi
|
|
if [[ ! -f "$ONNX_STATE" ]]; then
|
|
echo "Could not find onnxruntime_pybind11_state*.so in the active Python environment." >&2
|
|
exit 1
|
|
fi
|
|
|
|
cd "$SCRIPT_DIR"
|
|
rm -rf build dist
|
|
|
|
"$PYTHON_BIN" -m PyInstaller \
|
|
--noconfirm \
|
|
--clean \
|
|
--onefile \
|
|
--hidden-import selenium.webdriver.ie.webdriver \
|
|
--add-binary "$ONNX_STATE:onnxruntime/capi" \
|
|
--add-data "$DDDDOCR_MODEL:ddddocr" \
|
|
easyspider_executestage.py
|
|
|
|
TARGET_DIR="$REPO_ROOT/ElectronJS/chrome_linux64"
|
|
mkdir -p "$TARGET_DIR"
|
|
install -m 755 dist/easyspider_executestage "$TARGET_DIR/easyspider_executestage"
|
|
echo "Linux x64 execution stage packaged successfully."
|