海外客户档案同步简bi修复

back_ground_module/update_BI_CRM_info.py

@@ -9,6 +9,8 @@ from back_ground_module import CommonModule
 import numpy as np
 from config import Config
 from log_config import configure_task_logger, configure_error_task_logger
+import os
+import json
 
 common_module = CommonModule()
 
@@ -18,6 +20,9 @@ logger = configure_task_logger()
 # 获取已经配置好的错误任务日志记录器
 error_task_logger = configure_error_task_logger()
 
+# 设置输出目录
+output_dir = "output"
+os.makedirs(output_dir, exist_ok=True)
 
 class CRMDataProcessor:
     """泰国CRM数据迁移到BI"""
@@ -187,10 +192,11 @@ class CRMDataProcessor:
 
     def process_data(self, df):
         """处理CRM数据"""
-        # 去掉前六列和后两列
-        df.to_csv("CRM.csv", index=False)
-        df = df.iloc[:, 6:-2]
-        print(df)
+        # 保留第一列，去掉2-7列和后两列
+        # df.to_csv(os.path.join(output_dir, "CRM.csv"), index=False)
+        df = df.copy()
+        df = df.iloc[:, [0] + list(range(6, df.shape[1] - 2))] # shape【1】含义，df的列数，第二维度的大小shape（行，列）
+        # df.to_csv(os.path.join(output_dir, "CRM_processed.csv"), index=False)
 
         # 生成URL
         base_url = f"https://www.jiandaoyun.com/dashboard/app/{self.api_key}/form/{self.entry_id}/data/"
@@ -303,6 +309,7 @@ class CRMDataProcessor:
             self.close_db()
 
     def import_data(self, df, table_name):
+        # 不支持json的值
         try:
             self.connect_db()