[Python知识库]网络并发编程进阶版 实列自制服务端电影下载

服务端代码

import json
import socket
import struct
#mac本多次刷新链接，需要添加
from socket import SOL_SOCKET, SO_REUSEADDR

#socket实现各个主机之间互连
server = socket.socket()
#mac本多次刷新链接，需要添加
server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
server.bind(('127.0.0.1', 8080))  # 拨号
#半链接池，设置一次可接待几个需要链接客户，否则服务端不能accept（）客户
server.listen(5)


while True:
    sock, addr = server.accept()
    while True:
        # 1.先接收固定长度为4的字典报头数据
        recv_first = sock.recv(4)
        # 2.解析字典报头struct.unpack()[0]固定搭配，解压当前长度为4的字典报头数据
        dict_length = struct.unpack('i', recv_first)[0]
        # 3.接收字典数据
        real_data = sock.recv(dict_length)
        # 4.解析字典(json格式的bytes数据 loads方法会自动先解码 后反序列化)
        real_dict = json.loads(real_data)
        print(real_dict)
        # 5.获取字典中的各项数据
        data_length = real_dict.get('size')
        file_name = real_dict.get('file_name')
        # 接收数据
        recv_size = 0
        with open(file_name, 'wb') as f:
            while recv_size < data_length:
                data = sock.recv(1024)
                recv_size += len(data)
                f.write(data)

客户端

import json
import socket
import struct
import os

client = socket.socket()  # 客户端的链接行为
client.connect(('127.0.0.1', 8080))  # 拨号
while True:
    data_path = r'/Users/XXXXX/Movies'
    # os.listdir列出data_path路径下所有的文件
    movie_name_list = os.listdir(data_path)
    # print(os.listdir(data_path))
    # enumerate可以遍历集合对象还可以得到当前元素的索引位置
    for i, j in enumerate(movie_name_list, 1):
        print(i, j)
    chioce = input('以下选项请选择编号下载: ')
    if chioce.isdigit():
        chioce = int(chioce)
        if chioce in range(1, len(movie_name_list) + 1):
            movie_name = movie_name_list[chioce - 1]
            print(movie_name)
            # 拼接文件绝对路径
            movie_path = os.path.join(data_path, movie_name)
            # 定义一个字典数据
            data_dict = {
                'file_name': '古典音乐视频集合.mp4',
                'size': os.path.getsize(movie_path),
            }
            # print(data_dict)
            data_json = json.dumps(data_dict)
            # print(data_json)
            # 制作字典报头
            data_first = struct.pack('i', len(data_json))

            # 发送字典报头
            client.send(data_first)
            # 发送字典
            client.send(data_json.encode('utf8'))
            # 发送真实数据
            with open(movie_path, 'rb') as f:
                for line in f:
                    client.send(line)
    break