的Python在TCP多个文件传输插座

Question

我试图用Python语言编写的程序可以在文件夹中通过套接字传输多个文件，我有以下的代码到目前为止

客户：

def uploadFiles(folder,dcip,PORT,filetype): 
    os.chdir(folder) 
    dirList = os.listdir(folder) 
    print dirList 
    ms = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    print PORT 
    ms.connect((dcip, int(PORT))) 
    for fname in dirList: 

     if fname.endswith(str(filetype)): 
      cmd = 'get\n%s\n' % (fname) 
      ms.sendall(cmd) 
      f = open(fname,'rb') 
      data = f.read() 
      f.close() 
      print data 
      r = ms.recv(2) 
      ms.sendall(data) 
      ms.sendall('done\n%s\n' %(fname)) 
    ms.sendall('end\n\n') 
    ms.close() 

服务器：

import socket,os 
listener_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
listener_socket.bind(('',33234)) 
filename = '' 
while 1: 
    listener_socket.listen(100) 
    connection,address_client = listener_socket.accept() 
    if not os.path.exists(str(address_client[0])): 
     os.makedirs(str(address_client[0])) 
    currdir = os.getcwd() 
    os.chdir('./'+str(address_client[0])) 

    while(1): 
     data = connection.recv(4096) 
     cmd = data[:data.find('\n')] 

     if cmd == 'get': 
      x,filename,x = data.split('\n',2) 
      connection.sendall('ok') 
      f = open(filename,'wb') 


     if cmd == 'done': 
      continue 

     f.write(data) 

     if cmd == 'end': 
      connection.close() 
      break 
    os.chdir(currdir) 

上面的代码进入一个无限循环，这点我明白，这是因为在if cmd == 'done'部分continue语句的，我不知道为什么它是做日在？我的意思是它从来没有从客户端得到完成的消息，任何人都可以请帮我修复代码？

Answer 1

因为命令和文件名是分开的行，所以最好通过行解析接收到的数据。块数据的最后部分不必以\n完成，所以它必须与下一个接收到的数据块合并。

这是（未测试）实施解析的由线接收的数据：

data = '' # contains last line of a read block if it didn't finish with \n 
in_get, in_done, reading_file, ended = False, False, False, False 
while not ended: 
    if len(data) > 100: # < update 
    f.write(data) # < 
    data = ''   # < 
    data += connection.recv(4096) 
    i = data.find('\n') 
    while i >= 0 and not ended: 
    line = data[:i] 
    data = data[i+1:] 
    if in_get: 
     filename = line 
     reading_file = True 
     f = open(filename,'wb') 
     in_get = False 
    elif in_done: 
     if line != filename: # done inside file content 
     f.write('done\n' + line + '\n') 
     else: 
     f.close() 
     reading_file = False 
     in_done = False 
    else: 
     if line == 'get' and not reading_file: 
     in_get = True 
     elif line == 'done': 
     in_done = True 
     elif line == 'end' and not reading_file: 
     ended = True 
     break; 
     else: 
     f.write(line + '\n') 
    i = data.find('\n')