怎么在python中利用socketserver实现并发套接字功能?针对这个问题,这篇文章详细介绍了相对应的分析和解答,希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
Python是一种编程语言,内置了许多有效的工具,Python几乎无所不能,该语言通俗易懂、容易入门、功能强大,在许多领域中都有广泛的应用,例如最热门的大数据分析,人工智能,Web开发等。
目录结构如下:
测试文件请放在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: PyCharm import json import time import hashlib import struct import os from 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()
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: PyCharm from socket import * import os,sys import hashlib import struct import math import json FILE_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)
关于怎么在python中利用socketserver实现并发套接字功能问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注亿速云行业资讯频道了解更多相关知识。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。