Python 基于 Socket 實(shí)現(xiàn)群聊

s = socket.socket (socket_family, socket_type, protocol = 0)

#sever.pyimport socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)host = socket.gethostname()port = 8088s.bind((host,port))try:    while True:        receive_data,addr = s.recvfrom(1024)        print("來自服務(wù)器" + str(addr) + "的消息:")        print(receive_data.decode('utf-8'))        msg = input('please input send to msg:')        s.sendto(msg.encode('utf-8'),addr)except:    s.close()

#client.pyimport sockets = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)try:    while True:        host = socket.gethostname()        port = 8088        send_data = input('please input msg:')        s.sendto(send_data.encode('utf-8'),(host,port))        msg,addr = s.recvfrom(1024)        print("來自服務(wù)器" + str(addr) + "的消息:")        print(msg.decode('utf-8'))except:    s.close()

#server.pyimport socketimport logging
def main():    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # 創(chuàng)建socket對(duì)象
    addr = ('127.0.0.1', 9999)    s.bind(addr)  # 綁定地址和端口
    logging.info('UDP Server on %s:%s...', addr[0], addr[1])
    user = {}  # 存放字典{addr:name}    while True:        try:            data, addr = s.recvfrom(1024)  # 等待接收客戶端消息存放在2個(gè)變量data和addr里            if not addr in user:  # 如果addr不在user字典里則執(zhí)行以下代碼                for address in user:  # 從user遍歷數(shù)據(jù)出來address                    s.sendto(data + ' 進(jìn)入聊天室...'.encode('utf-8'), address)  # 發(fā)送user字典的data和address到客戶端                user[addr] = data.decode('utf-8')  # 接收的消息解碼成utf-8并存在字典user里,鍵名定義為addr                continue  # 如果addr在user字典里，跳過本次循環(huán)
            if 'EXIT'.lower() in data.decode('utf-8'):#如果EXIT在發(fā)送的data里                name = user[addr]   #user字典addr鍵對(duì)應(yīng)的值賦值給變量name                user.pop(addr)      #刪除user里的addr                for address in user:    #從user取出address                    s.sendto((name + ' 離開了聊天室...').encode(), address)     #發(fā)送name和address到客戶端            else:                   print('"%s" from %s:%s' %(data.decode('utf-8'), addr[0], addr[1]))                  for address in user:    #從user遍歷出address                    if address != addr:  #address不等于addr時(shí)間執(zhí)行下面的代碼                        s.sendto(data, address)     #發(fā)送data和address到客戶端
        except ConnectionResetError:            logging.warning('Someone left unexcept.')
if __name__ == '__main__':    main()

#clinet.pyimport socketimport threading
def recv(sock, addr):    '''    一個(gè)UDP連接在接收消息前必須要讓系統(tǒng)知道所占端口    也就是需要send一次，否則win下會(huì)報(bào)錯(cuò)    '''    sock.sendto(name.encode('utf-8'), addr)    while True:        data = sock.recv(1024)        print(data.decode('utf-8'))

def send(sock, addr):    '''        發(fā)送數(shù)據(jù)的方法        參數(shù)：            sock：定義一個(gè)實(shí)例化socket對(duì)象            server：傳遞的服務(wù)器IP和端口    '''    while True:        string = input('')        message = name + ' : ' + string        data = message.encode('utf-8')        sock.sendto(data, addr)        if string.lower() == 'EXIT'.lower():            break
def main():    '''        主函數(shù)執(zhí)行方法，通過多線程來實(shí)現(xiàn)多個(gè)客戶端之間的通信    '''    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)    server = ('127.0.0.1', 9999)    tr = threading.Thread(target=recv, args=(s, server), daemon=True)    ts = threading.Thread(target=send, args=(s, server))    tr.start()    ts.start()    ts.join()    s.close()
if __name__ == '__main__':    print("-----歡迎來到聊天室,退出聊天室請(qǐng)輸入'EXIT(不分大小寫)'-----")    name = input('請(qǐng)輸入你的名稱:')    print('-----------------%s------------------' % name)    main()