ClearData.py 940 B

12345678910111213141516171819202122232425
  1. import datetime
  2. from etl.common.PathsAndTable import PathsAndTable
  3. from service.plt_service import update_trans_transfer_progress
  4. from utils.log.trans_log import trans_print
  5. class ClearData(object):
  6. def __init__(self, pathsAndTable: PathsAndTable):
  7. self.pathsAndTable = pathsAndTable
  8. def clean_data(self):
  9. self.pathsAndTable.delete_tmp_files()
  10. if self.pathsAndTable.save_db:
  11. self.pathsAndTable.delete_batch_db()
  12. self.pathsAndTable.delete_batch_files()
  13. def run(self):
  14. trans_print("开始清理数据,临时文件夹:", self.pathsAndTable.get_tmp_path())
  15. begin = datetime.datetime.now()
  16. self.clean_data()
  17. update_trans_transfer_progress(self.pathsAndTable.batch_no, self.pathsAndTable.read_type, 5,
  18. self.pathsAndTable.save_db)
  19. trans_print("清理数据结束,耗时:", datetime.datetime.now() - begin)