# 錯誤示例：直接創(chuàng)建軟鏈接
ln -sf /usr/bin/python3 /usr/bin/python

# 結(jié)果：yum無法正常工作
yum install package
# 報錯：/usr/bin/python: No module named yum

# 用戶執(zhí)行了以下操作（?? 危險操作?。?
cd /usr/bin
rm python
ln -s python3 python

# 結(jié)果：系統(tǒng)瞬間崩潰！
yum update
# 錯誤：Traceback (most recent call last):
#   File "/usr/bin/yum", line 30, in <module>
#     import yum
# ImportError: No module named yum

# 系統(tǒng)無法更新，因為yum不可用
yum update
# 系統(tǒng)陷入無法更新的狀態(tài)
# 最終結(jié)果：只能重裝系統(tǒng)！??

#!/bin/bash

# CentOS7 Python3 一鍵安裝腳本（優(yōu)化版）
# 默認(rèn)使用 python 命令執(zhí)行 Python3，同時確保不影響 yum 等系統(tǒng)功能

set -e  # 遇到錯誤立即退出

# 顏色定義
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 國內(nèi)鏡像源配置
PYTHON_MIRROR="https://mirrors.huaweicloud.com/python/3.8.18/Python-3.8.18.tgz"
BACKUP_MIRROR="https://npm.taobao.org/mirrors/python/3.8.18/Python-3.8.18.tgz"

# 日志函數(shù)
log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

log_debug() {
    echo -e "${BLUE}[DEBUG]${NC} $1"
}

# 檢查文件完整性
check_file_integrity() {
    local file_path=$1
    
    if [[ ! -f "$file_path" ]]; then
        return 1
    fi
    
    if [[ ! -s "$file_path" ]]; then
        return 1
    fi
    
    if [[ "$file_path" == *.tgz ]] || [[ "$file_path" == *.tar.gz ]]; then
        if ! tar -tzf "$file_path" >/dev/null 2>&1; then
            return 1
        fi
    fi
    
    return 0
}

# 安全下載函數(shù)
safe_download() {
    local output=$1
    local max_retries=3
    local retry_count=0
    
    while [[ $retry_count -lt $max_retries ]]; do
        log_info "下載嘗試 $((retry_count + 1))/$max_retries"
        
        # 刪除可能存在的損壞文件
        rm -f "$output"
        
        # 使用 wget 下載，顯示進(jìn)度，設(shè)置超時
        if wget --progress=bar:force --timeout=30 -O "$output" "$PYTHON_MIRROR"; then
            # 驗證文件完整性
            if check_file_integrity "$output"; then
                log_info "下載完成且文件完整"
                return 0
            else
                log_warn "文件下載不完整，將重試..."
                rm -f "$output"
            fi
        else
            log_warn "下載失敗，將重試..."
        fi
        
        # 如果第一次失敗，嘗試備用鏡像
        if [[ $retry_count -eq 0 ]]; then
            log_info "嘗試備用鏡像源..."
            PYTHON_MIRROR="$BACKUP_MIRROR"
        fi
        
        retry_count=$((retry_count + 1))
        if [[ $retry_count -lt $max_retries ]]; then
            log_info "等待 3 秒后重試..."
            sleep 3
        fi
    done
    
    log_error "所有鏡像源下載失敗，請檢查網(wǎng)絡(luò)連接"
    return 1
}

# 檢查是否為 root 用戶
check_root() {
    if [[ $EUID -eq 0 ]]; then
        log_warn "腳本正在以 root 用戶運行，建議使用普通用戶執(zhí)行"
        read -p "是否繼續(xù)? (y/n): " -n 1 -r
        echo
        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
            exit 1
        fi
    fi
}

# 檢查操作系統(tǒng)
check_os() {
    if [[ ! -f /etc/redhat-release ]]; then
        log_error "此腳本僅支持 CentOS/RHEL 系統(tǒng)"
        exit 1
    fi
    
    CENTOS_VERSION=$(grep -oE '[0-9]+\.[0-9]+' /etc/redhat-release | cut -d. -f1)
    if [[ $CENTOS_VERSION -ne 7 ]]; then
        log_warn "檢測到系統(tǒng)版本: CentOS $CENTOS_VERSION，此腳本專為 CentOS 7 設(shè)計"
        read -p "是否繼續(xù)? (y/n): " -n 1 -r
        echo
        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
            exit 1
        fi
    fi
}

