python利用socketserver實現並發通訊端功能，pythonsocketserver

python利用socketserver實現並發通訊端功能，pythonsocketserver

本文實現利用python的socketserver這個強大的模組實現通訊端的並發

目錄結構如下：

測試檔案請放在server_file檔案夾裡面

server.py

#!/usr/bin/env python# -*- coding: gbk -*-# @Version  : Python 3.5.2# @Time    : 2018/1/24 10:29# @Author  : Ncp# @File    : server.py# @Software: PyCharmimport jsonimport timeimport hashlibimport structimport osfrom socketserver import *FILE_PATH = os.path.dirname(os.path.abspath(__file__))+'\\server_file'class MYserver(BaseRequestHandler):   # 設定一個類，基礎BaseRequestHandler這個類    def handle(self):                 # 這個方法下添加通訊功能（和上面建立類一樣，這是socketserver的固定模式）        print(self.client_address)        '''        :functions: 使用socketserver的並發通訊端，提供用戶端下載檔案，並對檔案進行MD5加密        '''        while True:            try:                data = self.request.recv(1024)                data_recv = data.decode('gbk').split()                if not os.path.exists(FILE_PATH+r'\%s' %data_recv[1]):                    self.request.send('file is not found'.encode('gbk'))                    continue                else:                    data = self.request.send('1'.encode('gbk'))  # 這裡發現小問題，不回複一個資訊的話，發送給用戶端的包頭居然成了沒有封裝                    FILE_SIZE = os.path.getsize(FILE_PATH+r'\%s' %data_recv[1])                    with open(FILE_PATH+r'\%s' %data_recv[1],'rb')as f:                        hash_file = f.read()                    m = hashlib.md5()                    m.update(hash_file)                    m_hex = m.hexdigest()                    file_header = {'filename':data_recv[1],                                   'filesize':FILE_SIZE,                                   'md5':m_hex,                                   'time':time.strftime('%Y-%m-%d-%X',time.localtime())                                   }                    # 包頭資訊序列化                    file_header_dump = json.dumps(file_header)                    # 編譯成2進位                    file_header_bytes = file_header_dump.encode('gbk')                    # 封裝前序                    file_header_struct = struct.pack('i',len(file_header_bytes))                    # 發送前序                    self.request.send(file_header_struct)                    # 發送報文內容                    self.request.send(file_header_bytes)                    # 傳送檔案資料                    send_size = 0                    with open(FILE_PATH+r'\%s' %data_recv[1] , 'rb')as f:                        for i in f:                            self.request.send(i)                            send_size += len(i)   # 這裡後續可以拓展一個進度或者網速顯示功能            except Exception:                self.request.close()if __name__ == '__main__':    server = ThreadingTCPServer(('127.0.0.1',8080),MYserver)  # windows下只能開啟多線程    server.serve_forever()

View Code

client.py

#!/usr/bin/env python# -*- coding: gbk -*-# @Version  : Python 3.5.2# @Time    : 2018/1/24 10:29# @Author  : Ncp# @File    : client.py# @Software: PyCharmfrom socket import *import os,sysimport hashlibimport structimport mathimport jsonFILE_PATH = os.path.dirname(os.path.abspath(__file__))+'\\client_file'# 顯示下載進度條功能，可以拓展為顯示下載速度（提示，因為每次傳輸4096個位元組，那麼下載網速為KB/S,1KB個位元組=1024B（位元組），那麼1s傳輸了多少個位元組呢？）def progress(recvd, total):    fraction = '{:.0%}'.format(recvd / total)    sys.stdout.write('\r[%-30s] %s' % ('#' * int(math.floor(recvd * 30 / total)), fraction))    sys.stdout.flush()    if recvd == total:        sys.stdout.write('\n')# 主函數def run(ip,addr):    client = socket(AF_INET,SOCK_STREAM)    client.connect((ip,addr))    while True:        user_input = input('>>').strip()        cmd = user_input.split()        if len(cmd) != 2:            print('input format is error please use:get xx')            continue        if cmd[0] == 'get':            client.send(user_input.encode('gbk'))            data = client.recv(1024)            data_recv = data.decode('gbk')            if data_recv == 'file is not found':                print(data_recv)                continue        else:            print('commands is not found')            continue        # 收包頭，然後一系列處理        header = client.recv(4)        if not header:break        header_json_len = struct.unpack('i', header)[0]        header_json_bytes = client.recv(header_json_len)        header_josn = header_json_bytes.decode('gbk')        header_dic = json.loads(header_josn)        # 去除包頭內容進行下載        print(header_dic)        data_len = header_dic['filesize']        data_file = header_dic['filename']        data_md5 = header_dic['md5']        recv_size = 0        with open(FILE_PATH+r'\%s' %data_file,'wb')as fw:            while recv_size < data_len:                recv_data = client.recv(4096)                recv_size += len(recv_data)                fw.write(recv_data)                progress(recv_size,data_len)            print('Download completion, start validation')        # 收到檔案後，讀取檔案進行加密，看是否與服務端下載的檔案一致！        with open(FILE_PATH+r'\%s' %data_file,'rb')as fr:            data_total = fr.read()        m = hashlib.md5()        m.update(data_total)        m_hex = m.hexdigest()        if m_hex == data_md5:            print('檔案驗證通過')        else:            print('檔案損壞,刪除檔案')            os.remove(FILE_PATH+r'\%s' %data_file)if __name__ == '__main__':    run('127.0.0.1',8080)

View Code

 

自己可以設定一個多使用者登入，然後測試，使用者下載同一個檔案，分別存入每個使用者自己的家目錄下面，效果更好。

當然現在也能測試，開啟兩個用戶端同時下載檔案。