main.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536
  1. import pandas as pd
  2. import time
  3. from LIB.BACKEND import DBManager,Log # 以相对路径的方式引入!!!!!
  4. from V_1_0_0 import core # 以相对路径的方式引入!!!!!
  5. import os
  6. if __name__ == '__main__':
  7. # 获取环境变量
  8. env_dist = os.environ
  9. host = env_dist.get("HOST", '120.25.223.1')
  10. port = env_dist.get("PORT", '4901')
  11. db = env_dist.get("DB", 'ali')
  12. user = env_dist.get("ROOT", 'root')
  13. password = env_dist.get("PASSWORD", '123456')
  14. # 日志配置
  15. now_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()).replace(":","_")
  16. log_path = 'log/' + now_str
  17. if not os.path.exists(log_path):
  18. os.makedirs(log_path)
  19. log = Log.Mylog(log_name='cal_sor', log_level = 'info')
  20. log.set_file_hl(file_name='{}/info.log'.format(log_path), log_level='info', size=1024* 1024 * 100)
  21. log.set_file_hl(file_name='{}/error.log'.format(log_path), log_level='error', size=1024* 1024 * 100)
  22. logger = log.get_logger()
  23. # 准备算法数据
  24. sn = "PK10001A326000123"
  25. st = '2021-07-06 00:00:00'
  26. et = '2021-07-07 20:00:00'
  27. dbManager = DBManager.DBManager()
  28. df_data = dbManager.get_data(sn=sn, start_time=st, end_time=et, data_groups=['bms', 'gps', 'accum', 'system'])
  29. df_bms = df_data['bms']
  30. # 调用核心算法
  31. calSor = core.CalSor();
  32. # 接收算法结果
  33. print(calSor.calSor(df_bms,df_bms))