文件夹传输：基于python、TCP

文件夹传输：基于python、TCP

服务端

import os
import socket
import time
# from traversalfolder import dirwalkermsg


# 定义允许传输的文件大小
RECVSIZE = 1024 * 1024 * 10
# 定义标签，用于标记文件夹名字与文件名字
label = '#97'*100


def dirwalker(foldername):
    """
    此方法用于遍历目标文件夹
    :param foldername:
    :return: 一个生成器
    """
    # 遍历文件夹，返回生成器
    walkers = os.walk(foldername)
    for walker in walkers:
        dirname = walker[0]
        files = walker[2]

        # 返回标记后文件夹的名称
        yield dirname + label

        for file in files:
            file = dirname + '/' + file
            # 返回标记后文件的名称
            yield label + file + label

            with open(file, 'rb') as f:
                msg = f.read()
                # 返回文件内容
                yield msg if msg != b'' else b' '


def dirwalkermsg(foldername, source='sources/'):
    """
    用于生产字节流信息
    :param foldername: 客户端要copy的文件夹名称
    :param source: 服务端数据源
    :return: 返回一个字节流生成器
    """

    foldername = source + foldername
    if not os.path.isdir(foldername):