Python基於Socket實現簡易多人聊天室
阿新 • • 發佈：2020-11-29
##前言
套接字(Sockets)是雙向通訊通道的端點。 套接字可以在一個程序內，在同一機器上的程序之間，或者在不同主機的程序之間進行通訊，主機可以是任何一臺有連線網際網路的機器。
套接字可以通過多種不同的通道型別實現：Unix域套接字，TCP，UDP等。 套接字型檔提供了處理公共傳輸的特定類，以及一個用於處理其餘部分的通用介面。
###socket模組：
要建立套接字，必須使用套接字模組中的socket.socket()函式，該函式具有一般語法
```
s = socket.socket (socket_family, socket_type, protocol = 0)
```
引數|描述|
----|----|
socket_family | 它的值可以是：AF_UNIX或AF_INET，如前所述。
socket_type | 它的值可以是：SOCK_STREAM或SOCK_DGRAM。
protocol | 這通常被省略，預設為0。
###常用方法:
序號|方法|描述
----|----|---
1|s.bind()|此方法將地址(主機名，埠號對)繫結到套接字。
2|s.recvfrom()|此方法接收UDP訊息，返回值是一對（位元組， 地址） ，其中位元組是代表接收到的資料的位元組物件，而地址是傳送資料的套接字的地址
3|s.sendto()|此方法傳送UDP訊息，將資料傳送到套接字。該套接字不應連線到遠端套接字，因為目標套接字是由address指定的
4|s.close()|此方法關閉套接字，套接字物件上所有以後的操作都將失敗。遠端將不再接收任何資料（在清除排隊的資料之後）。套接字在被垃圾回收時會自動關閉
5|s.gethostname()|返回主機名，返回一個字串，其中包含當前正在執行Python直譯器的計算機的主機名。
##示例1
###伺服器端
```
#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("來自伺服器" + 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("來自伺服器" + str(addr) + "的訊息:")
print(msg.decode('utf-8'))
except:
s.close()
```
服務端示例
客戶端示例
簡易的UDP聊天實現了,下面我們來優化一下示例。
##示例2
服務端：
```
#server.py
import socket
import logging
def main():
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 建立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字典裡則執行以下程式碼
for address in user: # 從user遍歷資料出來address
s.sendto(data + ' 進入聊天室...'.encode('utf-8'), address) # 傳送user字典的data和address到客戶端
user[addr] = data.decode('utf-8') # 接收的訊息解碼成utf-8並存在字典user裡,鍵名定義為addr
continue # 如果addr在user字典裡，跳過本次迴圈
if 'EXIT'.lower() in data.decode('utf-8'):#如果EXIT在傳送的data裡
name = user[addr] #user字典addr鍵對應的值賦值給變數name
user.pop(addr) #刪除user裡的addr
for address in user: #從user取出address
s.sendto((name + ' 離開了聊天室...').encode(), address) #傳送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時間執行下面的程式碼
s.sendto(data, address) #傳送data和address到客戶端
except ConnectionResetError:
logging.warning('Someone left unexcept.')
if __name__ == '__main__':
main()
```
客戶端：
```
#clinet.py
import socket
import threading
def recv(sock, addr):
'''
一個UDP連線在接收訊息前必須要讓系統知道所佔埠
也就是需要send一次，否則win下會報錯
'''
sock.sendto(name.encode('utf-8'), addr)
while True:
data = sock.recv(1024)
print(data.decode('utf-8'))
def send(sock, addr):
'''
傳送資料的方法
引數：
sock：定義一個例項化socket物件
server：傳遞的伺服器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():
'''
主函式執行方法，通過多執行緒來實現多個客戶端之間的通訊
'''
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()
```
支援多人的簡易聊天室示例，多個客戶端通過一個伺服器進行之間通訊
![](https://img2020.cnblogs.com/blog/1510016/202011/1510016-20201129130322475-4266589