Python基于Socket實現(xiàn)簡易多人聊天室的示例代碼

更新時間：2020年11月29日 14:55:44 作者：Huny

這篇文章主要介紹了Python基于Socket實現(xiàn)簡易多人聊天室的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前言

套接字(Sockets)是雙向通信信道的端點。套接字可以在一個進程內(nèi)，在同一機器上的進程之間，或者在不同主機的進程之間進行通信，主機可以是任何一臺有連接互聯(lián)網(wǎng)的機器。

套接字可以通過多種不同的通道類型實現(xiàn)：Unix域套接字，TCP，UDP等。套接字庫提供了處理公共傳輸?shù)奶囟悾约耙粋€用于處理其余部分的通用接口。

socket模塊：

要創(chuàng)建套接字，必須使用套接字模塊中的socket.socket()函數(shù)，該函數(shù)具有一般語法

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

參數(shù)	描述
socket_family	它的值可以是：AF_UNIX或AF_INET，如前所述。
socket_type	它的值可以是：SOCK_STREAM或SOCK_DGRAM。
protocol	這通常被省略，默認為0。

常用方法:

序號	方法	描述
1	s.bind()	此方法將地址(主機名，端口號對)綁定到套接字。
2	s.recvfrom()	此方法接收UDP消息，返回值是一對（字節(jié)，地址），其中字節(jié)是代表接收到的數(shù)據(jù)的字節(jié)對象，而地址是發(fā)送數(shù)據(jù)的套接字的地址
3	s.sendto()	此方法發(fā)送UDP消息，將數(shù)據(jù)發(fā)送到套接字。該套接字不應(yīng)連接到遠程套接字，因為目標套接字是由address指定的
4	s.close()	此方法關(guān)閉套接字，套接字對象上所有以后的操作都將失敗。遠端將不再接收任何數(shù)據(jù)（在清除排隊的數(shù)據(jù)之后）。套接字在被垃圾回收時會自動關(guān)閉
5	socket.gethostname()	返回主機名，返回一個字符串，其中包含當前正在執(zhí)行Python解釋器的計算機的主機名。

示例1

服務(wù)器端

#sever.py
import socket

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
host = socket.gethostname()
port = 8088
s.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.py
import socket
s = 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()

服務(wù)端示例

客戶端示例

簡易的UDP聊天實現(xiàn)了,下面我們來優(yōu)化一下示例。

示例2

服務(wù)端：

#server.py
import socket
import logging

def main():
  s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 創(chuàng)建socket對象

  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個變量data和addr里
      if not addr in user: # 如果addr不在user字典里則執(zhí)行以下代碼
        for address in user: # 從user遍歷數(shù)據(jù)出來address
          s.sendto(data + ' 進入聊天室...'.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鍵對應(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時間執(zhí)行下面的代碼
            s.sendto(data, address)   #發(fā)送data和address到客戶端

    except ConnectionResetError:
      logging.warning('Someone left unexcept.')

if __name__ == '__main__':
  main()

客戶端：

#clinet.py
import socket
import threading

def recv(sock, addr):
  '''
  一個UDP連接在接收消息前必須要讓系統(tǒng)知道所占端口
  也就是需要send一次，否則win下會報錯
  '''
  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：定義一個實例化socket對象
      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í)行方法，通過多線程來實現(xiàn)多個客戶端之間的通信
  '''
  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("-----歡迎來到聊天室,退出聊天室請輸入'EXIT(不分大小寫)'-----")
  name = input('請輸入你的名稱:')
  print('-----------------%s------------------' % name)
  main()

支持多人的簡易聊天室示例，多個客戶端通過一個服務(wù)器進行之間通信