的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')