#!/bin/bash
# 用户购课完课统计定时任务脚本
# 执行时间：每周一9:00
# 输出：Excel报表，发送到指定群

# 配置参数
DB_HOST="bj-postgres-16pob4sg.sql.tencentcdb.com"
DB_PORT="28591"
DB_NAME="vala_bi"
DB_USER="ai_member"
DB_PASS="LdfjdjL83h3h3^$&**YGG*"
SQL_PATH="/root/.openclaw/workspace/scripts/用户购课完课统计.sql"
OUTPUT_CSV="/tmp/user_course_stat_$(date +%Y%m%d).csv"
OUTPUT_EXCEL="/root/.openclaw/workspace/用户购课完课统计_$(date +%Y%m%d).xlsx"
TARGET_CHAT="chat:oc_af81515caefe26918736ad1941286224"
LOG_FILE="/var/log/user_course_stat.log"

# 记录日志
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 开始执行用户购课完课统计任务" >> $LOG_FILE

# 1. 运行SQL导出CSV
export PGPASSWORD=$DB_PASS
psql -h $DB_HOST -p $DB_PORT -U $DB_USER -d $DB_NAME --csv -f $SQL_PATH > $OUTPUT_CSV 2>> $LOG_FILE

if [ $? -ne 0 ]; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] SQL执行失败" >> $LOG_FILE
    exit 1
fi

# 2. CSV转Excel
python3 - << 'EOF'
import pandas as pd
import numpy as np
try:
    np._get_promotion_state = lambda *args, **kwargs: 0
except:
    pass
df = pd.read_csv('$OUTPUT_CSV', low_memory=False)
with pd.ExcelWriter('$OUTPUT_EXCEL') as writer:
    df.to_excel(writer, sheet_name='用户统计数据', index=False)
EOF

if [ $? -ne 0 ]; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] Excel生成失败" >> $LOG_FILE
    exit 1
fi

# 3. 发送到指定群
openclaw message send --channel feishu --target $TARGET_CHAT --message "【每周定时统计】$(date +%Y年%m月%d日)用户购课与完课情况统计，请查收附件。" --media $OUTPUT_EXCEL 2>> $LOG_FILE

if [ $? -ne 0 ]; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] 消息发送失败" >> $LOG_FILE
    exit 1
fi

# 清理临时文件
rm -f $OUTPUT_CSV
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 任务执行完成，文件已发送到群" >> $LOG_FILE
exit 0