库存资产excel文件导出

inventory-backend/app/api/v1/inbound/base.py

@@ -1,7 +1,9 @@
 # 文件路径: app/api/v1/inbound/base.py
-from flask import Blueprint, request, jsonify
+
+from flask import Blueprint, request, jsonify, send_file
 from app.services.inbound.base_service import MaterialBaseService
 import traceback
+import datetime
 
 inbound_base_bp = Blueprint('stock_base', __name__)
 
@@ -26,12 +28,13 @@ def search_base():
 @inbound_base_bp.route('/list', methods=['GET'])
 def get_list():
     try:
-        page = request.args.get('pageNum', 1, type=int)  # 前端传的是 pageNum
+        page = request.args.get('pageNum', 1, type=int)
         limit = request.args.get('pageSize', 10, type=int)
 
         # 构造筛选条件
         filters = {
             'keyword': request.args.get('keyword', ''),
+            'company': request.args.get('company', ''),
             'category': request.args.get('category', ''),
             'type': request.args.get('type', ''),
             'isEnabled': request.args.get('isEnabled', None)
@@ -45,7 +48,7 @@ def get_list():
 
 
 # ==============================================================================
-# 2.1 选项接口 (GET /api/v1/inbound/base/options) [新增]
+# 2.1 选项接口 (GET /api/v1/inbound/base/options)
 # ==============================================================================
 @inbound_base_bp.route('/options', methods=['GET'])
 def get_options():
@@ -57,9 +60,45 @@ def get_options():
         return jsonify({"code": 500, "msg": str(e)}), 500
 
 
+# ==============================================================================
+# 2.2 导出接口 (GET /api/v1/inbound/base/export)
+# ==============================================================================
+@inbound_base_bp.route('/export', methods=['GET'])
+def export_data():
+    try:
+        # 获取筛选条件
+        filters = {
+            'keyword': request.args.get('keyword', ''),
+            'company': request.args.get('company', ''),
+            'category': request.args.get('category', ''),
+            'type': request.args.get('type', ''),
+            'isEnabled': request.args.get('isEnabled', None)
+        }
+
+        # 生成 Excel 文件流
+        file_stream = MaterialBaseService.export_excel(filters)
+
+        # 生成文件名：库存统计+年月日+时分秒 (北京时间 UTC+8)
+        # 简单处理：UTC时间 + 8小时
+        beijing_time = datetime.datetime.utcnow() + datetime.timedelta(hours=8)
+        filename = f"库存统计_{beijing_time.strftime('%Y%m%d_%H%M%S')}.xlsx"
+
+        # 发送文件
+        # 注意：download_name 仅在较新 Flask 版本有效，旧版本可能需要手动 header，
+        # 但通常浏览器下载名由前端 Blob 处理或 Content-Disposition 决定。
+        return send_file(
+            file_stream,
+            mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+            as_attachment=True,
+            download_name=filename
+        )
+    except Exception as e:
+        traceback.print_exc()
+        return jsonify({"code": 500, "msg": f"导出失败: {str(e)}"}), 500
+
+
 # ==============================================================================
 # 3. 新增接口 (POST /api/v1/inbound/base/)
-# 注意：前端 material_base.ts 可能会请求 / 或 /add，这里统一匹配
 # ==============================================================================
 @inbound_base_bp.route('/', methods=['POST'])
 def create():