159 lines
4.9 KiB
Python
159 lines
4.9 KiB
Python
"""
|
|
产品审核 API
|
|
"""
|
|
import uuid
|
|
import json
|
|
from datetime import datetime
|
|
from flask import Blueprint, request, jsonify, session
|
|
from config import DATA_DIR, MODELS_FILE, GPUS_FILE, CPUS_FILE
|
|
from utils import load_data, save_data
|
|
|
|
reviews_bp = Blueprint('api_reviews', __name__)
|
|
|
|
PENDING_FILE = DATA_DIR / 'pending_reviews.json'
|
|
|
|
|
|
@reviews_bp.route('/api/reviews')
|
|
def api_reviews():
|
|
"""获取待审核列表"""
|
|
reviews = load_data(PENDING_FILE)
|
|
|
|
# 筛选参数
|
|
status = request.args.get('status', 'pending') # pending, approved, rejected, all
|
|
limit = int(request.args.get('limit', 100))
|
|
|
|
if status != 'all':
|
|
reviews = [r for r in reviews if r.get('status', 'pending') == status]
|
|
|
|
# 按时间倒序
|
|
reviews = sorted(reviews, key=lambda x: x.get('created_at', ''), reverse=True)
|
|
|
|
return jsonify(reviews[:limit])
|
|
|
|
|
|
@reviews_bp.route('/api/reviews/count')
|
|
def api_reviews_count():
|
|
"""获取待审核数量"""
|
|
reviews = load_data(PENDING_FILE)
|
|
pending_count = len([r for r in reviews if r.get('status', 'pending') == 'pending'])
|
|
return jsonify({'count': pending_count})
|
|
|
|
|
|
@reviews_bp.route('/api/reviews/<review_id>')
|
|
def api_review_detail(review_id):
|
|
"""获取审核详情"""
|
|
reviews = load_data(PENDING_FILE)
|
|
review = next((r for r in reviews if r['id'] == review_id), None)
|
|
if not review:
|
|
return jsonify({'error': 'Review not found'}), 404
|
|
return jsonify(review)
|
|
|
|
|
|
@reviews_bp.route('/api/reviews/<review_id>/approve', methods=['POST'])
|
|
def api_approve_review(review_id):
|
|
"""通过审核"""
|
|
reviews = load_data(PENDING_FILE)
|
|
review = next((r for r in reviews if r['id'] == review_id), None)
|
|
if not review:
|
|
return jsonify({'error': 'Review not found'}), 404
|
|
|
|
if review.get('status') != 'pending':
|
|
return jsonify({'error': '该申请已处理'}), 400
|
|
|
|
# 获取数据文件
|
|
category_id = review.get('category_id')
|
|
data_file = get_data_file(category_id)
|
|
if not data_file:
|
|
return jsonify({'error': 'Unknown category'}), 400
|
|
|
|
# 添加数据到正式文件
|
|
items = load_data(data_file)
|
|
item_data = review.get('data', {})
|
|
|
|
# 确保有ID和时间戳
|
|
if 'id' not in item_data or not item_data['id']:
|
|
item_data['id'] = uuid.uuid4().hex[:12]
|
|
item_data['created_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
item_data['visible'] = True
|
|
item_data['approved'] = True
|
|
item_data['approved_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
|
|
items.append(item_data)
|
|
save_data(data_file, items)
|
|
|
|
# 更新审核状态
|
|
review['status'] = 'approved'
|
|
review['approved_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
review['approved_by'] = session.get('username', 'admin')
|
|
save_data(PENDING_FILE, reviews)
|
|
|
|
return jsonify({'success': True, 'item': item_data})
|
|
|
|
|
|
@reviews_bp.route('/api/reviews/<review_id>/reject', methods=['POST'])
|
|
def api_reject_review(review_id):
|
|
"""拒绝审核"""
|
|
reviews = load_data(PENDING_FILE)
|
|
review = next((r for r in reviews if r['id'] == review_id), None)
|
|
if not review:
|
|
return jsonify({'error': 'Review not found'}), 404
|
|
|
|
if review.get('status') != 'pending':
|
|
return jsonify({'error': '该申请已处理'}), 400
|
|
|
|
data = request.get_json() or {}
|
|
reason = data.get('reason', '')
|
|
|
|
# 更新审核状态
|
|
review['status'] = 'rejected'
|
|
review['rejected_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
review['rejected_by'] = session.get('username', 'admin')
|
|
review['reject_reason'] = reason
|
|
save_data(PENDING_FILE, reviews)
|
|
|
|
return jsonify({'success': True})
|
|
|
|
|
|
def get_data_file(category_id):
|
|
"""获取类别对应的数据文件"""
|
|
builtin_map = {
|
|
'ai-models': MODELS_FILE,
|
|
'gpus': GPUS_FILE,
|
|
'cpus': CPUS_FILE
|
|
}
|
|
|
|
if category_id in builtin_map:
|
|
return builtin_map[category_id]
|
|
|
|
# 动态分类
|
|
return DATA_DIR / f'items_{category_id}.json'
|
|
|
|
|
|
# ─── 内部函数：提交审核 ───────────────────────────────────────────────────────
|
|
|
|
def submit_for_review(category_id, data, source='web', submitter=None):
|
|
"""
|
|
提交产品到审核队列
|
|
|
|
参数:
|
|
category_id: 分类ID
|
|
data: 产品数据(dict)
|
|
source: 来源 web/api
|
|
submitter: 提交者
|
|
"""
|
|
reviews = load_data(PENDING_FILE)
|
|
|
|
review = {
|
|
'id': uuid.uuid4().hex[:12],
|
|
'category_id': category_id,
|
|
'data': data,
|
|
'source': source,
|
|
'submitter': submitter or 'anonymous',
|
|
'status': 'pending', # pending, approved, rejected
|
|
'created_at': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
}
|
|
|
|
reviews.append(review)
|
|
save_data(PENDING_FILE, reviews)
|
|
|
|
return review |