# 安裝依賴包
install_dependencies() {
    log_info "安裝必要的依賴包..."
    
    # 配置阿里云yum源加速下載
    if [[ -f /etc/yum.repos.d/CentOS-Base.repo ]]; then
        sudo cp /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.backup
    fi
    
    sudo curl -o /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo 2>/dev/null || true
    
    # 清理緩存
    sudo yum clean all
    sudo yum makecache
    
    # 安裝依賴
    sudo yum groupinstall "Development Tools" -y
    sudo yum install -y openssl-devel libffi-devel bzip2-devel sqlite-devel wget \
        zlib-devel ncurses-devel readline-devel gdbm-devel xz-devel tk-devel
    
    # 恢復(fù)原yum配置
    if [[ -f /etc/yum.repos.d/CentOS-Base.repo.backup ]]; then
        sudo mv /etc/yum.repos.d/CentOS-Base.repo.backup /etc/yum.repos.d/CentOS-Base.repo
    fi
    
    log_info "依賴包安裝完成"
}

# 清理舊的安裝文件
cleanup_old_files() {
    log_info "清理可能存在的損壞文件..."
    
    local python_version="3.8.18"
    local tar_file="/tmp/Python-${python_version}.tgz"
    local source_dir="/tmp/Python-${python_version}"
    
    if [[ -f "$tar_file" ]]; then
        if ! check_file_integrity "$tar_file"; then
            log_warn "刪除損壞的壓縮包: $tar_file"
            sudo rm -f "$tar_file"
        fi
    fi
    
    if [[ -d "$source_dir" ]]; then
        log_info "清理舊的源碼目錄: $source_dir"
        sudo rm -rf "$source_dir"
    fi
}

# 安裝Python 3.8
install_python() {
    local python_version="3.8.18"
    local install_dir="/usr/local/python38"
    local tar_file="/tmp/Python-${python_version}.tgz"
    local source_dir="/tmp/Python-${python_version}"
    
    log_info "開始安裝 Python ${python_version}..."
    
    cd /tmp
    
    # 清理舊文件
    cleanup_old_files
    
    # 下載Python源碼
    if [[ ! -f "$tar_file" ]]; then
        safe_download "$tar_file"
    else
        if check_file_integrity "$tar_file"; then
            log_info "使用已存在的完整源碼包"
        else
            log_warn "已存在的源碼包損壞，重新下載..."
            safe_download "$tar_file"
        fi
    fi
    
    # 驗證壓縮包完整性
    log_info "驗證壓縮包完整性..."
    if ! tar -tzf "$tar_file" >/dev/null 2>&1; then
        log_error "壓縮包驗證失敗，文件可能已損壞"
        exit 1
    fi
    
    # 解壓
    log_info "解壓源碼包..."
    if ! tar -xzf "$tar_file"; then
        log_error "解壓失敗，文件可能已損壞"
        exit 1
    fi
    
    cd "$source_dir"
    
    # 配置編譯選項
    log_info "配置編譯參數(shù)..."
    ./configure --prefix=${install_dir} --with-ssl --enable-shared --enable-optimizations
    
    # 獲取CPU核心數(shù)
    local core_num=$(nproc)
    log_info "使用 ${core_num} 個核心進(jìn)行編譯..."
    
    # 編譯和安裝
    make -j${core_num}
    sudo make altinstall
    
    # 配置共享庫
    echo "/usr/local/python38/lib" | sudo tee /etc/ld.so.conf.d/python38.conf >/dev/null
    sudo ldconfig
    
    log_info "Python ${python_version} 安裝完成"
}

# 創(chuàng)建安全的符號鏈接（確保不影響系統(tǒng)功能）
setup_symlinks() {
    local install_dir="/usr/local/python38"
    
    log_info "配置符號鏈接..."
    
    # 創(chuàng)建目錄（如果不存在）
    sudo mkdir -p /usr/local/bin
    
    # 刪除可能存在的舊鏈接（只刪除我們創(chuàng)建的鏈接）
    sudo rm -f /usr/local/bin/python3
    sudo rm -f /usr/local/bin/pip3
    sudo rm -f /usr/local/bin/python
    sudo rm -f /usr/local/bin/pip
    
    # 創(chuàng)建符號鏈接到 /usr/local/bin（用戶路徑）
    sudo ln -sf ${install_dir}/bin/python3.8 /usr/local/bin/python3
    sudo ln -sf ${install_dir}/bin/pip3.8 /usr/local/bin/pip3
    
    # 關(guān)鍵步驟：創(chuàng)建 python 和 pip 鏈接到 /usr/local/bin
    # 這樣用戶默認(rèn)會使用 Python3，但系統(tǒng)腳本仍然使用 /usr/bin/python (Python2.7)
    sudo ln -sf ${install_dir}/bin/python3.8 /usr/local/bin/python
    sudo ln -sf ${install_dir}/bin/pip3.8 /usr/local/bin/pip
    
    log_info "符號鏈接配置完成"
}

