param-hub-python/modules/routes/api_gpus.py

"""
GPU CRUD API
"""
import uuid
import json
from datetime import datetime
from flask import Blueprint, request, jsonify, Response
from config import GPUS_FILE
from utils import load_data, save_data, parse_date_to_timestamp

gpus_bp = Blueprint('api_gpus', __name__)


def _safe_sort_key(x, key):
    val = x.get(key)
    if val is None:
        return 0 if key in ['memory_gb', 'cuda_cores', 'tensor_cores', 'views'] else ''
    return val


@gpus_bp.route('/api/gpus')
def api_gpus():
    gpus = load_data(GPUS_FILE)
    hide_hidden = request.args.get('all', '0') == '0'
    if hide_hidden:
        gpus = [g for g in gpus if g.get('visible', True)]
    keyword = request.args.get('q', '').strip().lower()
    if keyword:
        gpus = [g for g in gpus if keyword in g.get('name', '').lower() or
                keyword in g.get('manufacturer', '').lower()]
    sort_by = request.args.get('sort', 'default')
    reverse = request.args.get('order', 'desc') == 'desc'
    if sort_by == 'default':
        gpus = sorted(gpus, key=lambda x: (
            not x.get('is_pinned', False),
            -(parse_date_to_timestamp(x.get('publish_date', '')) or
              parse_date_to_timestamp(x.get('created_at', '')) or 0)
        ))
    elif sort_by in ['name', 'memory_gb', 'price_usd', 'created_at', 'publish_date',
                      'views', 'updated_at', 'release_year']:
        gpus = sorted(gpus, key=lambda x: _safe_sort_key(x, sort_by), reverse=reverse)
    return jsonify(gpus)


@gpus_bp.route('/api/gpus/<gpu_id>')
def api_gpu_detail(gpu_id):
    gpus = load_data(GPUS_FILE)
    gpu = next((g for g in gpus if g['id'] == gpu_id), None)
    if not gpu:
        return jsonify({'error': 'GPU not found'}), 404
    return jsonify(gpu)


@gpus_bp.route('/api/gpus', methods=['POST'])
def api_create_gpu():
    data = request.get_json()
    gpus = load_data(GPUS_FILE)
    data['id'] = uuid.uuid4().hex[:12]
    data['created_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    data['visible'] = data.get('visible', True)
    data['publish_date'] = data.get('publish_date', '')
    data['views'] = data.get('views', 0)
    data['is_pinned'] = data.get('is_pinned', False)
    gpus.append(data)
    save_data(GPUS_FILE, gpus)
    return jsonify(data)


@gpus_bp.route('/api/gpus/<gpu_id>', methods=['PUT'])
def api_update_gpu(gpu_id):
    data = request.get_json()
    gpus = load_data(GPUS_FILE)
    gpu = next((g for g in gpus if g['id'] == gpu_id), None)
    if not gpu:
        return jsonify({'error': 'GPU not found'}), 404
    gpu.update(data)
    gpu['updated_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    save_data(GPUS_FILE, gpus)
    return jsonify(gpu)


@gpus_bp.route('/api/gpus/<gpu_id>', methods=['DELETE'])
def api_delete_gpu(gpu_id):
    gpus = load_data(GPUS_FILE)
    gpus = [g for g in gpus if g['id'] != gpu_id]
    save_data(GPUS_FILE, gpus)
    return jsonify({'success': True})


@gpus_bp.route('/api/gpus/<gpu_id>/visible', methods=['POST'])
def api_toggle_gpu_visible(gpu_id):
    gpus = load_data(GPUS_FILE)
    gpu = next((g for g in gpus if g['id'] == gpu_id), None)
    if not gpu:
        return jsonify({'error': 'GPU not found'}), 404
    gpu['visible'] = not gpu.get('visible', True)
    save_data(GPUS_FILE, gpus)
    return jsonify({'success': True, 'visible': gpu['visible']})


# ─── 导出导入 ───────────────────────────────────────────────────────

@gpus_bp.route('/api/gpus/export', methods=['GET'])
def api_export_gpus():
    """导出所有GPU数据"""
    try:
        gpus = load_data(GPUS_FILE)
        export_data = {
            'type': 'gpus',
            'items': gpus,
            'count': len(gpus),
            'export_time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
            'version': '1.0'
        }
        json_str = json.dumps(export_data, ensure_ascii=False, indent=2)
        response = Response(
            json_str,
            mimetype='application/json',
            headers={'Content-Disposition': f'attachment; filename=gpus-export-{datetime.now().strftime("%Y%m%d%H%M%S")}.json'}
        )
        return response
    except Exception as e:
        return jsonify({'error': str(e)}), 500


@gpus_bp.route('/api/gpus/import', methods=['GET', 'POST'])
def api_import_gpus():
    """导入GPU数据"""
    try:
        if request.method == 'GET':
            return jsonify({'endpoint': '/api/gpus/import', 'method': 'POST', 'params': {'mode': 'merge 或 replace'}})
        
        import_data = request.get_json()
        if not import_data or 'items' not in import_data:
            return jsonify({'error': '缺少 items 字段'}), 400
        
        mode = request.args.get('mode', 'merge')
        gpus = load_data(GPUS_FILE)
        imported_items = import_data['items']
        
        result = {'success': True, 'imported': 0, 'updated': 0, 'skipped': []}
        
        for item in imported_items:
            existing = next((g for g in gpus if g['id'] == item['id']), None)
            if existing:
                if mode == 'replace':
                    existing.update(item)
                    result['updated'] += 1
                else:
                    result['skipped'].append(item['id'])
            else:
                gpus.append(item)
                result['imported'] += 1
        
        save_data(GPUS_FILE, gpus)
        return jsonify(result)
    except Exception as e:
        return jsonify({'error': str(e)}), 500