socketserver 模块实现并发及文件下载程序

本文主要是介绍socketserver 模块实现并发及文件下载程序，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

上章回顾

在讲解 socketserver 模块之前先补充一下上一章节的一个示例:

实现客户端从服务端下载文件的功能,能hash校验(Windows和Linux测试成功,代码比较low仅供观望)

• 服務端

# coding=utf-8
from socket import *
import json
import struct
import os,hashlibserver = socket(AF_INET,SOCK_STREAM)
# server.bind(("192.168.12.222",8090))
server.bind(("127.0.0.1",8090))
server.listen(5)while 1:print("connection...")conn,addr = server.accept()print(f"from {addr} conn")while 1:try:file_path = conn.recv(1024)file_path = os.path.normpath(file_path.decode("utf-8"))if not os.path.isfile(file_path):conn.send("4044".encode("utf-8"))else:file_size = os.path.getsize(file_path)file_name = os.path.basename(file_path)m = hashlib.md5()m.update(str(file_size).encode("utf-8"))md5 = m.hexdigest()header_dic = {"file_name":file_name,"file_size":file_size,"hash":md5}header_json = json.dumps(header_dic)header_bytes = header_json.encode("utf-8")header_bytes_len = struct.pack("i",len(header_bytes))conn.send(header_bytes_len)conn.send(header_bytes)with open(file_path,"rb")as f:for line in f:conn.send(line)except Exception:break

• 客户端

# coding=utf-8
from socket import *
import json
import struct
import os
import hashlib# 打印进度条
def progress(percent, symbol='█', width=40):if percent > 1:  # 超过 100% 的时候让其停在 1percent = 1  # 可以避免进度条溢出show_progress = ("▌%%-%ds▌" % width) % (int(percent * width) * symbol)print("\r%s %.2f%%" % (show_progress, percent * 100), end='')client = socket(AF_INET,SOCK_STREAM)
# client.connect(("192.168.12.222",8090))
client.connect(("127.0.0.1",8090))while True:file_path = input("Please enter the file path(q/exit)>>").strip()if file_path.lower() == "q":breakif len(file_path) == 0:continueto_path = input("Please enter the save directory(q/back)>>").strip()if to_path.lower() == "q":continueif not os.path.isdir(to_path):print("not find");continueelse:file_name = input("Please enter filename(q/back)>>").strip()if file_name.lower() == "q":continuegoal_path = os.path.join(to_path,file_name)client.send(file_path.encode("utf-8"))bytes_4 = client.recv(4)if bytes_4.decode("utf-8") == "4044":print("not find");continueelse:header_bytes_len = struct.unpack("i",bytes_4)[0]header_bytes = client.recv(header_bytes_len)header_dic = json.loads(header_bytes.decode("utf-8"))date_len = header_dic["file_size"]hash_md5 = header_dic["hash"]recv_len = 0with open(goal_path,"wb")as f:while 1:date = client.recv(1024)recv_len += len(date)percent = recv_len / date_len  # 接收的比例progress(percent, width=40)    # 进度条的宽度40f.write(date)if recv_len == date_len: breakm = hashlib.md5()m.update(str(os.path.getsize(goal_path)).encode("utf-8"))if hash_md5 == m.hexdigest():          # hash 值校验print("\nHash auth succeed\nFile saved...")else:os.remove(goal_path)               # 校验失败内容删除print("Hash auth failed!!")

• Windows 下测试

• Linux 下测试

• 两次的文件

使用 socketserver 模块实现并发

1.socketserver 模块简单介绍

基于 TCP 的套接字, 关键就是两个循环, 一个是连接循环, 另一个是通信循环, 分成两件事去做

socketserver 模块中有两大类, 一个是 server 类, 专门干连接的事, 一个是 request 类, 专门干通信的事

目前只是简单使用 socketserver 模块来实现并发效果, 后面章节再深入研究

2.基于 TCP 的 socketserver 自定义类中的属性和方法介绍

• handle( ) : 用于连接循环
• self.server : 套接字对象
• self.request : 建成的连接, 相当于前面用的 ‘conn’
• self.client_address : 客户端地址和端口号, 是个元组
• self.serve_forever( ) : 永久提供服务, 一个死循环, 对应的是连接循环
• socketserver.ThreadingTCPServer(bind_and_activate=True) : 第一个参数为服务端绑定的ip和端口, 第二的为指定的类, 第三个可以不写,默认为True,设置监听

3.基于 UDP 的 socketserver 自定义类中属性和方法介绍

• handle( ) : 用于连接循环
• self.request : 是一个元组, 第一个元素是客户端发来的数据, 第二部分是服务端的udp套接字对象
• self.client_address : 客户端ip和端口
• self.serve_forever( ) : 永久提供服务, 一个死循环, 对应的是连接循环
• socketserver.ThreadingUDPServer((“127.0.0.1”,8089),MyRequestHandler) : 第一个参数为服务端绑定的ip和端口, 第二的为指定的类

4.基于 TCP 实现并发

• 服务端

import socketserverclass MyRequestHandler(socketserver.BaseRequestHandler):  # 必须继承这个类来使用它的功能def handle(self):  # 用于通信循环while True:try:data = self.request.recv(1024)if len(data) == 0:breakself.request.send(data.upper())except ConnectionResetError:breakself.request.close()# 做绑定 ip和端口并设置监听的事, "bind_and_activate" 默认等于 "True"
s = socketserver.ThreadingTCPServer(("127.0.0.1",8089),MyRequestHandler,bind_and_activate=True)
s.serve_forever()  # 用于建立连接, 之后交给 handle 进行通信循环

• 客户端测试例( 可以开启多台)

from socket import *client = socket(AF_INET,SOCK_STREAM)
client.connect(("127.0.0.1",8089))while True:msg = input(">>").strip()if len(msg) == 0:continueclient.send(msg.encode("utf-8"))data = client.recv(1024)print(data.decode("utf-8"))client.close()

5.基于 UDP 实现并发

• 服务端

import socketserverclass MyRequestHandle(socketserver.BaseRequestHandler):def handle(self):while True:date,conn = self.requestprint(f"来自[{self.client_address}]的信息 : {date.decode('utf-8')}")conn.sendto(date.upper(),self.client_address)s = socketserver.ThreadingUDPServer(("127.0.0.1",8080),MyRequestHandle)
s.serve_forever()

• 客户端(可开启多台)

from socket import *client = socket(AF_INET,SOCK_DGRAM)while True:date = input(">>").strip()client.sendto(date.encode("utf-8"),("127.0.0.1",8080))res,addr = client.recvfrom(1024)print(f"来自服务端的消息 : {res.decode('utf-8')}")

这篇关于socketserver 模块实现并发及文件下载程序的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---