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

124次阅读
没有评论

共计 4482 个字符,预计需要花费 12 分钟才能阅读完成。

上章回顾

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

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

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

  • 服務端
# coding=utf-8
from socket import *
import json
import struct
import os,hashlib

server = 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% 的时候让其停在 1
        percent = 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":break
    if len(file_path) == 0:continue
    to_path = input("Please enter the save directory(q/back)>>").strip()
    if to_path.lower() == "q":continue
    if not os.path.isdir(to_path):
        print("not find");continue
    else:
        file_name = input("Please enter filename(q/back)>>").strip()
        if file_name.lower() == "q":continue
        goal_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");continue
    else:
        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 = 0
        with 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)    # 进度条的宽度 40
                f.write(date)
                if recv_len == date_len: break

        m = 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 下测试

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

  • Linux 下测试

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

  • 两次的文件

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

使用 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 socketserver

class MyRequestHandler(socketserver.BaseRequestHandler):  # 必须继承这个类来使用它的功能
    def handle(self):  # 用于通信循环
        while True:
            try:
                data = self.request.recv(1024)
                if len(data) == 0:break
                self.request.send(data.upper())
            except ConnectionResetError:
                break
        self.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:continue
    client.send(msg.encode("utf-8"))
    data = client.recv(1024)
    print(data.decode("utf-8"))

client.close()

5. 基于 UDP 实现并发

  • 服务端
import socketserver

class MyRequestHandle(socketserver.BaseRequestHandler):
    def handle(self):
        while True:
            date,conn = self.request
            print(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')}")
正文完
 
shawn
版权声明:本站原创文章,由 shawn 2023-06-16发表,共计4482字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)