p2pool/tests/src/stratum_dummy.py

import socket
import time
import sys

f = open('stratum_dummy' + sys.argv[1] + '.log', 'wb');

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(1)
sock.setblocking(True)

time.sleep(5)

while sock.connect_ex(('127.0.0.1', 3333)) != 0:
	f.write(b'.\n')
	print('.')
	time.sleep(1)

diff = '';

if (sys.argv[1] == '2'):
	diff = '+1000';
if (sys.argv[1] == '3'):
	diff = '+10000000';

request = '{"id":1,"method":"login","params":{"login":"x' + diff + '"}}\n'
sock.sendall(request.encode('utf-8'))

while True:
	data = sock.recv(1024)
	if len(data) == 0:
		break;
	f.write(data)
	f.flush()
	print(data.decode('utf-8'))

sock.close()
f.close()