259 lines
7.0 KiB
Bash
259 lines
7.0 KiB
Bash
#!/usr/bin/env bash
|
||
# crypto_monitor 一键环境部署（Linux / macOS / Git Bash）
|
||
#
|
||
# 用法:
|
||
# bash deploy/setup_env.sh
|
||
# bash deploy/setup_env.sh --only binance,gate_bot
|
||
# bash deploy/setup_env.sh --skip-pm2
|
||
# bash deploy/setup_env.sh --recreate-venv
|
||
# bash deploy/setup_env.sh --install-system-deps # root + apt 时安装 python*-venv
|
||
#
|
||
set -e
|
||
set -u
|
||
# 避免 Windows CRLF 导致 set -euo pipefail 一行报错；pipefail 仅 bash 支持
|
||
if [ -n "${BASH_VERSION:-}" ]; then
|
||
set -o pipefail
|
||
fi
|
||
|
||
DEPLOY_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
REPO_ROOT="$(cd "${DEPLOY_DIR}/.." && pwd)"
|
||
REQ_FILE="${REPO_ROOT}/requirements.txt"
|
||
HUB_REQ="${REPO_ROOT}/manual_trading_hub/requirements.txt"
|
||
|
||
ONLY="all"
|
||
SKIP_PM2=0
|
||
SKIP_ENV_COPY=0
|
||
RECREATE_VENV=0
|
||
INSTALL_APT_DEPS=0
|
||
PY=""
|
||
|
||
usage() {
|
||
sed -n '2,12p' "$0" | sed 's/^# \?//'
|
||
exit "${1:-0}"
|
||
}
|
||
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
--only) ONLY="${2:-all}"; shift 2 ;;
|
||
--skip-pm2) SKIP_PM2=1; shift ;;
|
||
--skip-env-copy) SKIP_ENV_COPY=1; shift ;;
|
||
--recreate-venv) RECREATE_VENV=1; shift ;;
|
||
--install-system-deps) INSTALL_APT_DEPS=1; shift ;;
|
||
-h|--help) usage 0 ;;
|
||
*) echo "未知参数: $1" >&2; usage 1 ;;
|
||
esac
|
||
done
|
||
|
||
step() { echo ""; echo "==> $*"; }
|
||
|
||
should_include() {
|
||
local key="$1"
|
||
if [[ "${ONLY}" == "all" ]]; then
|
||
return 0
|
||
fi
|
||
local item
|
||
IFS=',' read -ra PARTS <<< "${ONLY}"
|
||
for item in "${PARTS[@]}"; do
|
||
item="$(echo "${item}" | tr '[:upper:]' '[:lower:]' | xargs)"
|
||
[[ "${item}" == "${key}" ]] && return 0
|
||
done
|
||
return 1
|
||
}
|
||
|
||
find_python() {
|
||
if command -v python3 >/dev/null 2>&1; then
|
||
echo python3
|
||
return
|
||
fi
|
||
if command -v python >/dev/null 2>&1; then
|
||
echo python
|
||
return
|
||
fi
|
||
echo "未找到 python3/python，请先安装 Python 3.10+" >&2
|
||
exit 1
|
||
}
|
||
|
||
check_python_version() {
|
||
local py="$1"
|
||
local ver
|
||
ver="$("${py}" -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')"
|
||
local major minor
|
||
major="${ver%%.*}"
|
||
minor="${ver#*.}"
|
||
if [[ "${major}" -lt 3 ]] || [[ "${major}" -eq 3 && "${minor}" -lt 10 ]]; then
|
||
echo "需要 Python 3.10+，当前: ${ver}" >&2
|
||
exit 1
|
||
fi
|
||
echo "Python: $("${py}" --version 2>&1)"
|
||
}
|
||
|
||
python_minor_version() {
|
||
local py="$1"
|
||
"${py}" -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")'
|
||
}
|
||
|
||
check_venv_available() {
|
||
local py="$1"
|
||
local tmp
|
||
tmp="$(mktemp -d 2>/dev/null || mktemp -d -t cmvenv)"
|
||
if "${py}" -m venv "${tmp}" >/dev/null 2>&1 && [[ -x "${tmp}/bin/python" ]]; then
|
||
rm -rf "${tmp}"
|
||
return 0
|
||
fi
|
||
rm -rf "${tmp}" 2>/dev/null || true
|
||
return 1
|
||
}
|
||
|
||
install_debian_venv_packages() {
|
||
local py="$1"
|
||
local ver
|
||
ver="$(python_minor_version "${py}")"
|
||
if ! command -v apt-get >/dev/null 2>&1; then
|
||
echo " 未检测到 apt-get，请手动安装 python${ver}-venv 与 python3-pip" >&2
|
||
return 1
|
||
fi
|
||
if [[ "$(id -u)" -ne 0 ]]; then
|
||
echo " 需要 root 安装系统包，请执行:" >&2
|
||
echo " sudo apt update && sudo apt install -y python${ver}-venv python3-pip curl" >&2
|
||
echo " 或: sudo bash deploy/setup_env.sh --install-system-deps" >&2
|
||
return 1
|
||
fi
|
||
step "安装系统依赖 (python${ver}-venv) ..."
|
||
export DEBIAN_FRONTEND=noninteractive
|
||
apt-get update -qq
|
||
if ! apt-get install -y "python${ver}-venv" python3-pip curl ca-certificates; then
|
||
apt-get install -y python3-venv python3-pip curl ca-certificates
|
||
fi
|
||
}
|
||
|
||
ensure_venv_prereqs() {
|
||
local py="$1"
|
||
if check_venv_available "${py}"; then
|
||
return 0
|
||
fi
|
||
echo " 当前 Python 无法创建 venv（缺少 ensurepip，常见于未安装 python*-venv）" >&2
|
||
if [[ "${INSTALL_APT_DEPS}" -eq 1 ]] || [[ "$(id -u)" -eq 0 ]]; then
|
||
install_debian_venv_packages "${py}" || exit 1
|
||
if check_venv_available "${py}"; then
|
||
return 0
|
||
fi
|
||
fi
|
||
local ver
|
||
ver="$(python_minor_version "${py}")"
|
||
echo "请安装后重试:" >&2
|
||
echo " apt update && apt install -y python${ver}-venv python3-pip" >&2
|
||
echo " bash deploy/setup_env.sh" >&2
|
||
exit 1
|
||
}
|
||
|
||
create_project_venv() {
|
||
local py="$1"
|
||
if [[ "${RECREATE_VENV}" -eq 1 && -d .venv ]]; then
|
||
echo " 删除旧 venv ..."
|
||
rm -rf .venv
|
||
fi
|
||
if [[ -d .venv && ! -x .venv/bin/python ]]; then
|
||
echo " 清理未完成的 venv ..."
|
||
rm -rf .venv
|
||
fi
|
||
if [[ -x .venv/bin/python ]]; then
|
||
return 0
|
||
fi
|
||
echo " 创建 venv ..."
|
||
if ! "${py}" -m venv .venv; then
|
||
rm -rf .venv 2>/dev/null || true
|
||
echo " venv 创建失败" >&2
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
setup_monitor() {
|
||
local dir_name="$1"
|
||
local proj="${REPO_ROOT}/${dir_name}"
|
||
if [[ ! -d "${proj}" ]]; then
|
||
echo " 跳过（目录不存在）: ${dir_name}"
|
||
return
|
||
fi
|
||
step "${dir_name}"
|
||
cd "${proj}"
|
||
create_project_venv "${PY}"
|
||
echo " 升级 pip ..."
|
||
.venv/bin/python -m pip install -U pip setuptools wheel -q
|
||
echo " 安装依赖 ..."
|
||
.venv/bin/pip install -r "${REQ_FILE}" -q
|
||
if [[ "${SKIP_ENV_COPY}" -eq 0 ]]; then
|
||
if [[ -f .env.example && ! -f .env ]]; then
|
||
cp -n .env.example .env 2>/dev/null || cp .env.example .env
|
||
echo " 已复制 .env.example -> .env"
|
||
elif [[ -f .env ]]; then
|
||
echo " 保留已有 .env"
|
||
else
|
||
echo " 无 .env.example，请手动配置 .env"
|
||
fi
|
||
fi
|
||
mkdir -p static/images/order_charts
|
||
echo " 完成: ${proj}/.venv/bin/python"
|
||
}
|
||
|
||
setup_hub() {
|
||
local proj="${REPO_ROOT}/manual_trading_hub"
|
||
if [[ ! -d "${proj}" ]]; then
|
||
echo " 跳过 hub（目录不存在）"
|
||
return
|
||
fi
|
||
step "manual_trading_hub"
|
||
cd "${proj}"
|
||
create_project_venv "${PY}"
|
||
.venv/bin/python -m pip install -U pip setuptools wheel -q
|
||
if [[ -f "${HUB_REQ}" ]]; then
|
||
.venv/bin/pip install -r "${HUB_REQ}" -q
|
||
fi
|
||
if [[ "${SKIP_ENV_COPY}" -eq 0 && -f .env.example && ! -f .env ]]; then
|
||
cp -n .env.example .env 2>/dev/null || cp .env.example .env
|
||
echo " 已复制 .env.example -> .env"
|
||
fi
|
||
echo " 完成: ${proj}/.venv/bin/python"
|
||
}
|
||
|
||
install_pm2() {
|
||
if [[ "${SKIP_PM2}" -eq 1 ]]; then
|
||
return
|
||
fi
|
||
step "PM2（可选）"
|
||
if ! command -v node >/dev/null 2>&1; then
|
||
echo " 未检测到 Node.js，跳过。安装后执行: npm install -g pm2"
|
||
return
|
||
fi
|
||
if command -v pm2 >/dev/null 2>&1; then
|
||
echo " PM2 已安装: $(pm2 -v)"
|
||
return
|
||
fi
|
||
echo " 正在安装 pm2 ..."
|
||
npm install -g pm2
|
||
echo " 各子目录: pm2 start ecosystem.config.cjs"
|
||
}
|
||
|
||
echo "crypto_monitor 环境部署"
|
||
echo "仓库根目录: ${REPO_ROOT}"
|
||
|
||
[[ -f "${REQ_FILE}" ]] || { echo "缺少 ${REQ_FILE}" >&2; exit 1; }
|
||
|
||
PY="$(find_python)"
|
||
check_python_version "${PY}"
|
||
ensure_venv_prereqs "${PY}"
|
||
|
||
should_include binance && setup_monitor crypto_monitor_binance
|
||
should_include gate && setup_monitor crypto_monitor_gate
|
||
should_include gate_bot && setup_monitor crypto_monitor_gate_bot
|
||
should_include okx && setup_monitor crypto_monitor_okx
|
||
should_include hub && setup_hub
|
||
|
||
install_pm2
|
||
|
||
echo ""
|
||
echo "部署完成。下一步："
|
||
echo " 1. 编辑各子目录 .env"
|
||
echo " 2. 启动示例: cd crypto_monitor_binance && source .venv/bin/activate && python app.py"
|
||
echo " 3. Windows 可用: powershell -File deploy/setup_env.ps1"
|
||
echo ""
|