read.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. import os
  2. import yaml
  3. topoCheckData = None
  4. # 读取yaml
  5. def read_yaml_file(filepath):
  6. with open(filepath, 'r') as file:
  7. data = yaml.load(file, Loader=yaml.FullLoader)
  8. return data
  9. # 获取当前脚本所在的目录
  10. current_dir = os.path.dirname(os.path.abspath(__file__))
  11. topoCheckData = read_yaml_file(f"{current_dir}\\topoCheck.yaml")
  12. # print(topoCheckData)
  13. # print(topoCheckData["topoCheck"]["A03"])
  14. # tempdata = topoCheckData["topoCheck"]
  15. # print(f"标识码 名称 说明")
  16. # for attr in tempdata:
  17. # print(
  18. # f"{attr} {tempdata[attr]['note']} {tempdata[attr]['UI']['description'].replace('说明: ', '').replace('说明:', '').replace('<br/>', '')}")
  19. def getTopoCheckSQL(type, my_dict):
  20. # sqllist = []
  21. data = topoCheckData["topoCheck"][type]
  22. tempSql = data["ALG"]
  23. for key in my_dict:
  24. value = my_dict[key]
  25. k = f"@{key}@"
  26. tempSql = tempSql.replace(k, value)
  27. # print(tempSql)
  28. return tempSql
  29. # 获取description
  30. def getTopoCheckDescription(type):
  31. # sqllist = []
  32. data = topoCheckData["topoCheck"][type]
  33. description = data["UI"]["description"]
  34. return description
  35. # 获取note
  36. def getTopoCheckNote(type):
  37. # sqllist = []
  38. data = topoCheckData["topoCheck"][type]
  39. note = data["note"]
  40. return f"{type}:{note}"
  41. # my_dict = {
  42. # "intable_s": "topology_table",
  43. # "outtable": "topology_table_temp"
  44. # }
  45. # sql = getTopoCheckSQL("A03", my_dict)
  46. # print(sql)
  47. # for s in sql.split(";"):
  48. # print(s)