06101013

post_processing/cdq_coe_gen.py

@@ -170,18 +170,11 @@ def prepare_request_body(df, config, predict):
     """
     data = df.copy()
     # 转换时间格式为字符串
-    if 'dateTime' in data.columns and isinstance(data['dateTime'].iloc[0], datetime):
-        data['dateTime'] = data['dateTime'].dt.strftime('%Y-%m-%d %H:%M:%S')
+    if config['col_time'] in data.columns and isinstance(data[config['col_time']].iloc[0], datetime):
+        data[config['col_time'] ] = data[config['col_time'] ].dt.strftime('%Y-%m-%d %H:%M:%S')
     data['model'] = predict
-    # 排除不需要的字段（如果有）
-    exclude_fields = ['_id']  # 通常排除MongoDB的默认_id字段
-
-    # 获取所有字段名（排除不需要的字段）
-    available_fields = [col for col in data.columns if col not in exclude_fields]
-
-    # 转换为记录列表（保留所有字段）
-    data = data[available_fields].to_dict('records')
-
+    # 保留必要的字段
+    data = data[[config['col_time'], config['col_power'], predict, 'model']].to_dict('records')
     # 构造请求体（固定部分+动态数据部分）
     request_body = {
         "stationCode": config['stationCode'],