# 配置環(huán)境變量（優(yōu)化PATH順序）
setup_environment() {
    log_info "配置環(huán)境變量..."
    
    # 備份原bashrc
    cp ~/.bashrc ~/.bashrc.backup.$(date +%Y%m%d%H%M%S) 2>/dev/null || true
    
    # 移除可能存在的舊配置
    sed -i '/PYTHON_HOME/d' ~/.bashrc
    sed -i '/alias python=/d' ~/.bashrc
    sed -i '/alias pip=/d' ~/.bashrc
    sed -i '/\/usr\/local\/bin/d' ~/.bashrc
    
    # 添加新的環(huán)境變量配置
    cat >> ~/.bashrc << 'EOF'

# Python3 環(huán)境配置（由安裝腳本自動添加）
export PYTHON_HOME=/usr/local/python38
export PATH="/usr/local/bin:$PATH"
alias python='/usr/local/bin/python3'
alias pip='/usr/local/bin/pip3'
export LD_LIBRARY_PATH="$PYTHON_HOME/lib:$LD_LIBRARY_PATH"

# 確保系統(tǒng)Python2.7仍然可用于系統(tǒng)腳本
alias python2='/usr/bin/python'
alias pip2='/usr/bin/pip' 2>/dev/null || true
EOF
    
    # 立即生效
    export PATH="/usr/local/bin:$PATH"
    export PYTHON_HOME="/usr/local/python38"
    export LD_LIBRARY_PATH="$PYTHON_HOME/lib:$LD_LIBRARY_PATH"
    
    log_info "環(huán)境變量配置完成"
}

# 配置pip（使用國內(nèi)鏡像）
setup_pip() {
    log_info "配置pip國內(nèi)鏡像..."
    
    # 創(chuàng)建pip配置目錄
    mkdir -p ~/.pip
    
    # 配置國內(nèi)鏡像源
    cat > ~/.pip/pip.conf << 'EOF'
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host = pypi.tuna.tsinghua.edu.cn
timeout = 120
EOF

    # 升級pip
    log_info "升級pip到最新版本..."
    /usr/local/bin/pip3 install --upgrade pip
    
    log_info "pip配置完成"
}

# 驗證安裝和系統(tǒng)兼容性
verify_installation() {
    log_info "驗證安裝結(jié)果和系統(tǒng)兼容性..."
    
    echo -e "\n${GREEN}=== Python 版本驗證 ===${NC}"
    
    # 測試 python 命令
    echo -n "python 命令指向: "
    if command -v python >/dev/null; then
        python_version=$(python -V 2>&1)
        echo -e "${GREEN}$python_version${NC}"
    else
        echo -e "${RED}未找到${NC}"
    fi
    
    # 測試 python3 命令
    echo -n "python3 命令指向: "
    if command -v python3 >/dev/null; then
        python3_version=$(python3 -V 2>&1)
        echo -e "${GREEN}$python3_version${NC}"
    else
        echo -e "${RED}未找到${NC}"
    fi
    
    # 測試系統(tǒng) python2.7
    echo -n "系統(tǒng) Python2.7: "
    if [[ -f /usr/bin/python ]]; then
        sys_python_version=$(/usr/bin/python -V 2>&1)
        echo -e "${GREEN}$sys_python_version${NC}"
    else
        echo -e "${RED}未找到${NC}"
    fi
    
    echo -e "\n${GREEN}=== 系統(tǒng)功能驗證 ===${NC}"
    
    # 驗證 yum 功能
    echo -n "yum 功能: "
    if yum --version >/dev/null 2>&1; then
        echo -e "${GREEN}正常${NC}"
    else
        echo -e "${RED}異常${NC}"
    fi
    
    # 驗證 yum 使用的 Python 版本
    echo -n "yum 使用的 Python: "
    yum_python=$(grep -oP '^#!/usr/bin/\Kpython[0-9.]*' /usr/bin/yum 2>/dev/null || echo "未知")
    if [[ -n "$yum_python" && "$yum_python" != "未知" ]]; then
        if [[ -f "/usr/bin/$yum_python" ]]; then
            yum_python_version=$("/usr/bin/$yum_python" -V 2>&1)
            echo -e "${GREEN}$yum_python_version${NC}"
        else
            echo -e "${YELLOW}腳本指向的Python不存在${NC}"
        fi
    else
        echo -e "${YELLOW}無法確定${NC}"
    fi
    
    echo -e "\n${GREEN}=== 路徑優(yōu)先級驗證 ===${NC}"
    echo -n "which python: "
    which_python=$(which python)
    echo -e "${BLUE}$which_python${NC}"
    
    echo -n "which python3: "
    which_python3=$(which python3)
    echo -e "${BLUE}$which_python3${NC}"
    
    echo -e "\n${GREEN}=== 使用說明 ===${NC}"
    echo -e "默認(rèn) Python 版本: ${GREEN}Python 3.8.18${NC}"
    echo -e "系統(tǒng) Python2.7: ${GREEN}/usr/bin/python${NC} (yum等系統(tǒng)工具使用)"
    echo -e "重新加載環(huán)境: ${GREEN}source ~/.bashrc${NC}"
}

