Python(文件上传)

day26

通过socket上传文件。

post_client.py

import socket
import os
sk = socket.socket()
print(sk)
address = ('127.0.0.1', 8003)
sk.connect(address)

BASE_DIR = os.path.dirname(os.path.abspath(__file__))#当前执行文件路径

while True:
    inp = input('>>>').strip()#post abc.png
    cmd, path = inp.split('|') #post, 需传送的文件即abc.png
    path = os.path.join(BASE_DIR, path)#完整的路径拼接
    filename = os.path.basename(path)#文件名字
    file_size = os.stat(path).st_size#查看文件大小

    file_info = 'post|%s|%s'%(filename, file_size)#文件信息
    sk.sendall(bytes(file_info, 'utf8'))

    f = open(path, 'rb')#字节b为bytes
    has_sent = 0
    while has_sent != file_size:#相同大小说明全部上传
        data = f.read(1024)#取一部分发
        sk.sendall(data)
        has_sent += len(data)

    f.close()
    print('上传成功')


sk.close()
#print(sk)

post_server.py

import socket
import subprocess
sk = socket.socket()#创建socket
import os
print(sk)

address = ('127.0.0.1', 8003)#本机地址
sk.bind(address)#为socket绑定IP和端口号

sk.listen(3)#限制排队的个数
print('waiting......')
BASE_DIR = os.path.dirname(os.path.abspath(__file__))

while True:
    conn, addr = sk.accept()

    while True:
        data = conn.recv(1024)
        cmd, filename, filesize = str(data, 'utf8').split('|')#获得信息
        path = os.path.join(BASE_DIR, 'far', filename)#新路径
        filesize = int(filesize)

        f = open(path, 'ab')
        has_receive = 0
        #接收
        while has_receive != filesize:
            data = conn.recv(1024)
            f.write(data)
            has_receive += len(data)

        f.close()#不能忘记关闭，否则出错
        print('接收完毕！')
sk.close()#全关

执行post_server.py,再执行post_client.py,在client中输入post|abc.png进行上传。

（其中abc.png和两个执行文件同级，在以上三个文件相同位置还存在一个far包）