取消levels接口，限制classifications接口仅为申万行业接口

app.py

@@ -6324,196 +6324,100 @@ def parse_json_field(field_value):
 # ==================== 行业API ====================
 @app.route('/api/classifications', methods=['GET'])
 def get_classifications():
-    """获取所有行业分类系统"""
+    """获取申银万国行业分类树形结构"""
     try:
+        # 查询申银万国行业分类的所有数据
         sql = """
-              SELECT DISTINCT f002v as classification_name
+              SELECT f003v as code, f004v as level1, f005v as level2, f006v as level3,f007v as level4
               FROM ea_sector
-              WHERE f002v NOT IN ('指数成份股', '市场分类', '概念板块', '地区省市分类', '中上协行业分类')
+              WHERE f002v = '申银万国行业分类'
-              ORDER BY classification_name \
+                AND f003v IS NOT NULL
+                AND f004v IS NOT NULL
+              ORDER BY f003v
               """
 
         result = db.session.execute(text(sql)).all()
 
-        classifications = [{'name': row.classification_name} for row in result]
+        # 构建树形结构
+        tree_dict = {}
 
-        return jsonify({
+        for row in result:
-            'success': True,
+            code = row.code
-            'data': classifications
+            level1 = row.level1
-        })
+            level2 = row.level2
+            level3 = row.level3
 
-    except Exception as e:
+            # 跳过空数据
-        return jsonify({
+            if not level1:
-            'success': False,
+                continue
-            'error': str(e)
-        }), 500
 
