test_run_local_piliang.py 3.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. # -*- coding: utf-8 -*-
  2. # @Time : 2024/6/11
  3. # @Author : 魏志亮
  4. import datetime
  5. import sys
  6. import traceback
  7. from os import *
  8. from utils.conf.read_conf import yaml_conf, read_conf
  9. def get_exec_data(batch_no=None, read_type=None, run_count=1):
  10. if batch_no and read_type:
  11. data = get_data_by_batch_no_and_type(batch_no, read_type)
  12. if data is None:
  13. raise ValueError(f"未找到批次号:{batch_no},类型:{read_type}")
  14. else:
  15. data = get_batch_exec_data(run_count)
  16. if data is None:
  17. trans_print("当前有任务在执行")
  18. sys.exit(0)
  19. elif len(data.keys()) == 0:
  20. trans_print("当前无任务")
  21. sys.exit(0)
  22. return data
  23. def run(data: dict = dict(), save_db=False):
  24. exec_process = None
  25. if data['transfer_type'] in ['second', 'minute']:
  26. exec_process = MinSecTrans(data=data, save_db=save_db)
  27. if data['transfer_type'] in ['fault', 'warn']:
  28. exec_process = FaultWarnTrans(data=data, save_db=save_db)
  29. if exec_process is None:
  30. raise Exception("No exec process")
  31. exec_process.run()
  32. if __name__ == '__main__':
  33. env = 'dev'
  34. if len(sys.argv) >= 2:
  35. env = sys.argv[1]
  36. conf_path = path.abspath(f"./conf/etl_config_{env}.yaml")
  37. environ['ETL_CONF'] = conf_path
  38. yaml_config = yaml_conf(conf_path)
  39. environ['env'] = env
  40. run_count = int(read_conf(yaml_config, "run_batch_count", 1))
  41. from utils.log.trans_log import trans_print
  42. from service.plt_service import get_batch_exec_data, get_data_by_batch_no_and_type
  43. from etl.wind_power.fault_warn.FaultWarnTrans import FaultWarnTrans
  44. from etl.wind_power.min_sec.MinSecTrans import MinSecTrans
  45. from utils.file.trans_methods import read_file_to_df
  46. begin = datetime.datetime.now()
  47. df = read_file_to_df("tmp_file/rebuild_data.csv")
  48. results = list()
  49. data = dict()
  50. for batch_code, batch_name, transfer_type, transfer_addr, field_code, field_name \
  51. in zip(df['batch_code'], df['batch_name'], df['transfer_type'], df['transfer_addr'], df['field_code'],
  52. df['field_name']):
  53. batch_begin = datetime.datetime.now()
  54. transfer_addr = transfer_addr.replace(r"/data/download/collection_data",
  55. r"/data/download/datang_shangxian")
  56. trans_print("开始执行批次:", batch_code, batch_name, transfer_type, field_code, field_name)
  57. trans_print("批次路径:", transfer_addr)
  58. data['batch_code'] = batch_code
  59. data['batch_name'] = batch_name
  60. data['transfer_type'] = transfer_type
  61. data['transfer_addr'] = transfer_addr
  62. data['field_code'] = field_code
  63. data['field_name'] = field_name
  64. try:
  65. run(data=data, save_db=True)
  66. results.append((batch_code, batch_name, transfer_type, field_code, field_name, 'success'))
  67. except Exception as e:
  68. results.append((batch_code, batch_name, transfer_type, field_code, field_name, 'error'))
  69. trans_print(traceback.format_exc())
  70. finally:
  71. trans_print("执行结束,耗时:", datetime.datetime.now() - batch_begin, "总耗时:", datetime.datetime.now() - begin)
  72. for data in results:
  73. trans_print(data)
  74. trans_print("执行结束,总耗时:", datetime.datetime.now() - begin)