import os import uuid import json import requests import io import threading import time from flask import Blueprint, request, jsonify, session, current_app from urllib.parse import quote from config import Config from extensions import s3_client, redis_client, db from models import GenerationRecord, User, SystemDict from middlewares.auth import login_required from services.logger import system_logger api_bp = Blueprint('api', __name__) def sync_images_background(app, record_id, raw_urls): """后台同步图片至 MinIO，带重试机制""" with app.app_context(): minio_urls = [] for raw_url in raw_urls: success = False for attempt in range(3): # 3 次重试机制 try: img_resp = requests.get(raw_url, timeout=30) if img_resp.status_code == 200: filename = f"gen-{uuid.uuid4().hex}.png" s3_client.upload_fileobj( io.BytesIO(img_resp.content), Config.MINIO["bucket"], filename, ExtraArgs={"ContentType": "image/png"} ) minio_urls.append(f"{Config.MINIO['public_url']}{quote(filename)}") success = True break except Exception as e: print(f"⚠️ 第 {attempt+1} 次同步失败: {e}") time.sleep(2 ** attempt) # 指数退避 if not success: # 如果最终失败，保留原始 URL 以便至少有内容可看 minio_urls.append(raw_url) # 更新数据库记录为持久化 URL try: record = GenerationRecord.query.get(record_id) if record: record.image_urls = json.dumps(minio_urls) db.session.commit() print(f"✅ 记录 {record_id} 图片已完成持久化同步") except Exception as e: print(f"❌ 更新记录失败: {e}") @api_bp.route('/api/config') def get_config(): """从本地数据库字典获取配置""" try: dicts = SystemDict.query.filter_by(is_active=True).order_by(SystemDict.sort_order.desc()).all() config = { "models": [], "ratios": [], "prompts": [], "sizes": [] } for d in dicts: item = {"label": d.label, "value": d.value} if d.dict_type == 'ai_model': item["cost"] = d.cost config["models"].append(item) elif d.dict_type == 'aspect_ratio': config["ratios"].append(item) elif d.dict_type == 'prompt_tpl': config["prompts"].append(item) elif d.dict_type == 'ai_image_size': config["sizes"].append(item) return jsonify(config) except Exception as e: return jsonify({"error": str(e)}), 500 @api_bp.route('/api/upload', methods=['POST']) @login_required def upload(): try: files = request.files.getlist('images') img_urls = [] for f in files: ext = os.path.splitext(f.filename)[1] filename = f"{uuid.uuid4().hex}{ext}" s3_client.upload_fileobj( f, Config.MINIO["bucket"], filename, ExtraArgs={"ContentType": f.content_type} ) img_urls.append(f"{Config.MINIO['public_url']}{quote(filename)}") return jsonify({"urls": img_urls}) except Exception as e: return jsonify({"error": str(e)}), 500 @api_bp.route('/api/generate', methods=['POST']) @login_required def generate(): try: user_id = session.get('user_id') user = User.query.get(user_id) data = request.json if request.is_json else request.form mode = data.get('mode', 'trial') is_premium = data.get('is_premium', False) input_key = data.get('apiKey') target_api = Config.AI_API api_key = None use_trial = False if mode == 'key': # 自定义 Key 模式：优先使用本次输入的，否则使用数据库存的 api_key = input_key or user.api_key if not api_key: return jsonify({"error": "请先输入您的 API 密钥"}), 400 else: # 积分/试用模式 if user.points > 0: # 核心修复：优质模式使用专属 Key api_key = Config.PREMIUM_KEY if is_premium else Config.TRIAL_KEY target_api = Config.TRIAL_API use_trial = True else: return jsonify({"error": "可用积分已耗尽，请充值或切换至自定义 Key 模式"}), 400 # 如果是 Key 模式且输入了新 Key，则自动更新到数据库保存 if mode == 'key' and input_key and input_key != user.api_key: user.api_key = input_key db.session.commit() # 获取模型及对应的消耗积分 model_value = data.get('model') model_dict = SystemDict.query.filter_by(dict_type='ai_model', value=model_value).first() cost = model_dict.cost if model_dict else 1 # 核心修复：优质模式积分消耗 X2 if use_trial and is_premium: cost *= 2 # --- 积分预扣除逻辑 (点击即扣) --- if use_trial: if user.points < cost: return jsonify({"error": f"可用积分不足，优质模式需要 {cost} 积分，您当前剩余 {user.points} 积分"}), 400 user.points -= cost db.session.commit() system_logger.info(f"积分预扣除 ({'优质' if is_premium else '普通'}试用)", phone=user.phone, cost=cost, remaining_points=user.points) try: prompt = data.get('prompt') model = model_value ratio = data.get('ratio') size = data.get('size') input_img_urls = data.get('image_urls', []) payload = { "prompt": prompt, "model": model, "response_format": "url", "aspect_ratio": ratio, "image": input_img_urls } if model == "nano-banana-2" and size: payload["image_size"] = size headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"} resp = requests.post(target_api, json=payload, headers=headers, timeout=300) if resp.status_code != 200: # API 报错，退还积分 if use_trial: user.points += cost db.session.commit() system_logger.warning(f"API 报错，积分已退还", phone=user.phone, status_code=resp.status_code) return jsonify({"error": resp.text}), resp.status_code api_result = resp.json() raw_urls = [item['url'] for item in api_result.get('data', [])] # 立即写入数据库（先存原始 URL） new_record = GenerationRecord( user_id=session.get('user_id'), prompt=prompt, model=model, image_urls=json.dumps(raw_urls) ) db.session.add(new_record) db.session.commit() # 写入系统日志 system_logger.info( f"用户生成图片成功", phone=user.phone, model=model, record_id=new_record.id ) # 启动后台线程同步图片，不阻塞前端返回 app = current_app._get_current_object() threading.Thread( target=sync_images_background, args=(app, new_record.id, raw_urls) ).start() # 立即返回原始 URL 给前端展示 return jsonify({ "data": [{"url": url} for url in raw_urls], "message": "生成成功！作品正在后台同步至云存储。" }) except Exception as e: # 发生系统异常，退还积分 if use_trial: user.points += cost db.session.commit() system_logger.error(f"生成异常，积分已退还", phone=user.phone, error=str(e)) return jsonify({"error": str(e)}), 500 except Exception as e: return jsonify({"error": str(e)}), 500 @api_bp.route('/api/history', methods=['GET']) @login_required def get_history(): """获取用户的历史生成记录 (支持分页，限 90 天内)""" try: from datetime import datetime, timedelta user_id = session.get('user_id') page = request.args.get('page', 1, type=int) per_page = request.args.get('per_page', 10, type=int) # 计算 90 天前的时间 ninety_days_ago = datetime.utcnow() - timedelta(days=90) pagination = GenerationRecord.query.filter( GenerationRecord.user_id == user_id, GenerationRecord.created_at >= ninety_days_ago ).order_by(GenerationRecord.created_at.desc())\ .paginate(page=page, per_page=per_page, error_out=False) return jsonify({ "history": [{ "id": r.id, "prompt": r.prompt, "model": r.model, "urls": json.loads(r.image_urls), "time": r.created_at.strftime('%Y-%m-%d %H:%M') } for r in pagination.items], "has_next": pagination.has_next, "total": pagination.total }) except Exception as e: return jsonify({"error": str(e)}), 500