Преглед изворни кода

全部提交,跟小龙哥共享

wangxq пре 1 месец
родитељ
комит
45238b1b41
2 измењених фајлова са 23 додато и 9 уклоњено
  1. 22 8
      app/api/data_resource/routes.py
  2. 1 1
      app/config/config.py

+ 22 - 8
app/api/data_resource/routes.py

@@ -157,15 +157,29 @@ def data_resource_save():
     try:
         # 获取表单数据
         receiver = request.json.get('receiver', {})
-        head_data = request.json.get('head_data', [])
-        data_resource = request.json.get('data_resource', {})
-        
-        if not receiver or not data_resource:
-            return jsonify(failed("参数不完整"))
-        
-        # 调用业务逻辑处理数据资源创建
-        resource_id = handle_node(receiver, head_data, data_resource)
+        additional_info = receiver['additional_info']
+
+        # 检查receiver是否存在
+        if not receiver:
+            return jsonify(failed("参数不完整:缺少receiver"))
+            
+        # 检查url是否存在
+        if 'url' not in receiver:
+            return jsonify(failed("参数不完整:缺少url"))
+
+        file_extension = receiver['url'].split('.')[-1]
         
+        if file_extension == 'xlsx' or file_extension == 'xls' or file_extension == 'csv':
+            head_data = additional_info['head_data']
+            data_resource = additional_info['data_resource']
+
+            if not receiver or not data_resource:
+                return jsonify(failed("参数不完整"))
+            # 调用业务逻辑处理数据资源创建
+            resource_id = handle_node(receiver, head_data, data_resource)
+        else:
+            return jsonify(failed("文件格式错误"))
+    
         return jsonify(success({"id": resource_id}))
     except Exception as e:
         logger.error(f"保存数据资源失败: {str(e)}")

+ 1 - 1
app/config/config.py

@@ -45,7 +45,7 @@ class Config:
     
     # Bucket configurations - 新配置
     # BUCKET_NAME = 'dataops-test'
-    # PREFIX = ''  # 由于 bucket_name 已经包含了所需信息,PREFIX 可以置空
+    PREFIX = ''  # 由于 bucket_name 已经包含了所需信息,PREFIX 可以置空
     
     # 新增端口配置基类设置
     PORT = 5500  # 默认端口