ngv添加保存数据

back_ground_module/update_NGV_data.py

@@ -13,7 +13,10 @@ error_task_logger = configure_error_task_logger()
 start_time = datetime.datetime.now()
 api_instance = API()
 common_module = CommonModule()
-
+output_dir = "output"  # 设置输出目录
+# 创建输出目录（如果不存在）
+import os
+os.makedirs(output_dir, exist_ok=True)
 
 class UpdateNGVData:
     """NGV数据每日新增"""
@@ -103,6 +106,10 @@ class UpdateNGVData:
             # all_data = [self.row_to_dict(row, self.field_mapping) for index, row in data_NGV_j1.iterrows()] # 前两天的全部数据
             # all_data = [self.row_to_dict(row, self.field_mapping) for index, row in data_NGV_j.iterrows()]  # 前一天的全部数据
             all_data = [self.row_to_dict(row, self.field_mapping) for index, row in filtered_df.iterrows()]  # 增量数据
+            try:
+                filtered_df.to_csv(output_dir + "\\" + f"{task_start_time}NGV.csv")
+            except Exception as e:
+                error_task_logger.error(f"NGV过滤后数据保存异常: {e}")
 
             #
             data = {'api_key': Config.SaaS_Tasks_APP_ID, 'entry_id': Config.NGV_TASKS_ENTRY_ID, "data_list": all_data}