-
+            # 第一层
-@app.route('/api/levels', methods=['GET'])
+            if level1 not in tree_dict:
-def get_industry_levels():
+                # 获取第一层的code（取前3位或前缀）
-    """获取行业层级数据"""
+                level1_code = code[:3] if len(code) >= 3 else code
-    try:
+                tree_dict[level1] = {
-        classification = request.args.get('classification')
+                    'value': level1_code,
-        level = request.args.get('level', type=int)
+                    'label': level1,
-        level1_name = request.args.get('level1_name', '')
+                    'children_dict': {}
-        level2_name = request.args.get('level2_name', '')
-        level3_name = request.args.get('level3_name', '')
-
-        if not classification or not level or level < 1 or level > 4:
-            return jsonify({
-                'success': False,
-                'error': 'Invalid parameters'
-            }), 400
-
-        # 层级到字段的映射
-        level_fields = {
-            1: "f004v",
-            2: "f005v",
-            3: "f006v",
-            4: "f007v"
-        }
-
-        field_name = level_fields[level]
-
-        # 构建查询
-        if level == 1:
-            sql = f"""
-            SELECT DISTINCT {field_name} as name, 
-                            MIN(f003v) as code
-            FROM ea_sector
-            WHERE f002v = :classification 
-              AND {field_name} IS NOT NULL
-            GROUP BY name
-            ORDER BY name
-            """
-            params = {"classification": classification}
-
-        elif level == 2:
-            sql = f"""
-            SELECT DISTINCT {field_name} as name, 
-                            MIN(f003v) as code
-            FROM ea_sector
-            WHERE f002v = :classification 
-              AND f004v = :level1_name
-              AND {field_name} IS NOT NULL
-            GROUP BY name
-            ORDER BY name
-            """
-            params = {"classification": classification, "level1_name": level1_name}
-
-        elif level == 3:
-            sql = f"""
-            SELECT DISTINCT {field_name} as name, 
-                            MIN(f003v) as code
-            FROM ea_sector
-            WHERE f002v = :classification 
-              AND f004v = :level1_name
-              AND f005v = :level2_name
-              AND {field_name} IS NOT NULL
-            GROUP BY name
-            ORDER BY name
-            """
-            params = {
-                "classification": classification,
-                "level1_name": level1_name,
-                "level2_name": level2_name
-            }
-
-        elif level == 4:
-            sql = f"""
-            SELECT DISTINCT f003v as code, 
-                            {field_name} as name
-            FROM ea_sector
-            WHERE f002v = :classification 
-              AND f004v = :level1_name
-              AND f005v = :level2_name
-              AND f006v = :level3_name
-              AND {field_name} IS NOT NULL
-            ORDER BY name
-            """
-            params = {
-                "classification": classification,
-                "level1_name": level1_name,
-                "level2_name": level2_name,
-                "level3_name": level3_name
-            }
-
-        results = db.session.execute(text(sql), params).all()
-
-        industries = [{"code": row.code, "name": row.name} for row in results if row.name]
-
-        return jsonify({
-            'success': True,
-            'data': industries
-        })
-
-    except Exception as e:
-        return jsonify({
-            'success': False,
-            'error': str(e)
-        }), 500
-
-
-@app.route('/api/info', methods=['GET'])
-def get_industry_info():
-    """获取行业详细信息"""
-    try:
-        classification = request.args.get('classification')
-        code = request.args.get('code')
-
-        if not classification or not code:
-            return jsonify({
-                'success': False,
-                'error': 'Missing parameters'
-            }), 400
-
-        # 根据代码长度确定字段
-        if len(code) >= 8:
-            field_name = "f007v"
-        elif len(code) >= 6:
-            field_name = "f006v"
-        elif len(code) >= 4:
-            field_name = "f005v"
-        else:
-            field_name = "f004v"
-
-        sql = f"""
-        SELECT {field_name} as name,
-               f004v as level1_name,
-               f005v as level2_name,
-               f006v as level3_name,
-               f007v as level4_name
-        FROM ea_sector
-        WHERE f002v = :classification 
-          AND f003v = :code
-          AND {field_name} IS NOT NULL
-        LIMIT 1
-        """
-
-        result = db.session.execute(text(sql), {
-            "classification": classification,
-            "code": code
-        }).first()
-
-        if not result:
-            return jsonify({
-                'success': False,
-                'error': 'Industry not found'
-            }), 404
-
-        return jsonify({
-            'success': True,
-            'data': {
-                'name': result.name,
-                'code': code,
-                'classification': classification,
-                'hierarchy': {
-                    'level1': result.level1_name,
-                    'level2': result.level2_name,
-                    'level3': result.level3_name,
-                    'level4': result.level4_name
                 }
+
+            # 第二层
+            if level2:
+                if level2 not in tree_dict[level1]['children_dict']:
+                    # 获取第二层的code（取前6位）
+                    level2_code = code[:6] if len(code) >= 6 else code
+                    tree_dict[level1]['children_dict'][level2] = {
+                        'value': level2_code,
+                        'label': level2,
+                        'children_dict': {}
+                    }
+
+                # 第三层
+                if level3:
+                    if level3 not in tree_dict[level1]['children_dict'][level2]['children_dict']:
+                        tree_dict[level1]['children_dict'][level2]['children_dict'][level3] = {
+                            'value': code,
+                            'label': level3
+                        }
+
+        # 转换为最终格式
+        result_list = []
+        for level1_name, level1_data in tree_dict.items():
+            level1_node = {
+                'value': level1_data['value'],
+                'label': level1_data['label']
             }
+
+            # 处理第二层
+            if level1_data['children_dict']:
+                level1_children = []
+                for level2_name, level2_data in level1_data['children_dict'].items():
+                    level2_node = {
+                        'value': level2_data['value'],
+                        'label': level2_data['label']
+                    }
+
+                    # 处理第三层
+                    if level2_data['children_dict']:
+                        level2_children = []
+                        for level3_name, level3_data in level2_data['children_dict'].items():
+                            level2_children.append({
+                                'value': level3_data['value'],
+                                'label': level3_data['label']
+                            })
+                        if level2_children:
+                            level2_node['children'] = level2_children
+
+                    level1_children.append(level2_node)
+
+                if level1_children:
+                    level1_node['children'] = level1_children
+
+            result_list.append(level1_node)
+
+        return jsonify({
+            'success': True,
+            'data': result_list
         })
 
     except Exception as e:
@@ -6523,6 +6427,29 @@ def get_industry_info():
         }), 500
 
 
+@app.route('/api/stocklist', methods=['GET'])
+def get_stock_list():
+    """获取股票列表"""
+    try:
+        sql = """
+              SELECT DISTINCT SECCODE as code, SECNAME as name
+              FROM ea_stocklist
+              ORDER BY SECCODE
+              """
+
+        result = db.session.execute(text(sql)).all()
+
+        stocks = [{'code': row.code, 'name': row.name} for row in result]
+
+        return jsonify(stocks)
+
+    except Exception as e:
+        return jsonify({
+            'success': False,
+            'error': str(e)
+        }), 500
+
+
 @app.route('/api/events', methods=['GET'], strict_slashes=False)
 def api_get_events():
     """
@@ -6544,10 +6471,8 @@ def api_get_events():
         date_range = request.args.get('date_range')
         recent_days = request.args.get('recent_days', type=int)
 
-        # 行业筛选参数
+        # 行业筛选参数（只支持申银万国行业分类）
-        industry_classification = request.args.get('industry_classification')
+        industry_code = request.args.get('industry_code')  # 申万行业代码，如 "S370502"
-        industry_code = request.args.get('industry_code')
-        industry_level = request.args.get('industry_level', type=int)
 
         # 概念/标签筛选参数
         tag = request.args.get('tag')
@@ -6593,9 +6518,10 @@ def api_get_events():
             query = query.filter_by(importance=importance)
         if creator_id:
             query = query.filter_by(creator_id=creator_id)
-        # 新增：行业代码过滤（MySQL JSON，对象数组模式）
+        # 新增：行业代码过滤（申银万国行业分类）
-        if industry_classification and industry_code:
+        if industry_code:
-            json_path = f'$[*]."{industry_classification}"'
+            # related_industries 格式: [{"申银万国行业分类": "S370502"}, ...]
+            json_path = '$[*]."申银万国行业分类"'
             query = query.filter(
                 text("JSON_CONTAINS(JSON_EXTRACT(related_industries, :json_path), :industry_code)")
             ).params(json_path=json_path, industry_code=json.dumps(industry_code))