Added example tcp client python script
This commit is contained in:
parent
1d6670f2f9
commit
75cda323a1
89
tcp.py
Normal file
89
tcp.py
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
from __future__ import print_function
|
||||||
|
|
||||||
|
from socket import socket, AF_INET, SOCK_STREAM
|
||||||
|
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
|
||||||
|
server = ('127.0.0.1', 2442)
|
||||||
|
|
||||||
|
|
||||||
|
def from_message(content):
|
||||||
|
try:
|
||||||
|
return json.loads(content)
|
||||||
|
except ValueError:
|
||||||
|
return {}
|
||||||
|
|
||||||
|
|
||||||
|
def to_message(typ, value='', params=None):
|
||||||
|
if params is None:
|
||||||
|
params = {}
|
||||||
|
return json.dumps({'type': typ, 'value': value, 'params': params})
|
||||||
|
|
||||||
|
|
||||||
|
class Client(object):
|
||||||
|
first = True
|
||||||
|
def process(self, message):
|
||||||
|
typ = message.get('type', '')
|
||||||
|
value = message.get('value', '')
|
||||||
|
params = message.get('params', {})
|
||||||
|
if not typ:
|
||||||
|
return
|
||||||
|
|
||||||
|
print('->', typ)
|
||||||
|
|
||||||
|
if value:
|
||||||
|
print('-> value', value)
|
||||||
|
|
||||||
|
if params:
|
||||||
|
print('-> params: ', params)
|
||||||
|
|
||||||
|
|
||||||
|
def send(self, *args, **kwargs):
|
||||||
|
params = kwargs.get('params', {})
|
||||||
|
if '_ID' not in params:
|
||||||
|
params['_ID'] = '{}'.format(int(time.time()*1000))
|
||||||
|
kwargs['params'] = params
|
||||||
|
message = to_message(*args, **kwargs)
|
||||||
|
print('outgoing message:', message)
|
||||||
|
self.sock.send(message + '\n') # remember to send the newline at the end :)
|
||||||
|
|
||||||
|
def connect(self):
|
||||||
|
print('connecting to', ':'.join(map(str, server)))
|
||||||
|
self.sock = socket(AF_INET, SOCK_STREAM)
|
||||||
|
self.sock.connect(server)
|
||||||
|
self.connected = True
|
||||||
|
try:
|
||||||
|
# send a simple example query after connected
|
||||||
|
self.send("STATION.GET_STATUS")
|
||||||
|
|
||||||
|
while self.connected:
|
||||||
|
content = self.sock.recv(65500)
|
||||||
|
if not content:
|
||||||
|
break
|
||||||
|
print('incoming message')
|
||||||
|
|
||||||
|
try:
|
||||||
|
message = json.loads(content)
|
||||||
|
except ValueError:
|
||||||
|
message = {}
|
||||||
|
|
||||||
|
if not message:
|
||||||
|
continue
|
||||||
|
|
||||||
|
self.process(message)
|
||||||
|
|
||||||
|
finally:
|
||||||
|
self.sock.close()
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
self.connected = False
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
s = Client()
|
||||||
|
s.connect()
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
Loading…
Reference in New Issue
Block a user