# 保護(hù)系統(tǒng)Python配置
protect_system_python() {
    log_info "檢查并保護(hù)系統(tǒng)Python配置..."
    
    # 檢查yum配置文件，確保它們使用系統(tǒng)Python
    local system_tools=("/usr/bin/yum" "/usr/libexec/urlgrabber-ext-down")
    
    for tool in "${system_tools[@]}"; do
        if [[ -f "$tool" ]]; then
            # 檢查shebang行
            local shebang=$(head -1 "$tool" 2>/dev/null)
            if [[ "$shebang" == *"/usr/bin/python"* ]]; then
                log_debug "$tool 已正確配置使用系統(tǒng)Python"
            else
                log_warn "$tool 可能未正確配置Python解釋器"
            fi
        fi
    done
    
    # 創(chuàng)建保護(hù)腳本，防止意外修改系統(tǒng)Python
    sudo tee /usr/local/bin/protect-system-python.sh >/dev/null << 'EOF'
#!/bin/bash
# 系統(tǒng)Python保護(hù)腳本

echo "警告：系統(tǒng)Python (/usr/bin/python) 被保護(hù)，不能修改"
echo "這是yum等系統(tǒng)工具依賴的Python 2.7版本"
echo "如需使用Python 3，請使用 'python3' 或配置PATH環(huán)境變量"
echo "當(dāng)前Python版本：$(/usr/bin/python -V 2>&1)"
EOF
    
    sudo chmod +x /usr/local/bin/protect-system-python.sh
    
    log_info "系統(tǒng)Python保護(hù)完成"
}

# 顯示安裝摘要
show_summary() {
    echo -e "\n${GREEN}=== 安裝摘要 ===${NC}"
    echo -e "? Python 3.8.18 安裝到: ${BLUE}/usr/local/python38${NC}"
    echo -e "? 符號鏈接創(chuàng)建到: ${BLUE}/usr/local/bin/${NC}"
    echo -e "? 用戶默認(rèn) python 命令: ${GREEN}Python 3.8.18${NC}"
    echo -e "? 系統(tǒng)保留 python2.7: ${GREEN}/usr/bin/python${NC}"
    echo -e "? yum等系統(tǒng)工具: ${GREEN}不受影響${NC}"
    echo -e "? pip配置: ${GREEN}國內(nèi)鏡像源${NC}"
}

# 主函數(shù)
main() {
    echo -e "${GREEN}"
    echo "=================================================="
    echo "  CentOS7 Python3 一鍵安裝腳本（優(yōu)化版）"
    echo "  默認(rèn)使用python命令執(zhí)行Python3，不影響系統(tǒng)功能"
    echo "=================================================="
    echo -e "${NC}"
    
    echo -e "${YELLOW}安裝目標(biāo)：${NC}"
    echo -e "  ? 安裝 Python 3.8.18"
    echo -e "  ? 設(shè)置 'python' 命令默認(rèn)指向 Python3"
    echo -e "  ? 確保 yum 等系統(tǒng)工具繼續(xù)使用 Python2.7"
    echo -e "  ? 配置 pip 使用國內(nèi)鏡像源"
    echo ""
    
    read -p "是否繼續(xù)安裝? (y/n): " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        log_info "安裝已取消"
        exit 0
    fi
    
    # 執(zhí)行安裝步驟
    check_root
    check_os
    install_dependencies
    install_python
    setup_symlinks
    setup_environment
    setup_pip
    protect_system_python
    verify_installation
    show_summary
    
    echo -e "\n${GREEN}=== 安裝完成 ===${NC}"
    log_info "請執(zhí)行以下命令使配置立即生效："
    echo -e "  ${YELLOW}source ~/.bashrc${NC}"
    echo ""
    log_info "驗證命令："
    echo -e "  ${YELLOW}python -V${NC}    # 應(yīng)該顯示 Python 3.8.18"
    echo -e "  ${YELLOW}python3 -V${NC}   # 應(yīng)該顯示 Python 3.8.18"
    echo -e "  ${YELLOW}/usr/bin/python -V${NC}  # 應(yīng)該顯示 Python 2.7.5"
    echo -e "  ${YELLOW}yum --version${NC}  # 應(yīng)該正常顯示版本信息"
}

# 腳本入口
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    main "$@"
fi

# 測試yum（CentOS/RHEL）
yum --version

# 測試apt（Debian/Ubuntu）
apt --version

# 測試系統(tǒng)Python
/usr/bin/python --version