from socket import socket, AF_INET, SOCK_DGRAM
import string, random
from time import time

from private_settings import CLIENT_ADDRESS

ECHO_PORT = 7

LEN_PACKET = 127
N_PACKETS = 5000
TOT_BITS = float(LEN_PACKET * N_PACKETS * 8) * 2
MEGA = float(1024 * 1024)
UPDATE_STEP = (N_PACKETS/10)

class UDP_EchoClient:
    s = socket(AF_INET, SOCK_DGRAM)
    
    def __init__(self, host):
        self.host = host
        self.packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
    
    def __packet(self):
        # Comment out the checks when measuring the throughput
        # packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
        UDP_EchoClient.s.sendto(packet, (self.host, ECHO_PORT))
        data = UDP_EchoClient.s.recv(LEN_PACKET)
        # assert packet == data, "packet error:\n%s\n%s\n" % (packet, data)
    
    def test(self):
        start = time()
        for i in range(N_PACKETS):
            if (i % UPDATE_STEP) == 0: print '%.2f%%' % ((float(i)/float(N_PACKETS)) * 100.)
            self.__packet()
        t = time() - start
        print 'Throughput: (%.2f)Mbits/s' % ((TOT_BITS / t)/MEGA)

while True:
    e = UDP_EchoClient(CLIENT_ADDRESS)
    e.test()
