50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
|
# A socket class for communicating in Go Text Protocol
|
||
|
|
#
|
||
|
|
# A simple connect method is defined, and everything else is
|
||
|
|
# suffixed with _gtp, so that this can be subclassed easily,
|
||
|
|
# e.g. for communicating with a server
|
||
|
|
|
||
|
|
import socket
|
||
|
|
|
||
|
|
class GTPSocket:
|
||
|
|
def __init__(self, info):
|
||
|
|
self.info = info
|
||
|
|
self.socket = None
|
||
|
|
|
||
|
|
|
||
|
|
def connect(self):
|
||
|
|
try:
|
||
|
|
self.socket = socket.create_connection(info)
|
||
|
|
return True
|
||
|
|
except:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def get_gtp(self):
|
||
|
|
msg = None
|
||
|
|
|
||
|
|
while msg is None:
|
||
|
|
try:
|
||
|
|
msg = self.socket.recv(1024)
|
||
|
|
if not self._validate_gtp(msg):
|
||
|
|
print 'Error: Incoming data was not a valid GTP message'
|
||
|
|
msg = None
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def send_gtp(self, msg):
|
||
|
|
try:
|
||
|
|
if _validate_gtp(msg):
|
||
|
|
self.socket.send(msg)
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
print 'Error: Outgoing data was not a valid GTP message'
|
||
|
|
return False
|
||
|
|
except:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def _validate_gtp(self, gtp):
|
||
|
|
return True
|