python - 如何将文件从客户端发送到服务器并以相同的文件名保存？

Question

我正在尝试使用 python 将文件从客户端发送到服务器。它正在发送，没有任何问题，但我想用相同的文件名保存收到的文件。我不知道如何使用与从客户端发送到服务器相同的文件名保存该文件。我为此编写的代码是:

客户端代码

import socket, os, shutil
from stat import ST_SIZE
HOST=raw_input("Please enter IP-address :  ")

PORT=int(raw_input("Please enter PORT Number : "))
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((HOST,PORT))
if s.recv(8)!='READY':
    raw_input('Unable to connect \n\n Press any key to exit ...')
    s.close()
    exit()
path=raw_input("Please enter the complete PATH of your file :  ")

f=open(path,'rb')
fsize=os.stat(f.name)[ST_SIZE]


s.sendall(str(fsize).zfill(8))
sfile = s.makefile("wb")
shutil.copyfileobj(f, sfile)
sfile.close()
s.close()
f.close()

服务器代码

import socket
import shutil
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
HOST = ''
PORT = 23240
s.bind((HOST, PORT))
s.listen(3)
conn, addr = s.accept()               
print 'conn at address',addr
conn.sendall('READY')
i=1
f = open(r'file_'+ str(i)+".txt",'wb')
i=i+1

print 'File size',fsize
sfile = conn.makefile("rb")
shutil.copyfileobj(sfile, f)
sfile.close()

f.write(conn.recv(fsize))           
f.close()
conn.close()
s.close()

Answer 1

您的代码不是很健壮。 recv(cnt) 最多可传送 cnt 字节或更少的数据。所以不确定，你读了整个文件。甚至不确定，您是否在一次接收中获得了“READY”。相反，你必须使用类似的东西:

def recv_all(sock, bufsize):
    result = ''
    while bufsize>0:
        data = sock.recv(min(bufsize, 4096))
        if not data:
            raise IOError("Socket closed")
        result += data
        bufsize -= len(data)
    return result

如果你想知道服务器上的文件名，你也必须将它传输到服务器。顺便说一句，“READY”有 5 个字符，而不是 8 个。