main copy.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503
  1. from datetime import datetime
  2. from multiprocessing import Pool
  3. import json
  4. import os
  5. import time
  6. import traceback
  7. import warnings
  8. from li_plted.V1_0_0.corepro_V1 import *
  9. #from keras.models import load_model
  10. import pickle
  11. from sqlalchemy import text, delete, and_, or_, update
  12. import pandas as pd
  13. from ZlwlAlgosCommon.utils.ProUtils import *
  14. from ZlwlAlgosCommon.service.iotp.IotpAlgoService import IotpAlgoService
  15. from ZlwlAlgosCommon.service.iotp.Beans import DataField
  16. from ZlwlAlgosCommon.orm.models import *
  17. from socdiag.V_1_0_0.SOCBatDiag import SocDiag
  18. from LowSocAlarm.V1_0_0.low_soc_alarm import Low_soc_alarm
  19. from SorCal.V_1_0_0.sorcal import sor_est
  20. from DataSplit.V_1_0_0 import data_status as ds ##充电状态标准化程序
  21. from DataSplit.V_1_0_0 import data_split as dt ##分段函数程序
  22. from DataSplit.V_1_0_0 import data_drive_stat as ddt ##行驶数据按行驶段汇总统计
  23. from DataSplit.V_1_0_0 import data_charge_stat as dct ##充电数据按充电段汇总
  24. from DataSplit.V_1_0_0 import data_stand_stat as dst ##静置数据按静置段汇总
  25. from DataSplit.V_1_0_0 import data_drive_stat_period as ddtp ##行驶数据按充电周期汇总统计
  26. from DataSplit.V_1_0_0 import trans_day as trd ##解决跨天的问题
  27. def update_param(db_engine, rc):#
  28. # 从redis中获取参数,如果redis中获取不到,则去数据库中获取
  29. data = rc.get("algo_param_from_mysql:algo_adjustable_param")
  30. #data=pd.DataFrame()
  31. if pd.isnull(data):
  32. df_algo_adjustable_param = pd.read_sql("select id, algo_id, pack_code, param from algo_adjustable_param", db_engine)
  33. else:
  34. df_algo_adjustable_param = pd.DataFrame(json.loads(data))
  35. data = rc.get("algo_param_from_mysql:algo_list")#pd.DataFrame()
  36. if pd.isnull(data):
  37. df_algo_list = pd.read_sql("select id, algo_id, algo_name, is_activate, global_param, fault_code, fault_influence from algo_list", db_engine)
  38. else:
  39. df_algo_list = pd.DataFrame(json.loads(data))
  40. data = rc.get("algo_param_from_mysql:algo_pack_param")
  41. if pd.isnull(data):
  42. df_algo_pack_param = pd.read_sql("select id, pack_code, param from algo_pack_param", db_engine)
  43. else:
  44. df_algo_pack_param = pd.DataFrame(json.loads(data))
  45. data = rc.get("algo_param_from_mysql:app_device")
  46. if pd.isnull(data):
  47. df_snpk_list = pd.read_sql("select sn, imei,pack_model,scrap_status from t_device", db_engine)
  48. df_snpk_list=df_snpk_list[df_snpk_list['scrap_status']<4]
  49. else:
  50. df_snpk_list = pd.DataFrame(json.loads(data))
  51. return df_algo_adjustable_param,df_algo_list,df_algo_pack_param,df_snpk_list
  52. def main(): #process_num
  53. # 程序不能停止
  54. #while(True):
  55. try:
  56. process_num=1
  57. warnings.filterwarnings("ignore")
  58. try:
  59. # 调用算法前的准备工作
  60. kafka_topic_key = 'topic_task_day_1_sxqtest'
  61. kafka_groupid_key = 'group_id_task_day_1_sxqtest'
  62. algo_list = ['socdiag','low_soc_diag','Sor_Diag','Li_Plted','Data_Split'] # 本调度所包含的算法名列表。
  63. loggers = sysUtils.get_loggers(algo_list, log_base_path, process_num) # 为每个算法分配一个logger
  64. logger_main.info(f"process-{process_num}: 配置中间件")
  65. # mysql
  66. mysql_algo_params = sysUtils.get_cf_param('mysql-algo')
  67. mysqlUtils = MysqlUtils()
  68. mysql_algo_engine, mysql_algo_Session= mysqlUtils.get_mysql_engine(mysql_algo_params)
  69. mysql_algo_conn = mysql_algo_engine.connect()
  70. # kafka
  71. kafka_params = sysUtils.get_cf_param('kafka')
  72. kafkaUtils = KafkaUtils()
  73. kafka_consumer = kafkaUtils.get_kafka_consumer(kafka_params, kafka_topic_key, kafka_groupid_key, client_id=kafka_topic_key)
  74. #Hbase
  75. hbase_params = sysUtils.get_cf_param('hbase-datafactory')#hbase
  76. iotp_service = IotpAlgoService(hbase_params=hbase_params)
  77. #redis
  78. redis_params = sysUtils.get_cf_param('redis')
  79. reidsUtils = RedisUtils()
  80. rc = reidsUtils.get_redis_conncect(redis_params)
  81. except Exception as e:
  82. logger_main.error(f'process-{process_num}: {e}')
  83. logger_main.error(f'process-{process_num}: {traceback.format_exc()}')
  84. # 开始准备调度
  85. logger_main.info(f"process-{process_num}: 监听topic {kafka_params[kafka_topic_key]}等待kafka 调度")
  86. # for message in kafka_consumer:
  87. path = 'D:/data/'#'/data/common/benchi/data/'
  88. sn_list = os.listdir(path)
  89. pack_code = 'CL3282A'
  90. df_algo_adjustable_param, df_algo_list, df_algo_pack_param, df_snpk_list= update_param(mysql_algo_conn,rc)#
  91. sql = "select * from algo_pack_param"
  92. df_algo_pack_param_all = pd.read_sql(sql, mysql_algo_conn)
  93. sql = "select * from algo_list"
  94. df_algo_param = pd.read_sql(sql, mysql_algo_conn)
  95. df_algo_pack_param = json.loads(df_algo_pack_param_all[df_algo_pack_param_all['pack_code'] == pack_code]['param'].iloc[0])
  96. sql = f"select sn, imei from t_device where sn in {tuple(sn_list)}"
  97. df_snlist = pd.read_sql(sql, mysql_algo_conn)
  98. start_time_dt = pd.to_datetime('2022-01-01')
  99. end_time_dt = pd.to_datetime('2022-11-30')
  100. path = 'D:/data/'#'/data/common/benchi/data/'
  101. sn_list = os.listdir(path)
  102. sn_list=sn_list[10:20]
  103. for sn in sn_list:
  104. st_time=time.time()
  105. try:
  106. snpath = path + sn + '/'
  107. times = os.listdir(snpath)
  108. times = sorted(times)
  109. for i in range(0, len(times), 2):
  110. # 获取当前读取的4个文件
  111. files_to_read = times[i:i+2]
  112. # 循环读取文件
  113. df_data_all=pd.DataFrame()
  114. rd_data_st_time=time.time()
  115. for sntime in files_to_read:
  116. # 读取Excel文件
  117. start_time = sntime.split('.')[0].split('_')[0]
  118. end_time = sntime.split('.')[0].split('_')[1]
  119. # 取数
  120. time_st = time.time()
  121. logger_main.info(f"process-{process_num}: 开始取数{sn_list}")
  122. columns = [ DataField.time, DataField.sn, DataField.pack_crnt, DataField.pack_volt, DataField.pack_soc,
  123. DataField.cell_voltage_count, DataField.cell_temp_count, DataField.cell_voltage, DataField.cell_temp,
  124. DataField.other_temp_value, DataField.bms_sta]
  125. df_data_t = pd.read_pickle(snpath+sntime, compression='zip')
  126. if len(df_data_all):
  127. df_data_all=df_data_all.append(df_data_t)
  128. else:
  129. df_data_all=df_data_t
  130. df_data_all=df_data_all.reset_index(drop=True)
  131. print(sn+'_第'+str(i)+'次/共'+str(len(times))+'次取数耗时:'+str(time.time()-rd_data_st_time)+',共{}条数据'.format(str(len(df_data_t))))
  132. if mysql_algo_conn.closed:
  133. mysql_algo_conn = mysql_algo_engine.connect() # 从连接池中获取一个myslq连接
  134. # if mysql_algo_conn.closed:
  135. # mysql_algo_conn = mysql_algo_engine.connect() # 从连接池中获取一个myslq连接
  136. # schedule_params = json.loads(message.value)
  137. # if (schedule_params is None) or (schedule_params ==''):
  138. # logger_main.info('{} kafka数据异常,跳过本次运算'.format(str(message.value)))
  139. # continue
  140. # # kafka 调度参数解析
  141. # df_snlist = pd.DataFrame(schedule_params['snlist'])
  142. # df_algo_adjustable_param = pd.DataFrame([(d['algo_id'], d['param'],d['param_ai']) for d in schedule_params['adjustable_param']], columns=['algo_id', 'param','param_ai'])
  143. # df_algo_pack_param = json.loads(schedule_params['pack_param'][0]['param'])
  144. # df_algo_pack_param = {k: eval(v) if isinstance(v, str) else v for k, v in df_algo_pack_param.items()}
  145. # df_algo_param = pd.DataFrame(schedule_params['algo_list'])
  146. # start_time = schedule_params['start_time']
  147. # end_time = schedule_params['end_time']
  148. # pack_code = schedule_params['pack_code']
  149. # cell_type = schedule_params['cell_type']
  150. # sn_list=df_snlist['sn'].tolist()
  151. # # 取数
  152. # time_st = time.time()
  153. # logger_main.info(f"process-{process_num}: 开始取数{sn_list}")
  154. # columns = [ DataField.time, DataField.sn, DataField.pack_crnt, DataField.pack_volt, DataField.pack_soc,
  155. # DataField.cell_voltage_count, DataField.cell_temp_count, DataField.cell_voltage, DataField.cell_temp,
  156. # DataField.other_temp_value, DataField.bms_sta, DataField.charge_sta,DataField.latitude,DataField.longitude]
  157. # df_data = iotp_service.get_data(sn_list=sn_list, columns=columns, start_time=start_time, end_time=end_time)
  158. # logger_main.info(f'process-{process_num},获取到{len(df_data)}条数据,取数耗时:{time.time()-time_st}')
  159. # # 将字符串转换成datetime对象
  160. # str_date = start_time
  161. # date_time =datetime.datetime.strptime(str_date, '%Y-%m-%d %H:%M:%S')
  162. # # 将datetime对象减去6小时
  163. # new_date_time = date_time - datetime.timedelta(hours=8)
  164. # # 将datetime对象转换成字符串
  165. # start_time_8h = new_date_time.strftime('%Y-%m-%d %H:%M:%S')
  166. # df_data_8h = iotp_service.get_data(sn_list=sn_list, columns=columns, start_time=start_time_8h, end_time=start_time)
  167. # logger_main.info(f'process-{process_num},获取到{len(df_data_8h)}条数据,取数耗时:{time.time()-time_st}')
  168. # except Exception as e:
  169. # logger_main.error(f"process-{process_num}:获取原始数据出错")
  170. # logger_main.error(f"process-{process_num}:{e}")
  171. # logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  172. # continue
  173. # 数据清洗
  174. try:
  175. time_st = time.time()
  176. logger_main.info(f'process-{process_num}数据清洗')
  177. #里程填充
  178. df_data_all['mileage'] = df_data_all['mileage'].replace(0, np.nan).ffill()
  179. df_data_all['mileage'] = df_data_all['mileage'].replace(0, np.nan).bfill()
  180. df_data_all['mileage']=df_data_all['mileage']/1000
  181. df_data, df_table, cellvolt_name, celltemp_name = iotp_service.datacleaning(df_algo_pack_param,df_data_all)#进行数据清洗
  182. # df_data_8h, df_table_t, cellvolt_name_t, celltemp_name_t = iotp_service.datacleaning(df_algo_pack_param,df_data_8h)#进行数据清洗
  183. print('洗数耗时:'+str(time.time()-time_st))
  184. if len(df_data) == 0:
  185. logger_main.info(f"process-{process_num}: 数据清洗耗时{time.time()-time_st}, 无有效数据,跳过本次运算")
  186. continue
  187. else:
  188. logger_main.info(f"process-{process_num}: {pack_code}, time_type:{df_data.loc[0, 'time']} ~ {df_data.iloc[-1]['time']}, 数据清洗完成耗时{time.time()-time_st}")
  189. except Exception as e:
  190. logger_main.error(f"process-{process_num}:数据清洗出错")
  191. logger_main.error(f"process-{process_num}:{e}")
  192. logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  193. continue
  194. # mysql数据读取
  195. try:
  196. time_st = time.time()
  197. logger_main.info(f'process-{process_num}开始读取mysql故障数据')
  198. if len(sn_list) == 1:
  199. sn_tuple = f"('{sn_list[0]}')"
  200. else:
  201. sn_tuple = tuple(sn_list)
  202. sql = "select * from algo_all_fault_info_ing where sn in {}".format(sn_tuple) #fault_code='{}' or fault_code='{}') and 'C599','C590',
  203. df_diag_ram = pd.read_sql(sql, mysql_algo_conn)
  204. sql = "select * from algo_ailipltd_result where sn in {}".format(sn_tuple) #fault_code='{}' or fault_code='{}') and 'C599','C590',
  205. Li_pltd_his = pd.read_sql(sql, mysql_algo_conn)
  206. logger_main.info(f'process-{process_num}读取mysql耗时{time.time()-time_st}')
  207. except Exception as e:
  208. logger_main.error(f"process-{process_num}:读取redis出错")
  209. logger_main.error(f"process-{process_num}:{e}")
  210. logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  211. continue
  212. #算法1_SOC诊断调用
  213. # try:
  214. # time_st = time.time()
  215. # loggers['socdiag'].info(f'开始执行算法{pack_code}, time:{start_time}~{end_time},\n sn_list:{sn_list}')
  216. # period = 24*60 #算法周期min
  217. # soc_diag = SocDiag(cell_type, df_algo_pack_param, df_algo_adjustable_param, df_algo_param, end_time, period, pack_code, df_snlist, df_data)
  218. # df_res_new_C109, df_res_end_C109= soc_diag.soc_block(df_diag_ram)
  219. # df_res_end_C107 = soc_diag.soc_jump()
  220. # df_res_new_soc = df_res_new_C109
  221. # df_res_end_soc = pd.concat([df_res_end_C107, df_res_end_C109])
  222. # loggers['socdiag'].info(f'算法运行完成{pack_code},算法耗时{time.time()-time_st}')
  223. # except Exception as e:
  224. # loggers['socdiag'].error('算法运行出错')
  225. # loggers['socdiag'].error(str(e))
  226. # loggers['socdiag'].error(traceback.format_exc())
  227. # df_res_end_soc=pd.DataFrame()
  228. # df_res_new_soc=pd.DataFrame()
  229. # # 算法2_低电量调用
  230. try:
  231. time_st = time.time()
  232. loggers['low_soc_diag'].info(f'开始执行算法{pack_code}, time:{start_time}~{end_time},\n sn_list:{sn_list}')
  233. low_soc_warning = Low_soc_alarm(df_data,cellvolt_name)
  234. df_res_new_lw_soc, df_res_update_lw_soc,df_res_end_lw_soc= low_soc_warning.diag(df_algo_pack_param,df_algo_param,df_algo_adjustable_param,df_data,df_table,df_diag_ram,df_snlist)
  235. start_time
  236. # month = date.month
  237. # day = date.day
  238. df_res_new_lw_soc.to_excel('lowsoc_sn_date_{}_{}.xlsx'.format(sn,start_time[0:10]))
  239. loggers['low_soc_diag'].info(f'算法运行完成{pack_code},算法耗时{time.time()-time_st}')
  240. except Exception as e:
  241. loggers['low_soc_diag'].error('算法运行出错')
  242. loggers['low_soc_diag'].error(str(e))
  243. loggers['low_soc_diag'].error(traceback.format_exc())
  244. df_res_new_lw_soc=pd.DataFrame()
  245. df_res_update_lw_soc=pd.DataFrame()
  246. df_res_end_lw_soc=pd.DataFrame()
  247. # # 算法3_SOR计算调用
  248. # try:
  249. # time_st = time.time()
  250. # loggers['Sor_Diag'].info(f'开始执行算法{pack_code}, time:{start_time}~{end_time},\n sn_list:{sn_list}')
  251. # Diagsor_temp = sor_est(df_data, df_algo_pack_param)#计算内阻
  252. # df_sor_add = Diagsor_temp.sor_cal()
  253. # loggers['Sor_Diag'].info(f'算法运行完成{pack_code},算法耗时{time.time()-time_st}')
  254. # except Exception as e:
  255. # loggers['Sor_Diag'].error('算法运行出错')
  256. # loggers['Sor_Diag'].error(str(e))
  257. # loggers['Sor_Diag'].error(traceback.format_exc())
  258. # df_sor_add=pd.DataFrame()
  259. # #算法4_析锂计算调用
  260. # try:
  261. # time_st = time.time()
  262. # loggers['Li_Plted'].info(f'开始执行算法{pack_code}, time:{start_time}~{end_time},\n sn_list:{sn_list}')
  263. # pkl_path='li_plted/V1_0_0/scaler.pkl'
  264. # md_path='li_plted/V1_0_0/model.h5'
  265. # scaler=pickle.load(open(pkl_path,'rb')) #读取标准化参数
  266. # model=load_model(md_path) #读取模型参数
  267. # data_set=df_data.groupby('sn').apply(prediction,scaler,model,cellvolt_name)
  268. # if not data_set.empty:
  269. # df_result=data_set.groupby('sn').apply(out_final,Li_pltd_his,df_algo_param,df_algo_pack_param)
  270. # # if not df_result.empty:
  271. # # df_res_lipltdchange,df_res_lipltd_new = zip(*df_result.groupby('sn').apply(alarme_final,Li_pltd_his,df_algo_pack_param))
  272. # loggers['Li_Plted'].info(f'算法运行完成{pack_code},算法耗时{time.time()-time_st}')
  273. # except Exception as e:
  274. # loggers['Li_Plted'].error('算法运行出错')
  275. # loggers['Li_Plted'].error(str(e))
  276. # loggers['Li_Plted'].error(traceback.format_exc())
  277. # df_res_lipltdchange=pd.DataFrame()
  278. # df_res_lipltd_new=pd.DataFrame()
  279. #算法5_日数据分段
  280. try:
  281. cal_time=time.time()
  282. loggers['Data_Split'].info(f'开始执行算法{pack_code}, time:{start_time}~{end_time},\n sn_list:{sn_list}')
  283. df_merge=ds.data_status(df_data,c_soc_dif_p=0.05,s_soc_dif_p=0,c_order_delta=1200,s_order_delta=300)
  284. ##基于各个状态码,进行分段,分段函数
  285. df_drive,df_charge,df_stand,df_data_split_rlt=dt.split(df_merge,celltemp_name,cellvolt_name,drive_interval_time_min=1200,charge_interval_time_min=1200,stand_interval_time_min=1200,single_num_min=3,drive_sts=3,charge_sts=[21,22],stand_sts=0)
  286. # date = datetime.datetime.strptime(start_time, '%Y-%m-%d-%H-%M-%S')
  287. # 获取datetime对象的月份
  288. # month = date.month
  289. # day = date.day
  290. print('计算耗时:'+str(time.time()-cal_time))
  291. df_data_split_rlt.to_excel('data_split_sn_date_{}_{}.xlsx'.format(sn,start_time[0:10]))
  292. loggers['Data_Split'].info(f'算法运行完成{pack_code},算法耗时{time.time()-time_st}')
  293. except Exception as e:
  294. loggers['Data_Split'].error('算法运行出错')
  295. loggers['Data_Split'].error(str(e))
  296. loggers['Data_Split'].error(traceback.format_exc())
  297. # #结果写入mysql
  298. # try:
  299. # df_res_new = pd.concat([df_res_new_soc, df_res_new_lw_soc]) #, res1
  300. # df_res_update=df_res_update_lw_soc#pd.concat([df_res_update_lw_soc,df_res_update_crnt, df_res_update_temp]) #, res1
  301. # df_res_end = pd.concat([df_res_end_soc,df_res_end_lw_soc]) #, res2
  302. # df_res_new.reset_index(drop=True, inplace=True)
  303. # df_res_update.reset_index(drop=True, inplace=True)
  304. # df_res_end.reset_index(drop=True, inplace=True)
  305. # time_st = time.time()
  306. # session = mysql_algo_Session()
  307. # if not df_res_new.empty:
  308. # df_res_new['date_info'] = df_res_new['start_time']
  309. # df_res_new['create_time'] = datetime.now()
  310. # df_res_new['create_by'] = 'algo'
  311. # df_res_new['is_delete'] = 0
  312. # df_res_new.to_sql("algo_all_fault_info_ing", con=mysql_algo_conn, if_exists="append", index=False)
  313. # logger_main.info(f'process-{process_num}新增未结束故障入库{pack_code}完成')
  314. # if not df_res_end.empty:
  315. # df_res_end=df_res_end.where(pd.notnull(df_res_end),None)
  316. # df_res_end=df_res_end.fillna(0)
  317. # for index in df_res_end.index:
  318. # df_t = df_res_end.loc[index:index]
  319. # sql = 'delete from algo_all_fault_info_ing where start_time=:start_time and fault_code=:fault_code and sn=:sn'
  320. # params = {'start_time': df_t['start_time'].values[0],
  321. # 'fault_code': df_t['fault_code'].values[0], 'sn': df_t['sn'].values[0]}
  322. # session.execute(sql, params=params)
  323. # sql = 'insert into algo_all_fault_info_done (date_info, start_time, end_time, sn, imei, model, fault_level, fault_code, fault_info,\
  324. # fault_reason, fault_advice, fault_location, device_status,odo, create_time, create_by,update_time, update_by, is_delete,comment) values \
  325. # (:date_info, :start_time, :end_time, :sn, :imei, :model,:fault_level, :fault_code, :fault_info,\
  326. # :fault_reason, :fault_advice, :fault_location, :device_status, :odo, :create_time, :create_by, :update_time,:update_by, :is_delete , :comment)'
  327. # params = {'date_info': datetime.now(),
  328. # 'start_time': df_t['start_time'].values[0],
  329. # 'end_time': df_t['end_time'].values[0],
  330. # 'sn': df_t['sn'].values[0],
  331. # 'imei': df_t['imei'].values[0],
  332. # 'model' :pack_code,
  333. # 'fault_level': df_t['fault_level'].values[0],
  334. # 'fault_code': df_t['fault_code'].values[0],
  335. # 'fault_info': df_t['fault_info'].values[0],
  336. # 'fault_reason': df_t['fault_reason'].values[0],
  337. # 'fault_advice': df_t['fault_advice'].values[0],
  338. # 'fault_location': df_t['fault_location'].values[0],
  339. # 'device_status': df_t['device_status'].values[0],
  340. # 'odo': df_t['odo'].values[0],
  341. # 'create_time': datetime.now(),
  342. # 'create_by': 'algo',
  343. # 'update_time': datetime.now(),
  344. # 'update_by': None,
  345. # 'is_delete': 0,
  346. # 'comment': None}
  347. # session.execute(sql, params=params)
  348. # session.commit()
  349. # logger_main.info(f'process-{process_num}结束故障入库{pack_code}完成')
  350. # if not df_res_update.empty:
  351. # df_res_update=df_res_update.where(pd.notnull(df_res_update),None)
  352. # df_res_update=df_res_update.fillna(0)
  353. # for index in df_res_update.index:
  354. # df_t = df_res_update.loc[index:index]
  355. # try:
  356. # # 更新数据
  357. # with mysql_algo_Session() as session:
  358. # session.execute(update(AlgoAllFaultInfoIng).where(
  359. # and_((AlgoAllFaultInfoIng.start_time == df_t['start_time'].values[0]),
  360. # (AlgoAllFaultInfoIng.fault_code == df_t['fault_code'].values[0]),
  361. # (AlgoAllFaultInfoIng.sn == df_t['sn'].values[0]))).
  362. # values(fault_level=df_t['fault_level'].values[0],
  363. # comment=df_t['comment'].values[0]))
  364. # session.commit()
  365. # except Exception as e:
  366. # logger_main.error(f"process-{process_num}:结果入库出错")
  367. # logger_main.error(f"process-{process_num}:{e}")
  368. # logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  369. # finally:
  370. # session.close()
  371. # logger_main.info(f"process-{process_num}: 更新入库完成")
  372. # else:
  373. # logger_main.info(f"process-{process_num}: 无更新故障")
  374. # if not df_sor_add.empty:
  375. # time_record = time.time()
  376. # df_sor_rlt = df_sor_add#df_sor_rlt.append()
  377. # df_sor_rlt.reset_index(drop = True, inplace = True)
  378. # df_sor_rlt.to_sql("algo_mid_sorout",con=mysql_algo_conn, if_exists="append",index=False)
  379. # write_mysql_time = write_mysql_time + time.time()-time_record
  380. # logger_main.info(f'process-{process_num}新增未结束故障入库{pack_code}完成')
  381. # logger_main.info(f"process-{process_num}: 结果入库耗时:{time.time()-time_st}")
  382. # except Exception as e:
  383. # logger_main.error(f"process-{process_num}:结果入库出错")
  384. # logger_main.error(f"process-{process_num}:{e}")
  385. # logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  386. try:
  387. if not df_data_split_rlt.empty:
  388. time_record = time.time()
  389. df_data_split_rlt.reset_index(drop = True, inplace = True)
  390. #df_data_split_rlt.to_sql("algo_charge_info",con=mysql_algo_conn, if_exists="append",index=False)
  391. # write_mysql_time = write_mysql_time + time.time()-time_record
  392. logger_main.info(f'process-{process_num}新增未结束故障入库{pack_code}完成')
  393. logger_main.info(f"process-{process_num}: 结果入库耗时:{time.time()-time_st}")
  394. except Exception as e:
  395. logger_main.error(f"process-{process_num}:数据分段结果入库出错")
  396. logger_main.error(f"process-{process_num}:{e}")
  397. logger_main.error(f"process-{process_num}:{traceback.format_exc()}")
  398. except Exception as e:
  399. logger_main.error(f'process-{process_num}: {e}')
  400. logger_main.error(f'process-{process_num}: {traceback.format_exc()}')
  401. print('本次计算耗时:'+str(time.time()-st_time))
  402. except Exception as e:
  403. logger_main.error(f'process-{process_num}: {e}')
  404. logger_main.error(f'process-{process_num}: {traceback.format_exc()}')
  405. if __name__ == '__main__':
  406. # while(True):
  407. try:
  408. # 配置量
  409. cur_env = 'dev' # 设置运行环境
  410. app_path = "D:/ZLWORK/code/zlwl-algos/" # 设置app绝对路径
  411. log_base_path = f"{os.path.dirname(os.path.abspath(__file__))}/log" # 设置日志路径
  412. app_name = "task_day_1_sxqtest" # 应用名
  413. sysUtils = SysUtils(cur_env, app_path)
  414. logger_main = sysUtils.get_logger(app_name, log_base_path)
  415. logger_main.info(f"本次主进程号: {os.getpid()}")
  416. main()
  417. # 读取配置文件 (该部分请不要修改)
  418. # processes = int(sysUtils.env_params.get("PROCESS_NUM_PER_NODE", '1')) # 默认为1个进程
  419. # pool = Pool(processes = int(processes))
  420. # logger_main.info("开始分配子进程")
  421. # for i in range(int(processes)):
  422. # pool.apply_async(main, (i, ))
  423. # pool.close()
  424. # logger_main.info("进程分配结束,堵塞主进程")
  425. # pool.join()
  426. except Exception as e:
  427. print(str(e))
  428. print(traceback.format_exc())
  429. logger_main.error(str(e))
  430. logger_main.error(traceback.format_exc())
  431. finally:
  432. handlers = logger_main.handlers.copy()
  433. for h in handlers:
  434. logger_main.removeHandler(h)
  435. # pool.terminate()