Python 文件上传(Socket)

1.服务器端

#__author: Lobin
#__date:   2018/1/21
import socket
import os
BASE_DIR=os.path.dirname(os.path.abspath(__file__))
sk=socket.socket()
address=('127.0.0.1',8080)
sk.bind(address)
sk.listen(3)
while True:
    conn,add=sk.accept()
    print(conn,add)
    while True:
        data=conn.recv(1024)
        cmd,filename,filesize=str(data,'utf8').split('|')
        path=os.path.join(BASE_DIR,'Up',filename)
        filesize=int(filesize)

        f=open(path,'ab')
        has_recv=0
        while has_recv!=filesize:
            data=conn.recv(1024)
            f.write(data)
            has_recv+=len(data)
        f.close()
        print('接收成功')
sk.close()

解析：约定服务器与客户端的上传格式为第一次发送文件大小信息，二次发送文件

1.1 cmd,filename,filesize=str(data,'utf8').split('|') 将客户端传送来的消息以“|”分割开来获取文件信息的命令cmd、文件名称filename、文件大小filesize

1.2 path=os.path.join(BASE_DIR,'Up',filename) 用路径拼接函数将本地路径BASE_DIR、文件夹Up、和文件名称拼接为服务器本地绝对路径

1.3  filesize=int(filesize) 将字符号转换为int

1.4  打开文件接收数据以bytes类型接收并写入ab模式

 

2.客户端

#__author: Lobin
#__date:   2018/1/21
import socket
import os
BASE_DIR=os.path.dirname(os.path.abspath(__file__))
sk=socket.socket()
address=('127.0.0.1',8080)
sk.connect(address)
while True:
    inp=input('>>>').strip()
    cmd,path=inp.split('|')
    path=os.path.join(BASE_DIR,path)
    file_name=os.path.basename(path)
    file_size=os.stat(path).st_size
    file_info='post|%s|%s'%(file_name,file_size)
    sk.sendall(bytes(file_info,'utf8'))

    f=open(path,'rb')
    has_sent=0
    while has_sent!=file_size:
        data=f.read(1024)
        sk.sendall(data)
        has_sent+=len(data)
    f.close()
    print('上传成功')
sk.close()

解析：约定服务器与客户端的上传格式为第一次发送文件大小信息，二次发送文件

2.1 path=os.path.join(BASE_DIR,'Up',filename) 用路径拼接函数将本地路径BASE_DIR、文件夹Up、和文件名称拼接为服务器本地绝对路径

2.2  file_size=os.stat(path).st_size 利用次函数获取文件的大小

2.3  打开文件发送数据以bytes类型读取并发送rb模式

注意：当操作完文件后，需要将其 关闭，f.close()否则无法再文件打开的模式下再次打开查看文件内容