Initial commit

stock-html/routes/watchlist.py

@@ -0,0 +1,47 @@
+"""
+关注列表 API 路由（纯数据库版）
+"""
+from flask import Blueprint, request, jsonify
+from db import (
+    login_required, get_current_user_id,
+    db_get_watchlist, db_add_to_watchlist, db_remove_from_watchlist
+)
+
+bp = Blueprint('watchlist', __name__, url_prefix='/api')
+
+
+@bp.route('/watchlist', methods=['GET'])
+@login_required
+def get_watchlist():
+    """获取关注列表"""
+    user_id = get_current_user_id()
+    watchlist = db_get_watchlist(user_id)
+    return jsonify({'success': True, 'watchlist': watchlist})
+
+
+@bp.route('/watchlist', methods=['POST'])
+@login_required
+def add_to_watchlist():
+    """添加到关注列表"""
+    try:
+        user_id = get_current_user_id()
+        data = request.get_json()
+        code = data.get('code')
+        name = data.get('name', f'股票{code}')
+        
+        watchlist, error = db_add_to_watchlist(user_id, code, name)
+        if error:
+            return jsonify({'success': False, 'error': error}), 400
+        
+        return jsonify({'success': True, 'watchlist': watchlist})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 400
+
+
+@bp.route('/watchlist/<stock_code>', methods=['DELETE'])
+@login_required
+def remove_from_watchlist(stock_code):
+    """从关注列表移除"""
+    user_id = get_current_user_id()
+    watchlist = db_remove_from_watchlist(user_id, stock_code)
+    return jsonify({'success': True, 'watchlist': watchlist or []})