from fastapi import APIRouter, Depends, HTTPException from sqlalchemy.orm import Session from typing import Dict from ...core.database import get_db from ...services.prediction_service import PredictionService from ...models.lottery import SSQLotteryBetRecord, DLTLotteryBetRecord, SSQLottery, DLTLottery import datetime import random from ...services.telegram_bot import send_message from ...services.analysis_service import LotteryAnalysisService import os router = APIRouter() @router.post("/train/{lottery_type}") def train_prediction_model( lottery_type: str, periods: int = 100, db: Session = Depends(get_db) ): """训练预测模型""" if lottery_type not in ['ssq', 'dlt']: raise HTTPException(status_code=400, detail="Invalid lottery type") service = PredictionService(db) return service.train_model(lottery_type, periods) @router.get("/predict/{lottery_type}") def predict_next_numbers( lottery_type: str, periods: int = 10, db: Session = Depends(get_db) ): """预测下一期号码""" if lottery_type not in ['ssq', 'dlt']: raise HTTPException(status_code=400, detail="Invalid lottery type") service = PredictionService(db) return service.predict_next_numbers(lottery_type, periods) @router.get("/pattern/{lottery_type}") def get_pattern_prediction( lottery_type: str, periods: int = 100, db: Session = Depends(get_db) ): """基于模式的预测""" if lottery_type not in ['ssq', 'dlt']: raise HTTPException(status_code=400, detail="Invalid lottery type") service = PredictionService(db) return service.get_pattern_based_prediction(lottery_type, periods) @router.get("/ensemble/{lottery_type}") def get_ensemble_prediction( lottery_type: str, periods: int = 100, db: Session = Depends(get_db) ): """集成预测""" if lottery_type not in ['ssq', 'dlt']: raise HTTPException(status_code=400, detail="Invalid lottery type") service = PredictionService(db) return service.get_ensemble_prediction(lottery_type, periods) @router.post("/recommend/today") def recommend_today(db: Session = Depends(get_db)): """ 每日推荐：一三六推荐大乐透，二四日推荐双色球，固定投注1注+推荐4注，共5注，推送内容合并快乐8。 """ # 固定投注号码（写死） FIXED_SSQ = {'red': [4, 6, 7, 12, 18, 19], 'blue': [9]} FIXED_DLT = {'red': [4, 6, 9, 18, 19], 'blue': [7, 12]} # 判断彩种 weekday = datetime.datetime.now().isoweekday() if weekday in [1, 3, 6]: lottery_type = 'dlt' BetModel = DLTLotteryBetRecord recommend_type = '集成预测-每日推荐' def number_format(red, blue): return ' '.join( f"{n:02d}" for n in red) + ' 蓝球：' + ' '.join(f"{n:02d}" for n in blue) LotteryModel = DLTLottery fixed = FIXED_DLT title = "你好，帮忙买下大乐透。" dlt_append = True else: lottery_type = 'ssq' BetModel = SSQLotteryBetRecord recommend_type = '集成预测-每日推荐' def number_format(red, blue): return ' '.join( f"{n:02d}" for n in red) + ' 蓝球：' + ' '.join(f"{n:02d}" for n in blue) LotteryModel = SSQLottery fixed = FIXED_SSQ title = "你好，帮忙买下双色球。" dlt_append = False # 生成batch_id today = datetime.datetime.now().strftime('%Y%m%d') batch_id = f"{today}-{random.randint(1000, 9999)}" # 1. 推荐4注（集成预测+智能选号补足） service = PredictionService(db) pred = service.get_ensemble_prediction(lottery_type, periods=200) if not pred.get('success'): raise HTTPException( status_code=500, detail=pred.get('message', '预测失败')) recs = pred.get('recommendations', []) results = [] used_numbers = set() # 先取集成预测推荐 for rec in recs: if len(results) >= 4: break red = rec.get('numbers') blue = [rec.get('blue')] if rec.get( 'blue') is not None else rec.get('blues', []) if not red or not blue: continue numbers = ','.join(f"{n:02d}" for n in red) + \ '|' + ','.join(f"{n:02d}" for n in blue) if numbers in used_numbers: continue used_numbers.add(numbers) last = db.query(LotteryModel).order_by( LotteryModel.open_time.desc()).first() issue = last.issue if last else '' bet = BetModel( batch_id=batch_id, issue=issue, numbers=numbers, recommend_type=rec.get('method', recommend_type) ) db.add(bet) db.commit() db.refresh(bet) results.append({ 'id': bet.id, 'batch_id': batch_id, 'issue': issue, 'numbers': numbers, 'red': red, 'blue': blue, 'recommend_type': rec.get('method', recommend_type) }) # 补足到4注 analysis_service = LotteryAnalysisService(db) while len(results) < 4: smart = analysis_service.generate_smart_numbers( lottery_type, strategy='balanced', count=1) if not smart: break red = smart[0].get('red_numbers') blue = smart[0].get('blue_numbers') numbers = ','.join(f"{n:02d}" for n in red) + \ '|' + ','.join(f"{n:02d}" for n in blue) if numbers in used_numbers: continue used_numbers.add(numbers) last = db.query(LotteryModel).order_by( LotteryModel.open_time.desc()).first() issue = last.issue if last else '' bet = BetModel( batch_id=batch_id, issue=issue, numbers=numbers, recommend_type='智能选号补足' ) db.add(bet) db.commit() db.refresh(bet) results.append({ 'id': bet.id, 'batch_id': batch_id, 'issue': issue, 'numbers': numbers, 'red': red, 'blue': blue, 'recommend_type': '智能选号补足' }) # 2. 组装推送内容（固定投注+4注推荐） msg_lines = [title] # 固定投注 msg_lines.append(f"红球：{' '.join(f'{n:02d}' for n in fixed['red'])} 蓝球：{ ' '.join(f'{n:02d}' for n in fixed['blue'])}") # 推荐号码 for rec in results: msg_lines.append(f"红球：{' '.join(f'{n:02d}' for n in rec['red'])} 蓝球：{ ' '.join(f'{n:02d}' for n in rec['blue'])}") if dlt_append: msg_lines.append("都追加，谢谢。") # 3. 快乐8推送（3注十选，合并到主消息） msg_lines.append("\n快乐8，单式，选十") for _ in range(3): nums = sorted(random.sample(range(1, 81), 10)) msg_lines.append(' '.join(f"{n:02d}" for n in nums)) # 4. 推送到Telegram try: token = os.getenv("TELEGRAM_BOT_TOKEN") chat_id = os.getenv("TELEGRAM_CHAT_ID") send_message('\n'.join(msg_lines), token=token, chat_id=chat_id) except Exception as e: print(f"[WARN] Telegram推送失败: {e}") return { 'success': True, 'lottery_type': lottery_type, 'batch_id': batch_id, 'recommend': results } @router.get("/recommend/today") def get_today_recommendations(db: Session = Depends(get_db)): """获取今日拼盘推荐""" try: today = datetime.datetime.now().strftime('%Y%m%d') # 查询今日的双色球推荐 ssq_recommendations = db.query(SSQLotteryBetRecord).filter( SSQLotteryBetRecord.batch_id.like(f'SSQ_{today}%') ).order_by(SSQLotteryBetRecord.created_at.desc()).all() # 查询今日的大乐透推荐 dlt_recommendations = db.query(DLTLotteryBetRecord).filter( DLTLotteryBetRecord.batch_id.like(f'DLT_{today}%') ).order_by(DLTLotteryBetRecord.created_at.desc()).all() # 确定当前彩票类型（根据推荐数量） if len(ssq_recommendations) > len(dlt_recommendations): current_lottery_type = 'ssq' today_recommendations = ssq_recommendations else: current_lottery_type = 'dlt' today_recommendations = dlt_recommendations # 格式化推荐数据 formatted_recommendations = [] for rec in today_recommendations: numbers_parts = rec.numbers.split('|') red_numbers = [int(n) for n in numbers_parts[0].split( ',')] if numbers_parts[0] else [] blue_numbers = [int(n) for n in numbers_parts[1].split(',')] if len( numbers_parts) > 1 and numbers_parts[1] else [] formatted_recommendations.append({ 'id': rec.id, 'batch_id': rec.batch_id, 'recommend_type': rec.recommend_type, 'numbers': rec.numbers, 'redNumbers': red_numbers, 'blueNumbers': blue_numbers, 'created_at': rec.created_at, 'is_winner': rec.is_winner, 'win_amount': rec.win_amount }) return { 'success': True, 'lottery_type': current_lottery_type, 'recommend': formatted_recommendations, 'count': len(formatted_recommendations) } except Exception as e: raise HTTPException(status_code=500, detail=f"获取今日推荐失败: {str(e)}")