1. 程式人生 > >用python實現的簡單Server/Client檔案傳輸

用python實現的簡單Server/Client檔案傳輸

還是那個題目(題目和流程見java版本),感覺光用java寫一點新意也沒有,恰巧剛學習了python,何不拿來一用,呵呵:

伺服器端:

python 程式碼
  1. importSocketServertime
  2. class MyServer(SocketServer.BaseRequestHandler):   
  3.     userInfo = {   
  4.         'yangsq'    : 'yangsq',   
  5.         'hudeyong'  : 'hudeyong',   
  6.         'mudan'     : 'mudan' }   
  7. def handle(self):   
  8. print 'Connected from', self.client_address   
  9. whileTrue:   
  10.             receivedData = self.request.recv(8192)   
  11. ifnot receivedData:   
  12. continue
  13. elif receivedData == 'Hi, server':   
  14. self.request.sendall('hi, client')   
  15. elif receivedData.startswith('name'):   
  16. self.clientName = receivedData.split(':')[-1]   
  17. if MyServer.userInfo.has_key(self.clientName):   
  18. self.request.sendall('valid')   
  19. else:   
  20. self.request.sendall('invalid')   
  21. elif receivedData.startswith('pwd'):   
  22. self.clientPwd = receivedData.split(':')[-1]   
  23. ifself.clientPwd == MyServer.userInfo[self.clientName]:   
  24. self.request.sendall('valid')   
  25. time.sleep(5)   
  26.                     sfile = open('PyNet.pdf', 'rb')   
  27. whileTrue:   
  28.                         data = sfile.read(1024)   
  29. ifnot data:   
  30. break
  31. whilelen(data) > 0:   
  32.                             intSent = self.request.send(data)   
  33.                             data = data[intSent:]   
  34. time.sleep(3)   
  35. self.request.sendall('EOF')   
  36. else:   
  37. self.request.sendall('invalid')   
  38. elif receivedData == 'bye':   
  39. break
  40. self.request.close()   
  41. print 'Disconnected from', self.client_address   
  42. print
  43. if__name__ == '__main__':   
  44. print 'Server is started\nwaiting for connection...\n'   
  45.     srv = SocketServer.ThreadingTCPServer(('localhost', 50000), MyServer)   
  46.     srv.serve_forever()   

說明:

  • line-55到line-58的作用就相當於java中某個類裡面的main函式,即一個類的入口。
  • python中SocketServer module裡提供了好多實用的現成的類,BaseRequestHandler就是一個,它的作用是為每一個請求fork一個執行緒,只要繼承它,就有這個能力了,哈哈,真是美事。
  • 當然,我們繼承了BaseRequestHandler,就是override它的handle方法,就像java中繼承了Thread後要實現run方法一樣。實際上這個handle方法的內容和我們的java版本的run函式實現的完全一樣。
  • line-30到line-43就是處理檔案下載的主要內容了。看著都挺眼熟的呵:)
  • 這裡在檔案傳送完後發了一個“EOF”,告訴client檔案傳完了。

客戶端:

python 程式碼
  1. importsockettime
  2. class MyClient:   
  3. def__init__(self):   
  4. print 'Prepare for connecting...'   
  5. def connect(self):   
  6.         sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)   
  7.         sock.connect(('localhost', 50000))   
  8.         sock.sendall('Hi, server')   
  9. self.response = sock.recv(8192)   
  10. print 'Server:', self.response   
  11. self.s = raw_input("Server: Do you want get the 'thinking in python' file?(y/n):")   
  12. ifself.s == 'y':   
  13. whileTrue:   
  14. self.name = raw_input('Server: input our name:')   
  15.                 sock.sendall('name:' + self.name.strip())   
  16. self.response = sock.recv(8192)   
  17. ifself.response == 'valid':   
  18. break
  19. else:   
  20. print 'Server: Invalid username'   
  21. whileTrue:   
  22. self.pwd = raw_input('Server: input our password:')   
  23.                 sock.sendall('pwd:' + self.pwd.strip())   
  24. self.response = sock.recv(8192)   
  25. ifself.response == 'valid':   
  26. print 'please wait...'   
  27.                     f = open('b.pdf', 'wb')   
  28. whileTrue:   
  29.                         data = sock.recv(1024)   
  30. if data == 'EOF':   
  31. break
  32.                         f.write(data)   
  33.                     f.flush()   
  34.                     f.close()   
  35. print 'download finished'   
  36. break
  37. else:   
  38. print 'Server: Invalid password'   
  39.         sock.sendall('bye')   
  40.         sock.close()   
  41. print 'Disconnected'   
  42. if__name__ == '__main__':   
  43.     client = MyClient()   
  44.     client.connect()   

line-34到line-41處理檔案下載,client收到server的“EOF”訊號後,就知道檔案傳完了。

最後需要說明一下python的檔案,由於是內建型別,所以不想java那樣有那麼多的reader,writer,input,ouput啊。python中,在開啟或建立一個檔案時,主要是通過模式(mode)來區別的。

python的網路程式設計確實簡單,因為它提供了各種功能的已經寫好的類,直接繼承就Ok了。

python還在學習中,上面的例子跑通是沒問題,但寫得肯定不夠好,還得學習啊