修改数据采集

lib/api/sum_data_api.py

@@ -149,26 +149,48 @@ def batch_delete_sum_data_api():
 
 @sum_data_bp.route('/sum-data/batch-query', methods=['POST'])
 def batch_query_sum_data():
-    """根据ID列表批量查询求和数据"""
+    """根据ID列表或条件批量查询求和数据"""
     try:
         data = request.json
         ids = data.get('ids', [])
-        log_info(f'[求和数据API] 批量查询求和数据，ID数量: {len(ids)}', 'api')
-        
-        if not ids:
-            return jsonify({'success': False, 'message': '请提供ID列表'}), 400
+        conditions = data.get('conditions', {})
         
         from lib.db import execute_query
         
-        placeholders = ','.join(['%s'] * len(ids))
-        sql = f'SELECT * FROM t_daily_report_sum_data WHERE id IN ({placeholders})'
+        # 支持按ID列表查询
+        if ids:
+            log_info(f'[求和数据API] 按ID批量查询求和数据，ID数量: {len(ids)}', 'api')
+            placeholders = ','.join(['%s'] * len(ids))
+            sql = f'SELECT * FROM t_daily_report_sum_data WHERE id IN ({placeholders})'
+            result = execute_query(sql, tuple(ids))
+            return jsonify({
+                'success': True,
+                'data': result
+            })
         
-        result = execute_query(sql, tuple(ids))
+        # 支持按条件查询
+        if conditions:
+            log_info(f'[求和数据API] 按条件查询求和数据，条件: {conditions}', 'api')
+            where_clauses = []
+            params = []
+            
+            for key, value in conditions.items():
+                if value is not None and value != '':
+                    where_clauses.append(f'{key} = %s')
+                    params.append(value)
+            
+            if not where_clauses:
+                return jsonify({'success': False, 'message': '请提供有效的查询条件'}), 400
+            
+            sql = f'SELECT * FROM t_daily_report_sum_data WHERE {" AND ".join(where_clauses)}'
+            result = execute_query(sql, tuple(params))
+            
+            return jsonify({
+                'success': True,
+                'data': result
+            })
         
-        return jsonify({
-            'success': True,
-            'data': result
-        })
+        return jsonify({'success': False, 'message': '请提供ID列表或查询条件'}), 400
     except Exception as e:
         log_error(f'[求和数据API] 批量查询求和数据失败: {e}', 'api')
         return jsonify({'success': False, 'message': str(e)}), 500