data_processor.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. from data_processor.importer import SalesDataImporter
  2. from data_processor.combined_sales_exporter import CombinedSalesExporter
  3. from data_processor.daily_sales_exporter import DailySalesExporter
  4. from data_processor.daily_zones_exporter import DailyZonesSalesExporter
  5. # 数据库配置建议从环境变量或配置文件中读取
  6. db_config = {
  7. 'host': '10.41.1.220',
  8. 'port': 3306,
  9. 'user': 'it_user',
  10. 'password': 'Goelia*199@5',
  11. 'database': 'sales_data'
  12. }
  13. def data_processor(excel_file, export_file):
  14. """
  15. 处理销售数据:导入 -> 导出每日销售 -> 导出区域销售 -> 合并导出
  16. """
  17. try:
  18. print("开始导入销售数据")
  19. importer = SalesDataImporter(db_config, excel_file)
  20. importer.import_data()
  21. print("销售数据导入完成")
  22. print("初始化每日销售导出器")
  23. daily_sales_exporter = DailySalesExporter(db_config)
  24. print("初始化区域销售导出器")
  25. daily_zones_sales_exporter = DailyZonesSalesExporter(db_config)
  26. print("开始合并导出")
  27. combined_exporter = CombinedSalesExporter(daily_sales_exporter, daily_zones_sales_exporter)
  28. combined_exporter.export(export_file)
  29. print("合并导出完成")
  30. except Exception as e:
  31. print(f"数据处理过程中发生错误: {e}")
  32. raise