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包）