#!/bin/bash
# 从 qdzs 项目导入 Excel 数据到 Neo4j
# 便捷的一键导入脚本

set -e

echo "============================================================"
echo "从 qdzs 项目导入 Excel 数据到 Neo4j"
echo "============================================================"

# 检查 Neo4j 是否运行
echo "检查 Neo4j 服务状态..."
if ! curl -s http://localhost:7474 > /dev/null 2>&1; then
    echo "❌ Neo4j 服务未运行，正在启动..."
    /data/neo4j/start_neo4j.sh start
    echo "等待 Neo4j 启动..."
    sleep 5
else
    echo "✓ Neo4j 服务正在运行"
fi

# 检查 Excel 文件
EXCEL_FILE="/data/qdzs/public/merged_all_interests_20251203_133603.xlsx"
if [ ! -f "$EXCEL_FILE" ]; then
    echo "❌ Excel 文件不存在: $EXCEL_FILE"
    exit 1
fi
echo "✓ Excel 文件存在: $EXCEL_FILE"

# 检查 Python 依赖
echo ""
echo "检查 Python 依赖..."
python3 -c "import pandas; import neo4j" 2>/dev/null || {
    echo "❌ 缺少必要的 Python 包，请安装:"
    echo "   pip3 install pandas openpyxl neo4j"
    exit 1
}
echo "✓ Python 依赖检查通过"

# 运行导入脚本
echo ""
echo "开始导入数据..."
echo "============================================================"
python3 /data/neo4j/import/import_from_qdzs.py

echo ""
echo "============================================================"
echo "导入完成！"
echo "============================================================"

