如何将文件从客户端发送到服务器并使用相同的文件名保存?

时间:2014-04-21 09:22:04

标签: python sockets networking client

我试图在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()

1 个答案:

答案 0 :(得分:1)

您的代码不是很强大。 recv(cnt)提供高达cnt字节的数据或更少。所以不确定,你读完整个文件。甚至不确定,你在一次recv中获得“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个字符。