python--基於socket網路編程_ZenDei技術網路在線

Python 提供了兩個基本的 socket 模塊。第一個是 Socket，它提供了標準的 BSD Sockets API。第二個是 SocketServer，它提供了伺服器中心類，可以簡化網路伺服器的開發。 1、Scoket類型套接字格式： socket(family,type[,prot ...

Python 提供了兩個基本的 socket 模塊。

第一個是 Socket，它提供了標準的 BSD Sockets API。

第二個是 SocketServer，它提供了伺服器中心類，可以簡化網路伺服器的開發。

1、Scoket類型

套接字格式：

socket(family,type[,protocal]) 使用給定的地址族、套接字類型、協議編號（預設為0）來創建套接字。

1.1、socket.AF_UNIX #只能夠用於單一的Unix系統進程間通信

1.2、socket.AF_INET #伺服器之間網路通信

1.3、socket.AF_INET6 #IPv6

1.4、socket.SOCK_STREAM #流式socket , for TCP

1.5、socket.SOCK_DGRAM #數據報式socket , for UDP

1.6、socket.SOCK_RAW #原始套接字，普通的套接字無法處理ICMP、IGMP等網路報文，而SOCK_RAW可以；其次，SOCK_RAW也可以處理特殊的IPv4報文；此外，利用原始套接字，可以通過IP_HDRINCL套接字選項由用戶構造IP頭。

1.7、socket.SOCK_SEQPACKET #可靠的連續數據包服務

1.8、創建TCP Socket： s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

1.9、創建UDP Socket： s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

2、Socket 函數
註意點:
1）TCP發送數據時，已建立好TCP連接，所以不需要指定地址。UDP是面向無連接的，每次發送要指定是發給誰。
2）服務端與客戶端不能直接發送列表，元組，字典。需要字元串化repr(data)。

2.1、服務端socket函數

s.bind(address) #將套接字綁定到地址, 在AF_INET下,以元組（host,port）的形式表示地址.

s.listen(backlog) #開始監聽TCP傳入連接。backlog指定在拒絕連接之前，操作系統可以掛起的最大連接數量。該值至少為1，大部分應用程式設為5就可以了。

s.accept() #接受TCP連接並返回（conn,address）,其中conn是新的套接字對象，可以用來接收和發送數據。address是連接客戶端的地址。

2.2、客戶端socket函數

s.connect(address) #連接到address處的套接字。一般address的格式為元組（hostname,port），如果連接出錯，返回socket.error錯誤。

s.connect_ex(adddress) #功能與connect(address)相同，但是成功返回0，失敗返回errno的值。

2.3、公共socket函數

s.recv(bufsize[,flag]) #接受TCP套接字的數據。數據以字元串形式返回，bufsize指定要接收的最大數據量。flag提供有關消息的其他信息，通常可以忽略。

s.send(string[,flag]) #發送TCP數據。將string中的數據發送到連接的套接字。返回值是要發送的位元組數量，該數量可能小於string的位元組大小。

s.sendall(string[,flag]) #完整發送TCP數據。將string中的數據發送到連接的套接字，但在返回之前會嘗試發送所有數據。成功返回None，失敗則拋出異常。

s.recvfrom(bufsize[.flag]) #接受UDP套接字的數據。與recv()類似，但返回值是（data,address）。其中data是包含接收數據的字元串，address是發送數據的套接字地址。

s.sendto(string[,flag],address) #發送UDP數據。將數據發送到套接字，address是形式為（ipaddr，port）的元組，指定遠程地址。返回值是發送的位元組數。

s.close() #關閉套接字。

s.getpeername() #返回連接套接字的遠程地址。返回值通常是元組（ipaddr,port）。

s.getsockname() #返回套接字自己的地址。通常是一個元組(ipaddr,port)

s.setsockopt(level,optname,value) #設置給定套接字選項的值。

s.getsockopt(level,optname[.buflen]) #返回套接字選項的值。

s.settimeout(timeout) #設置套接字操作的超時期，timeout是一個浮點數，單位是秒。值為None表示沒有超時期。一般，超時期應該在剛創建套接字時設置，因為它們可能用於連接的操作（如connect()）

s.gettimeout() #返回當前超時期的值，單位是秒，如果沒有設置超時期，則返回None。

s.fileno() #返回套接字的文件描述符。

s.setblocking(flag) #如果flag為0，則將套接字設為非阻塞模式，否則將套接字設為阻塞模式（預設值）。非阻塞模式下，如果調用recv()沒有發現任何數據，或send()調用無法立即發送數據，那麼將引起socket.error異常。

s.makefile() #創建一個與該套接字相關連的文件

3、socket編程思路

3.1、TCP服務端：

1 創建套接字，綁定套接字到本地IP與埠

# socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.bind()

2 開始監聽連接 #s.listen()

3 進入迴圈，不斷接受客戶端的連接請求 #s.accept()

4 然後接收傳來的數據，併發送給對方數據 #s.recv() , s.sendall()

5 傳輸完畢後，關閉套接字 #s.close()

3.2、TCP客戶端:

1 創建套接字，連接遠端地址

# socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.connect()

2 連接後發送數據和接收數據 # s.sendall(), s.recv()

3 傳輸完畢後，關閉套接字 #s.close()

4、模擬聊天機器人實例

4.1、socket_server.py 服務端

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
Socket伺服器端
"""
import socket

sk = socket.socket()
sk.bind(('127.0.0.1', 8080,))
sk.listen(5)
#接收客戶端請求
#連接，客戶端地址信息
while True:
    conn, address = sk.accept()  #accept()阻塞
    conn.sendall(bytes('歡迎致電10086', encoding='utf-8'))
    while True:
        ret_byte = conn.recv(1024)
        ret_str = str(ret_byte, encoding='utf-8')
        if ret_str == 'q':
            break
        conn.sendall(bytes('工號10086回覆您：' + ret_str, encoding='utf-8'))

4.2、socket_client.py 客戶端

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
socket客戶端
"""
import socket

ck = socket.socket()
ck.connect(('127.0.0.1', 8080))
ret_byte = ck.recv(1024)
ret_str = str(ret_byte, encoding='utf-8')
print(ret_str)
while True:
    inp = input("請輸入內容，按q退出 \n >>>")
    if inp == 'q':
        ck.sendall(bytes(inp, encoding='utf-8'))
        break
    else:
        ck.sendall(bytes(inp, encoding='utf-8'))
        ret_str = str(ck.recv(1024), encoding='utf-8')
        print(ret_str)

ck.close()

5、基於socket實現文件上傳實例

5.1、ftp_server.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
模擬發送文件的伺服器端
"""

import socket

sk = socket.socket()
sk.bind(('127.0.0.1', 8080,))  #綁定套接字到本地IP與埠
sk.listen(5)  #開始監聽連接
#接收客戶端請求
#連接，客戶端地址信息
while True:
    conn, address = sk.accept()  # accept()阻塞
    conn.sendall(bytes('歡迎訪問我的ftp', encoding='utf-8'))
    #先接收文件大小，然後再接收文件
    ret_size = str(conn.recv(1024), encoding='utf-8')
    conn.sendall(bytes('start', encoding='utf-8')) #給客戶端發送一個確認信息，通知客戶端開始發送文件，解決數據粘包的問題。
    total_size = int(ret_size)  #文件總大小
    has_recv = 0  #已經接收的文件大小
    f = open('new.png', 'wb')
    #接收文件內容，直到獲取完成
    while True:
        if total_size == has_recv:
            break
        data = conn.recv(1024)
        f.write(data)
        has_recv += len(data)
    f.close()

5.2、ftp_client.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
模擬發送文件的客戶端
"""
import socket
import os

ck = socket.socket()
ck.connect(('127.0.0.1', 8080))
ret_byte = ck.recv(1024)  #接收伺服器端返回的位元組數據
ret_str = str(ret_byte, encoding='utf-8')  #將位元組數據轉換成字元串
print(ret_str)
#首先向伺服器發送文件的大小
file_size = os.stat('file1.png').st_size
ck.sendall(bytes(str(file_size), encoding='utf-8'))

ck.recv(1024)  #接收一下伺服器端的確認信息，再開始發送文件，解決數據粘包引起的系統異常問題
#發送文件
with open('file1.png', 'rb') as f:
    for line in f:
        ck.sendall(line)
ck.close()

6、socketserver模塊實現併發操作

使用：
1、創建類，必須繼承socketserver.BaseRequestHandler
2、定義handle方法
3、server_forever

socketserver模塊實現併發操作實例：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
socketserver模塊實現併發操作
"""
import socketserver

class MyServer(socketserver.BaseRequestHandler):

    def handle(self):
        #self.request, self.client_address, self.server
        conn = self.request
        conn.sendall(bytes('歡迎致電10086', encoding='utf-8'))
        while True:
            ret_byte = conn.recv(1024)
            ret_str = str(ret_byte, encoding='utf-8')
            if ret_str == 'q':
                break
            conn.sendall(bytes('工號10086回覆您：' + ret_str, encoding='utf-8'))

if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyServer)
    server.serve_forever()