import os
import datetime

from account import AccountManagerSQLite


def export_database_to_excel(output_directory=".", file_name_prefix="exported_accounts"):
    """
    从数据库导出数据到 Excel 文件。

    参数:
        output_directory (str): 输出文件夹路径，默认为当前目录。
        file_name_prefix (str): 文件名前缀，默认是 'exported_accounts'。
    """
    try:
        # 确保输出目录存在
        if not os.path.exists(output_directory):
            os.makedirs(output_directory)

        # 生成 Excel 文件名，包含时间戳
        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
        file_name = f"{file_name_prefix}_{timestamp}.xlsx"
        excel_path = os.path.join(output_directory, file_name)

        # 创建数据库管理实例并导出数据
        db_manager = AccountManagerSQLite()
        db_manager.export_to_excel(excel_path)

        print(f"数据成功导出到: {excel_path}")
        return excel_path
    except Exception as e:
        print(f"导出失败: {e}")
        raise

# 如果此文件作为主程序运行
if __name__ == "__main__":
    # 设置导出文件夹（可以修改为你想要的路径）
    output_directory = "./exports"

    # 调用导出功能
    export_database_to_excel(output_directory)