Browse Source

python用例:tcp

scbc.sat2 5 years ago
parent
commit
4f1ffb5fa5
1 changed files with 113 additions and 0 deletions
  1. 113 0
      SATService/tcp_client.py

+ 113 - 0
SATService/tcp_client.py

@@ -0,0 +1,113 @@
+# -*- coding:utf-8 -*-
+import os, sys, time
+import abc
+import socket
+import json
+import numpy as np
+import struct
+import binascii
+
+#创建ProHeader数据类型
+# 定义:
+DataHeader = np.dtype({'names': ['protocol', 'len', 'cmd'], 'formats': ['u1', 'u4', 'u1']}, align=False)
+UserInfo = np.dtype({'names': ['userName', 'password'], 'formats': ['|S260', '|S260']}, align=False)
+
+class BaseClient(object):
+    __metaclass__ = abc.ABCMeta
+
+    def __init__(self):
+        """设备id"""
+        self.device_id = 1
+        '''设备名称'''
+        self.device_name = "name"
+        '''设备命令'''
+        self.device_cmd = "rtn 11;"
+        '''设备通讯超时值:毫秒'''
+        self.device_timeout = 300
+
+        '''通信sock'''
+        self.sock = None
+        '''通信端口号'''
+        self.port = 5588
+        '''通信状态'''
+        self.constate = False
+
+    def __def__(self):
+        self.disconnect()
+
+    '''连接服务器'''
+    def connect(self):
+        try:
+            self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            self.sock.connect(('127.0.0.1', self.port))  # 没有返回值;
+            self.sock.sendall("2222")
+            self.constate = True
+        except Exception, e:
+            print "BaseClient=> socket connect error:", e, self.port
+            self.constate = False
+
+        return self.constate
+
+    # 发送消息;
+    def sendmsg(self):
+        '''是否连接'''
+        if self.constate is False:
+            if self.connect() is False:
+                return None
+
+        '''拼装要发送的数据包'''
+        userinfo = np.array([("superAdmin", "123456")], dtype=UserInfo)
+
+        '''发送请求数据'''
+        try:
+            self.sock.settimeout(5)
+            phead = np.array([(0xAA, userinfo.itemsize + np.array([(0xAA, 0, 0x01)], dtype=DataHeader).itemsize, 0x00)], dtype=DataHeader)
+            '''一次性发送完整包'''
+            # self.sock.sendall(bytearray(phead.tobytes() + bytes(msg_json)))
+            '''分包发送:头、体'''
+            self.sock.sendall(phead.tobytes())
+            self.sock.sendall(userinfo.tobytes())
+            self.sock.settimeout(None)
+        except Exception, e:
+            print "BaseClient=> send Error:", e
+            self.disconnect()
+            self.connect()
+            return None
+        
+        '''接收返回数据'''
+        try:
+            # 等待数据返回;(串口若请求失败,超时值应该设置在3秒以上)
+            self.sock.settimeout(10)
+            data = bytes(self.sock.recv(1024 * 8))
+            print "recv:" + binascii.hexlify(data)
+            self.sock.settimeout(None)
+
+            phead = data[0: np.array([(0xAA, 0, 0x01)], dtype=DataHeader).itemsize]
+            respone = data[np.array([(0xAA, 0, 0x01)], dtype=DataHeader).itemsize:]
+            print "respone:" + binascii.hexlify(respone)
+            '''小端<I, 大端>I, 网络端 !'''
+            protocol, len, cmd = struct.unpack('<BIB', phead)
+            loginresult = int(struct.unpack('<c', respone))
+            '''返回头是否有效'''
+            if protocol != 0xAA and len != data.__len__():
+                return None
+            
+            return loginresult
+        except Exception, e:
+            print "BaseClient=> recv Error:", e
+            return None
+
+    '''断开连接'''
+    def disconnect(self):
+        try:
+            self.sock.shutdown(2)
+            self.sock.close()
+        except Exception, e:
+            print "BaseClient=> socket disconnect error:", e
+
+
+if __name__ == "__main__":
+    bc = BaseClient()
+    bc.sendmsg()
+
+    print "ok!"