bingo.py

import argparse, socket, zmq, time, random, threading

def checkBingo(clientList, currentList):
    check = [False] * 25
    for i in range(25):
        for j in range(25):
            if currentList[i] == clientList[j]:
                check[j] = True
    line = 0

    for i in range(5):
        true = True
        for j in range(5):
            if check[j*5+i] == False:
                true = False
            if true == False:
                break
        if true == True:
            line+=1

    for j in range(5):
        true = True
        for i in range(5):
            if check[j*5+i] == False:
                true = False
            if true == False:
                break
        if true == True:
            line+=1

    true = True
    for i in range(5):
        if check[i*5+i] == False:
            true = False
        if true == False:
            break
    if true == True:
        line+=1

    true = True
    for i in range(5):
        if check[(4-i)*5+i] == False:
            true = False
        if true == False:
            break
    if true == True:
        line+=1
    #print("You already have {} lines.".format(line))
    if line >= 5:
        return True
    else:
        return False

numbers= 0
maps = {} #username : chessboard[]
second = 60
Chessboard = []
currentChessboard = [0]*25
new_connection = True
end = False
winner = ''

def server_broadcast_socket(host, port):
    global numbers, maps, second, Chessboard, currentChessboard, new_connection, end, winner
    url = "tcp://{}:{}".format(host, port)
    context = zmq.Context()
    b_socket = context.socket(zmq.PUB)
    b_socket.bind(url)
    
    for i in range(60):
        if i % 5 == 0:
            print("[PUB] Waiting for {} seconds...".format(second))
            b_socket.send_string("Time,{}".format(second))
        time.sleep(1)
        second-=1

    print("\nGame Start\n")
    b_socket.send_string("\nGame Start !\n")
    new_connection = False
    
    Chessboard = list(range(1,26))
    random.shuffle(Chessboard)
    
    for i in range(25):
        print(Chessboard[i], end =' ')
        if i % 5 == 4:
            print()
    print()

    i = 0
    while i < 25:
        if end:
            break
        currentChessboard[i] = Chessboard[i]
        b_socket.send_string('Num,{}'.format(str(Chessboard[i])))
        print('[PUB] Number is {}'.format(Chessboard[i]))
        time.sleep(3)
        i += 1

    b_socket.send_string("Bingo,{}".format(winner))
    print("[PUB] Congratulations to '{}' win the game!".format(winner))
    print("[PUB] End")

#set to deamon
def server_ordinary_socket(host, port):
    global numbers, maps, second, Chessboard, currentChessboard, new_connection, end, winner
    url = "tcp://{}:{}".format(host, port)
    context = zmq.Context()
    socket = context.socket(zmq.REP)
    #socket.setsockopt(zmq.LINGER, 0)
    socket.bind(url)
    poller = zmq.Poller()
    poller.register(socket, zmq.POLLIN)

    
    while new_connection:#change to poll
        sockets = dict(poller.poll(500))  #poll   #idle for 0.5 second
        if socket in sockets:
            data = socket.recv_string()
        #data = socket.recv_string()
            splitText= data.split(',')
            thisMap = list(map(int,splitText[1:]))
            name = splitText[0]
            maps[splitText[0]] = thisMap  #this
            numbers += 1
            message = "{},{}".format(numbers, second)
            socket.send_string(message)
            print("'{}' chooses: {}".format(name, maps[name]))


    while True:
        data = socket.recv_string()
        splitData = data.split(',')
        if splitData[0] == 'Bingo':

            if checkBingo(maps[splitData[1]], currentChessboard):
                winner = splitData[1]    #name
                end = True
                socket.send_string("You are the Winner!")
                break
            else:
                socket.send_string("You are wrong!")



def publisher(interface = '127.0.0.1', port = '11129'):
    if interface == '':
        interface = '*'

    t1 = threading.Thread(target=server_ordinary_socket, args=((interface, port)))
    t2 = threading.Thread(target=server_broadcast_socket, args=(interface, port+1))
    t1.start()
    t2.start()


def client_ordinary_socket(host, port, ready): 
    global Chessboard, end
    url = "tcp://{}:{}".format(host, port)
    context = zmq.Context()
    socket = context.socket(zmq.REQ)
    socket.connect(url)
    text = input("Please enter your name, and 25 different numbers from 1 to 25 in random order, seperated by ','\n")
    socket.send_string(text)
    textList = text.split(',')
    Chessboard = list(map(int,textList[1:]))
    name = textList[0]
    print("\nThis is your chessboard:\n")
    for i in range(25):
        print('%2d' % Chessboard[i], end=' ')
        if i%5==4:
            print()
    print()
    message = socket.recv_string()
    mesList = message.split(',')
    print("You are the {} player.\nThe game will start in {} seconds.".format(mesList[0],mesList[1]))
    ready.set()
    
    while True:
        if end:
            socket.send_string('Bingo,{}'.format(name))
            data = socket.recv_string()
            print(data)
            if data != 'Failure!':
                break


def client_broadcast_socket(host, port, ready):
    global Chessboard, currentChessboard, end
    ready.wait()
    url = "tcp://{}:{}".format(host, port)
    context = zmq.Context()
    b_socket = context.socket(zmq.SUB)
    b_socket.connect(url)
    b_socket.setsockopt_string(zmq.SUBSCRIBE,'')
    i = 0
    while True:
        #print("here!")
        data = b_socket.recv_string();
        splitdata = data.split(',')
        if splitdata[0] == 'Bingo':
            print("Congratulations to '{}' win the game!".format(splitdata[1]))
            break
        elif splitdata[0] == 'Time':
            print('{} seconds left to start.'.format(splitdata[1]))
        elif splitdata[0] == 'Num':
            currentChessboard[i] = int(splitdata[1])
            print('Number is :', splitdata[1])
            i += 1
            if checkBingo(Chessboard, currentChessboard):
                end = True
                print('Bingo!(You)')
        else :
            print(data)




def subscriber(network, port):
    ready = threading.Event()
    if network == '':
        network = '*'
    t1 = threading.Thread(target=client_ordinary_socket, args=(network, port, ready), daemon=True)  #daemon
    t2 = threading.Thread(target=client_broadcast_socket, args=(network, port+1, ready))
    t1.start()
    t2.start()

if __name__ == '__main__':
    choices = {'subscriber': subscriber, 'publisher': publisher}
    parser = argparse.ArgumentParser(description='Send, receive UDP broadcast')
    parser.add_argument('role', choices=choices, help='which role to take')
    parser.add_argument('host', help='interface the server listens at;'
                               ' network the client sends to')
    parser.add_argument('-p', metavar='port', type=int, default=1060,
                        help='TCP port (default 1060)')
    args = parser.parse_args()
    function = choices[args.role]
    function(args.host, args.p)