前言

套接字(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()

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