dekun
211 lines
7.1 KiB
PowerShell
211 lines
7.1 KiB
PowerShell
#Requires -Version 5.1
|
||
<#
|
||
.SYNOPSIS
|
||
crypto_monitor 一键环境部署（Windows PowerShell）
|
||
|
||
.DESCRIPTION
|
||
- 为各子项目创建 Python venv 并安装依赖
|
||
- 从 .env.example 复制 .env（不覆盖已有）
|
||
- 创建 static/images 等运行时目录
|
||
- 可选安装 PM2（需已安装 Node.js）
|
||
|
||
.EXAMPLE
|
||
.\deploy\setup_env.ps1
|
||
.\deploy\setup_env.ps1 -Only binance,gate_bot
|
||
.\deploy\setup_env.ps1 -SkipPm2
|
||
#>
|
||
param(
|
||
[string]$Only = "all",
|
||
[switch]$SkipPm2,
|
||
[switch]$SkipEnvCopy,
|
||
[switch]$RecreateVenv
|
||
)
|
||
|
||
$ErrorActionPreference = "Stop"
|
||
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
|
||
|
||
$DeployDir = $PSScriptRoot
|
||
$RepoRoot = (Resolve-Path (Join-Path $DeployDir "..")).Path
|
||
$ReqFile = Join-Path $RepoRoot "requirements.txt"
|
||
$HubReqFile = Join-Path $RepoRoot "manual_trading_hub\requirements.txt"
|
||
|
||
$MonitorProjects = @(
|
||
@{ Key = "binance"; Dir = "crypto_monitor_binance" },
|
||
@{ Key = "gate"; Dir = "crypto_monitor_gate" },
|
||
@{ Key = "gate_bot"; Dir = "crypto_monitor_gate_bot" },
|
||
@{ Key = "okx"; Dir = "crypto_monitor_okx" }
|
||
)
|
||
$HubProject = @{ Key = "hub"; Dir = "manual_trading_hub" }
|
||
|
||
function Write-Step([string]$Msg) {
|
||
Write-Host ""
|
||
Write-Host "==> $Msg" -ForegroundColor Cyan
|
||
}
|
||
|
||
function Test-Python310 {
|
||
$py = Get-Command python -ErrorAction SilentlyContinue
|
||
if (-not $py) {
|
||
throw "未找到 python。请安装 Python 3.10+ 并加入 PATH：https://www.python.org/downloads/"
|
||
}
|
||
$verText = & python -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')"
|
||
$parts = $verText.Trim() -split "\."
|
||
$major = [int]$parts[0]
|
||
$minor = [int]$parts[1]
|
||
if ($major -lt 3 -or ($major -eq 3 -and $minor -lt 10)) {
|
||
throw "需要 Python 3.10+，当前: $verText"
|
||
}
|
||
Write-Host "Python: $(python --version 2>&1)" -ForegroundColor DarkGray
|
||
}
|
||
|
||
function Should-Include([string]$Key, [string[]]$Selected) {
|
||
if ($Selected -contains "all") { return $true }
|
||
return $Selected -contains $Key
|
||
}
|
||
|
||
function Setup-MonitorProject([hashtable]$Proj) {
|
||
$projPath = Join-Path $RepoRoot $Proj.Dir
|
||
if (-not (Test-Path $projPath)) {
|
||
Write-Host " 跳过（目录不存在）: $($Proj.Dir)" -ForegroundColor Yellow
|
||
return
|
||
}
|
||
Write-Step "$($Proj.Dir)"
|
||
Push-Location $projPath
|
||
try {
|
||
$venvDir = Join-Path $projPath ".venv"
|
||
$venvPy = Join-Path $venvDir "Scripts\python.exe"
|
||
$venvPip = Join-Path $venvDir "Scripts\pip.exe"
|
||
|
||
if ($RecreateVenv -and (Test-Path $venvDir)) {
|
||
Write-Host " 删除旧 venv ..."
|
||
Remove-Item -Recurse -Force $venvDir
|
||
}
|
||
if (-not (Test-Path $venvPy)) {
|
||
Write-Host " 创建 venv ..."
|
||
& python -m venv .venv
|
||
}
|
||
|
||
Write-Host " 升级 pip ..."
|
||
& $venvPy -m pip install -U pip setuptools wheel -q
|
||
|
||
Write-Host " 安装依赖 (requirements.txt) ..."
|
||
& $venvPip install -r $ReqFile -q
|
||
|
||
if (-not $SkipEnvCopy) {
|
||
$envExample = Join-Path $projPath ".env.example"
|
||
$envFile = Join-Path $projPath ".env"
|
||
if ((Test-Path $envExample) -and -not (Test-Path $envFile)) {
|
||
Copy-Item $envExample $envFile
|
||
Write-Host " 已复制 .env.example -> .env" -ForegroundColor Green
|
||
} elseif (Test-Path $envFile) {
|
||
Write-Host " 保留已有 .env" -ForegroundColor DarkGray
|
||
} else {
|
||
Write-Host " 无 .env.example，请手动配置 .env" -ForegroundColor Yellow
|
||
}
|
||
}
|
||
|
||
$staticDirs = @(
|
||
"static\images",
|
||
"static\images\order_charts"
|
||
)
|
||
foreach ($d in $staticDirs) {
|
||
$full = Join-Path $projPath $d
|
||
if (-not (Test-Path $full)) {
|
||
New-Item -ItemType Directory -Path $full -Force | Out-Null
|
||
}
|
||
}
|
||
Write-Host " 完成: $venvPy" -ForegroundColor Green
|
||
} finally {
|
||
Pop-Location
|
||
}
|
||
}
|
||
|
||
function Setup-HubProject() {
|
||
$projPath = Join-Path $RepoRoot $HubProject.Dir
|
||
if (-not (Test-Path $projPath)) {
|
||
Write-Host " 跳过 hub（目录不存在）" -ForegroundColor Yellow
|
||
return
|
||
}
|
||
Write-Step $HubProject.Dir
|
||
Push-Location $projPath
|
||
try {
|
||
$venvDir = Join-Path $projPath ".venv"
|
||
$venvPy = Join-Path $venvDir "Scripts\python.exe"
|
||
$venvPip = Join-Path $venvDir "Scripts\pip.exe"
|
||
|
||
if ($RecreateVenv -and (Test-Path $venvDir)) {
|
||
Remove-Item -Recurse -Force $venvDir
|
||
}
|
||
if (-not (Test-Path $venvPy)) {
|
||
& python -m venv .venv
|
||
}
|
||
& $venvPy -m pip install -U pip setuptools wheel -q
|
||
if (Test-Path $HubReqFile) {
|
||
& $venvPip install -r $HubReqFile -q
|
||
}
|
||
|
||
if (-not $SkipEnvCopy) {
|
||
$envExample = Join-Path $projPath ".env.example"
|
||
$envFile = Join-Path $projPath ".env"
|
||
if ((Test-Path $envExample) -and -not (Test-Path $envFile)) {
|
||
Copy-Item $envExample $envFile
|
||
Write-Host " 已复制 .env.example -> .env" -ForegroundColor Green
|
||
}
|
||
}
|
||
Write-Host " 完成: $venvPy" -ForegroundColor Green
|
||
} finally {
|
||
Pop-Location
|
||
}
|
||
}
|
||
|
||
function Install-Pm2IfNeeded() {
|
||
if ($SkipPm2) { return }
|
||
$node = Get-Command node -ErrorAction SilentlyContinue
|
||
if (-not $node) {
|
||
Write-Host "未检测到 Node.js，跳过 PM2。安装 Node 后执行: npm install -g pm2" -ForegroundColor Yellow
|
||
return
|
||
}
|
||
Write-Step "PM2（可选进程托管）"
|
||
$pm2 = Get-Command pm2 -ErrorAction SilentlyContinue
|
||
if ($pm2) {
|
||
Write-Host " PM2 已安装: $(pm2 -v)" -ForegroundColor Green
|
||
return
|
||
}
|
||
Write-Host " 正在全局安装 pm2 ..."
|
||
& npm install -g pm2
|
||
Write-Host " PM2 安装完成。在各子目录执行: pm2 start ecosystem.config.cjs" -ForegroundColor Green
|
||
}
|
||
|
||
# --- main ---
|
||
Write-Host "crypto_monitor 环境部署" -ForegroundColor White
|
||
Write-Host "仓库根目录: $RepoRoot" -ForegroundColor DarkGray
|
||
|
||
if (-not (Test-Path $ReqFile)) {
|
||
throw "缺少 $ReqFile"
|
||
}
|
||
|
||
Test-Python310
|
||
|
||
$selected = ($Only -split "[,;\s]+" | ForEach-Object { $_.Trim().ToLower() } | Where-Object { $_ })
|
||
if (-not $selected -or $selected.Count -eq 0) { $selected = @("all") }
|
||
|
||
foreach ($p in $MonitorProjects) {
|
||
if (Should-Include $p.Key $selected) {
|
||
Setup-MonitorProject $p
|
||
}
|
||
}
|
||
if (Should-Include $HubProject.Key $selected) {
|
||
Setup-HubProject
|
||
}
|
||
|
||
Install-Pm2IfNeeded
|
||
|
||
Write-Host ""
|
||
Write-Host "部署完成。下一步：" -ForegroundColor Green
|
||
Write-Host " 1. 编辑各子目录 .env（API Key、密码、代理等）"
|
||
Write-Host " 2. 启动示例（Binance）："
|
||
Write-Host " cd crypto_monitor_binance"
|
||
Write-Host " .\.venv\Scripts\activate"
|
||
Write-Host " python app.py"
|
||
Write-Host " 3. Linux 服务器可用: bash deploy/setup_env.sh"
|
||
Write-Host ""
|