#!C:\Python3.6.5\python.exe
# -*- coding: gbk -*-

import socket
import threading

class WSGIServer(object):
 def __init__(self, port):
 """初始化對象"""
 # 創建套接字
 self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 # 解決程序端口占用問題
 self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 # 綁定本地ip地址
 self.tcp_server_socket.bind(("", port))
 # 將套接字變為監聽套接字，最大連接數量為100
 self.tcp_server_socket.listen(100)

 def run_forever(self):
 """設備連接"""
 while True:
 # 1.等待設備連接(通過ip地址和端口建立tcp連接)
 # 如果有設備連接，則會生成用于設備和服務器通訊的套接字：new_socket
 # 會獲取到設備的ip地址和端口
 new_socket, client_addr = self.tcp_server_socket.accept()
 print("設備{0}已連接".format(client_addr))

 # 2.創建線程處理設備的需求
 t1 = threading.Thread(target=self.service_machine, args=(new_socket, client_addr))
 t1.start()

 def service_machine(self, new_socket, client_addr):
 """業務處理"""
 while True:
 # 3.接收設備發送的數據，單次最大1024字節，按‘gbk'格式解碼
 receive_data = new_socket.recv(1024).decode("gbk")
 # 4.如果設備發送的數據不為空
 if receive_data:
 # 4.1 打印接收的數據，這里可以將設備發送的數據寫入到文件中
 # 獲取設備的ID信息
 print(receive_data)
 if receive_data[0:6] == "report":
  response = "SET OK:" + receive_data
 else:
  receive_data = receive_data[6:].split(",")[0]
  # 拼接響應數據
  response = "alarm=" + receive_data + ",Switch:clear"
 print(response)
 # 4.2 返回原數據作為應答，按‘utf-8'格式編碼
 new_socket.send(response.encode("utf-8"))
 # 5.當設備斷開連接時，會收到空的字節數據，判斷設備已斷開連接
 else:
 print('設備{0}斷開連接...'.format(client_addr))
 break

 # 關閉套接字
 new_socket.close()


def main(port):
 """創建一個WEB服務器"""
 wsgi_server = WSGIServer(port)
 print("服務器已開啟")
 wsgi_server.run_forever()



if __name__ == '__main__':
 port = 8125 # 指定端口
 main(8125)