Gmail/export_to_excel.py

42 lines
1.3 KiB
Python
Raw Normal View History

2024-11-23 14:27:50 +00:00
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)