tcp_strategy_client_py2.py

import io
import json
from socket import socket, AF_INET, SOCK_STREAM

class SocketIO(io.RawIOBase):
    def __init__(self, sock):
        self.sock = sock
    def read(self, sz=-1):
        if (sz == -1): sz=0x7FFFFFFF
        return self.sock.recv(sz)
    def seekable(self):
        return False

class TcpClient:
	def __init__(self, addr=('127.0.0.1', 8000)):
		self.host, self.port = addr
		self.sock = None
		self.buf = None

	def new_game(self, solution_id):
	    self.sock = socket(AF_INET, SOCK_STREAM)
	    self.sock.connect((self.host, self.port))
	    self.sock.settimeout(5)
	    self.buf = SocketIO(self.sock)
	    return self.sock.sendall(json.dumps({"solution_id": solution_id}) + '\n')

	def tick(self):
		if self.buf is None: return None
		return json.loads(self.buf.readline().decode('utf-8'))

	def command(self, command, debug=None):
		if self.sock is None: return None
		return self.sock.sendall(json.dumps({ "command": command,
			"debug": debug }) + '\n')

	def end_game(self):
		self.sock.close()
		self.buf = None
		self.sock = None

if __name__ == '__main__':
    c = TcpClient()
    c.new_game(12345)
    while True:
        try:
            data = c.tick()
            c.command("right")
        except ValueError:
            c.end_game()
            break