# -*- coding: utf-8 -*-
import pandas as pd
import datetime
from config import Config
from api import API
import pymysql  # 使用 pymysql 替代 mysql.connector
from back_ground_module import CommonModule

start_time = datetime.datetime.now()
api_instance = API()
common_module = CommonModule()


class importDifferentIndustriesData:
    """异业合作数据支撑"""
    def __init__(self):
        self.field_mapping = {}
        self.fields()

    def main(self):
        task_start_time =datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        payload = {"api_key": Config.SaaS_Tasks_APP_ID,
                   "entry_id":  Config.DifferentIndustries_form_ID,
                   }
        abnormal_service = api_instance.entry_data_list(payload)
        abnormal_list = abnormal_service.get("data")  # api请求格式，将数据封装在data字典里
        for i in range(0,len(abnormal_list)):  # 删除历史数据
            try:
                delete_data = {"api_key": Config.SaaS_Tasks_APP_ID,
                                "entry_id": Config.DifferentIndustries_form_ID,
                                "data_id": abnormal_list[i]['_id']}
                api_instance.entry_data_delete(delete_data)
            except:
                pass


        data_commission = common_module.get_differentindustries_details()

        # 生成包含所有行转换后的字典列表
        data_commission = [self.row_to_dict(row, self.field_mapping) for index, row in data_commission.iterrows()] 
        data_commission = {'api_key': Config.SaaS_Tasks_APP_ID, 'entry_id': Config.DifferentIndustries_form_ID, "data_list": data_commission}
        
        result = api_instance.entry_data_batch_create(data_commission)
        result_str = str(result)
        # print(result_str[:500])

        # 保存到Excel文件
        # output_path = r'D:\Idea Project\F6+宜搭+其它(1)\new\文件输出\ngv明细1.xlsx'
        # data_NGV_j.to_excel(output_path, index=False)

        end_time = datetime.datetime.now()

        time_diff = end_time - start_time

        # 打印天数、秒数和微秒数
        print(f"执行时间: {time_diff.days} 天, {time_diff.seconds} 秒, {time_diff.microseconds} 微秒")
        common_module.send_task_status(task_start_time, "异业合作数据支撑")

    @staticmethod
    def row_to_dict(row, field_mapping):
        """将一行数据转换为指定格式的字典"""
        result = {}
        for col_name, widget_id in field_mapping.items():
            if col_name in row:
                value = row[col_name]
                clean_value = None if pd.isna(value) else value
                result[widget_id] = {"value": clean_value}
        return result

    def fields(self):
        self.field_mapping = {
                                "门店id": "_widget_1742884829007",
                                "商品名称": "_widget_1742884829008",
                                "服务期结束时间": "_widget_1742884829009",
                                "门店名称": "_widget_1748931208851",
                                "公司id": "_widget_1748930826642",
                                "门店编码": "_widget_1748931208852"
                            }
if __name__ == '__main__':
    start = importDifferentIndustriesData()
    start.main()