使用python的socket进行多个文件传输

使用python的socket进行多个文件传输

在网上学习到了基本的socket使用，但与自己想要的效果不同，所以进行了修改实现了在局域网内进行多文件传输，但是还有一些小bug不知道为什么有时候进行传输时，会显示您的主机中的软件中止了一个已建立的连接。接下来先看代码吧。
首先是cilient端代码

import socket
import os



def postData():
    # 实例化
    sk = socket.socket()
    # 定义连接的ip和port
    ip_port = ('127.0.0.1', 9999)
    # 服务器连接
    sk.connect(ip_port)
    # 文件上传，获取传送文件目录
    filenames = os.listdir(r"D:\temp")
    print(filenames)
    newfilename = "/".join(filenames)
    #将目录上传
    sk.send(newfilename.encode())
    for name in filenames:
    #打开文件
        with open('D:\\temp\\'+name,'rb') as f:
            #按每一段分割文件上传
            for i in f:
                sk.send(i)
                #等待接收完成标志
                data=sk.recv(1024)
                #判断是否真正接收完成
                if data != b'success':
                    break
        #给服务端发送结束信号
        sk.send('quit'.encode())
    sk.close()
if __name__ == "__main__":
      postData()

然后是server端代码

import socket, threading

#实例化
sk = socket.socket()
#定义连接的ip和port
ip_port = ('127.0.0.1',9999)
#绑定端口
sk.bind(ip_port)
#最大连接数
sk.listen(5)
#进入循环接收数据
conn, address = sk.accept()
print("文件接收开始")
testFilenames=""
i=1
list1=["1"]
def setValue(value):
    list1=value
    return list1



#根据filename创建相应的文件来接受数据
def creatName(filenames):
        for name in filenames:
                while True:
                        with open(name,'ab') as f:
                            #接收数据
                            data = conn.recv(1024)
                            if data == b'quit':
                                break
                            #写入文件
                            f.write(data)
                            #接受完成标志
                            conn.send('success'.encode())
        print("文件接收完成")
if __name__ == "__main__":
        newfilenames = conn.recv(1024)
        testFilenames = newfilenames.decode()
        list1 = setValue(testFilenames.split("/"))
        creatName(list1)
        #关闭连接
        sk.close()

大部分的代码网上都是从网络上查到的只是进行了一部分改动，而且现在还只是即用即传未实现文件连接常开。第一次进行博客发表，就先这样吧。