Pygame multijoueur

import pygame as pg
import socket

#################################SETUP PYGAME##################################

pg.init()

screen = pg.display.set_mode((640,640))


###################SET THE CONNEXION WITH THE OTHER PLAYER#####################

def setup_serveur_connexion():
    my_socket = socket.socket()

    host_this_player = socket.gethostname()

    host_other_player = ''

    #meant to be changed

    if "imac" in host_this_player:
        host_other_player = 'pluton-1.home'
    else:
        host_other_player = 'imac-de-christian-mayeux.home'
        

    port = 6667

    my_socket.bind((host_this_player,port))

    my_socket.listen(5)

    print("done")
    connexion, adr = my_socket.accept()

    connexion.send((host_this_player+" is connected").encode())

    print(connexion.recv(1024).decode())
    return connexion

def setup_client_connexion():
    my_socket = socket.socket()

    host_this_player = socket.gethostname()

    host_other_player = ''

    #meant to be changed

    if "imac" in host_this_player:
        host_other_player = 'pluton-1.home'
    else:
        host_other_player = 'imac-de-christian-mayeux.home'
        

    port = 6667


    my_socket.connect((host_other_player,port))

    print("done")

    my_socket.send((host_this_player+" is connected").encode())

    print(my_socket.recv(1024).decode())

    return my_socket

if 'imac' in socket.gethostname():
    cxn =  setup_serveur_connexion()
else:
    cxn = setup_client_connexion()


###############################DEFINE CLASSES####################################

class Grille:
    def __init__(self,w,h):
        self.data = []
        for i in range(h):
            self.data.append([])
            for j in range(w):
                self.data[-1].append(None)
        for ligne in self.data:
            print(ligne)

Grille(4,3)

###################################MAIN LOOP##############################s######

done = False

tour = 0

ready = False

while not done:
    for event in pg.event.get():
        if event.type == pg.QUIT:
            done = True
    screen.fill("white")
    

    pg.display.update()

pg.quit()
cxn.close()