socket实现文件上传（客户端向服务器端上传照片示例）

本示例在对socket有了基本了解之后，可以实现基本的文件上传。首先先介绍一下目录结构，server_data文件夹是用来存放客户端上传的文件，client_data是模拟客户端文件夹（目的是为了测试文件夹下的文件能否上传） ，还有两个主要的py文件，onload_server.py  onload_client.py 。

目录结构如图：

下面直接上两段主要py文件代码：

客户端（上传端）代码：

#__author__:Kelvin
#date:2020/2/16 11:09

import socket,os

sk=socket.socket()
address=("127.0.0.1",8000)
sk.connect(address)

while True:
    inp=input("请输入 命令|文件名：") # post|a.jpg
    size=os.path.getsize(inp.split("|")[1])
    # print(size)
    inp=inp+"|%s"%size
    sk.send(inp.encode("utf8"))
    sk.recv(1024) # 接收 避免粘包发送的无效数据
    f=open(inp.split("|")[1],"rb")
    sent_size=0
    while sent_size!=size:
        sent_data=f.read(1024)
        sk.send(sent_data)
        sent_size+=len(sent_data)
        # print(sent_size)
    f.close()
    print("%s 文件上传成功..."%inp.split("|")[1].split("|")[-1])
    break
sk.close()

服务器端（接收端）代码：

#__author__:Kelvin
#date:2020/2/16 11:09

import socket

sk=socket.socket()
address=("127.0.0.1",8000)
sk.bind(address)
sk.listen(5)
while True:
    conn,addr=sk.accept()
    print(addr,"已连接...")
    while True:
        data=conn.recv(1024).decode("utf8")
        size=int(data.split("|")[2])
        file_name=data.split("|")[1].split("/")[-1]
        conn.send("1".encode("utf8")) # 避免粘包，无实际作用
        # print(size,file_name)
        f=open("server_data/%s"%file_name,"wb")
        saved_size=0
        while saved_size!=size:
            rev_data=conn.recv(1024)
            f.write(rev_data)
            saved_size+=len(rev_data)
            # print(saved_size)
        f.close()
        print("%s 文件接收成功..."%file_name)
        conn.close()
        break

运行输出结果展示：