s7data.py 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. import snap7, csv, threading, warnings, time
  2. from utils.statepoint import *
  3. S7Client = snap7.client.Client
  4. class S7data:
  5. def __init__(self, csvfile):
  6. self.logger = None
  7. self.S7Client = None
  8. self.lock = threading.Lock()
  9. self.thread_run = False
  10. self.threads = []
  11. self.nodes = {}
  12. self.node_data = {}
  13. self.target_from_name = {}
  14. with open(csvfile) as f:
  15. for i in csv.DictReader(f):
  16. if i['name'] in self.nodes:
  17. raise Exception(f"S7配置文件节点名称重复:{i['name']}")
  18. else:
  19. self.nodes[i['name']] = i
  20. self.node_data[i['name']] = bytearray(int(i['size']))
  21. def set_logger(self, logger):
  22. self.logger = logger
  23. def set_S7Client(self, s7c):
  24. self.S7Client = s7c
  25. def get_S7Client(self):
  26. return self.S7Client
  27. def send(self, name):
  28. if self.nodes[name]['type'] == 'int':
  29. data = snap7.util.get_int(self.node_data[name], 0)
  30. elif self.nodes[name]['type'] == 'bool':
  31. data = snap7.util.get_bool(self.node_data[name], 0, int(self.nodes[name]['offset']))
  32. elif self.nodes[name]['type'] == 'boollist':
  33. data = [(self.node_data[name][0] >> i) & 1 for i in range(8)]
  34. elif self.nodes[name]['type'] == 'real':
  35. data = snap7.util.get_real(self.node_data[name], 0)
  36. elif self.nodes[name]['type'] == 'string':
  37. data = name, snap7.util.get_string(self.node_data[name], 0)
  38. elif self.nodes[name]['type'] == 'wstring':
  39. data = self.node_data[name][4:].decode(encoding='utf-16be')
  40. else:
  41. warnings.warn('暂不支持的类型:' + self.nodes[name]['type'])
  42. if self.logger:
  43. self.logger.error('暂不支持的类型:' + self.nodes[name]['type'])
  44. return None
  45. if name in self.target_from_name:
  46. for i in self.target_from_name[name]:
  47. i.inject(data)
  48. if self.nodes[name]['type'] == 'boollist' and name + '*' in self.target_from_name:
  49. for i in range(8):
  50. for j in self.target_from_name[name+'*'][i]:
  51. j.inject(data[i])
  52. def update(self, name):
  53. nodeinfo = self.nodes[name]
  54. try:
  55. while True:
  56. if not self.thread_run:
  57. return None
  58. self.lock.acquire()
  59. if not self.S7Client.get_connected():
  60. warnings.warn('S7Client连接中断')
  61. if self.logger:
  62. self.logger.error('S7Client连接中断')
  63. self.thread_run = False
  64. self.lock.release()
  65. return None
  66. tmp = self.S7Client.db_read(int(nodeinfo['db']), int(nodeinfo['start']), int(nodeinfo['size']))
  67. self.lock.release()
  68. if self.node_data[name] != tmp:
  69. self.node_data[name] = tmp
  70. self.send(name)
  71. time.sleep(float(nodeinfo['frequency']) / 1000)
  72. except RuntimeError as reason:
  73. warnings.warn(reason)
  74. if self.logger:
  75. self.logger.error(reason)
  76. self.thread_run = False
  77. self.lock.release()
  78. def start_auto_update(self):
  79. if self.thread_run:
  80. return None
  81. self.threads = []
  82. if self.S7Client == None:
  83. warnings.warn('未初始化S7Client')
  84. if self.logger:
  85. self.logger.error('未初始化S7Client')
  86. return None
  87. if not self.S7Client.get_connected():
  88. warnings.warn('S7Client未连接')
  89. if self.logger:
  90. self.logger.error('S7Client未连接')
  91. return None
  92. for key, value in self.nodes.items():
  93. if value['read_allow'].upper() != 'FALSE':
  94. self.threads.append(threading.Thread(target=self.update, args=(value['name'],)))
  95. self.thread_run = True
  96. for i in self.threads:
  97. i.start()
  98. def end_auto_update(self):
  99. self.thread_run = False
  100. for i in self.threads:
  101. i.join()
  102. def make_point(self, name):
  103. index = -1
  104. solvedname = name
  105. if len(name) > 3 and name[-3] == '[' and name[-1] == ']' and name[-2].isdigit() and 0 <= int(name[-2]) < 8:
  106. index = int(name[-2])
  107. name = name[:-3]
  108. solvedname = name + '*'
  109. if name not in self.nodes:
  110. raise ValueError("创建了未配置的点")
  111. if solvedname not in self.target_from_name:
  112. if index == -1:
  113. self.target_from_name[solvedname] = []
  114. else:
  115. self.target_from_name[solvedname] = [[],[],[],[],[],[],[],[]]
  116. res = Statepoint()
  117. if index == -1:
  118. self.target_from_name[solvedname].append(res)
  119. else:
  120. self.target_from_name[solvedname][index].append(res)
  121. return res