#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2025/3/15 15:22
# @Author  : AngesZhu
# @File    : data_file_utils.py
# @Desc    : excel表格相关操作
import pandas as pd
from utils.logger_utils import logger


class ExcelHandler:
    def __init__(self, file_path: str):
        """
        初始化 ExcelHandler 类。
        :param file_path: Excel 文件路径
        """
        self.file_path = file_path

    def read_excel(self, sheet_name=0, header=0, usecols=None):
        """
        读取 Excel 文件。
        :param sheet_name: 需要读取的 sheet 名称或索引（默认第一个 sheet）
        :param header: 作为列名的行号（默认第一行）
        :param usecols: 指定需要读取的列（默认读取所有列）
        :return: pandas DataFrame
        """
        try:
            df = pd.read_excel(self.file_path, sheet_name=sheet_name, header=header, usecols=usecols)
            # print(df)
            return df.fillna("")
        except Exception as e:
            logger.error(f"读取 Excel 文件失败: {e}")
            return None

    def write_excel(self, data: pd.DataFrame, sheet_name='Sheet1', mode='w', index=False):
        """
        写入 Excel 文件。
        :param data: 需要写入的 DataFrame
        :param sheet_name: 目标 sheet 名称（默认 'Sheet1'）
        :param mode: 写入模式（'w' 覆盖写入，'a' 追加写入）
        :param index: 是否写入索引（默认 False）
        """
        try:
            with pd.ExcelWriter(self.file_path, engine='openpyxl', mode=mode) as writer:
                data.to_excel(writer, sheet_name=sheet_name, index=index)
            logger.info("Excel 文件写入成功。")
        except Exception as e:
            logger.error(f"写入 Excel 文件失败: {e}")

    def append_to_excel(self, data: pd.DataFrame, sheet_name='Sheet1', index=False):
        """
        追加数据到 Excel 文件的指定 sheet。
        :param data: 需要追加的 DataFrame
        :param sheet_name: 目标 sheet 名称（默认 'Sheet1'）
        :param index: 是否写入索引（默认 False）
        """
        try:
            with pd.ExcelWriter(self.file_path, engine='openpyxl', mode='a', if_sheet_exists='overlay') as writer:
                startrow = writer.sheets[sheet_name].max_row if sheet_name in writer.sheets else 0
                data.to_excel(writer, sheet_name=sheet_name, startrow=startrow, index=index, header=(startrow == 0))
            logger.info("数据成功追加到 Excel 文件。")
        except Exception as e:
            logger.error(f"追加 Excel 文件失败: {e}")