用python實現的簡單Server/Client檔案傳輸
阿新 • • 發佈:2019-01-22
還是那個題目(題目和流程見java版本),感覺光用java寫一點新意也沒有,恰巧剛學習了python,何不拿來一用,呵呵:
伺服器端:
python 程式碼- importSocketServer, time
- class MyServer(SocketServer.BaseRequestHandler):
- userInfo = {
- 'yangsq' : 'yangsq',
- 'hudeyong' : 'hudeyong',
- 'mudan' : 'mudan' }
- def handle(self):
- print 'Connected from', self.client_address
- whileTrue:
- receivedData = self.request.recv(8192)
- ifnot receivedData:
- continue
- elif receivedData == 'Hi, server':
- self.request.sendall('hi, client')
- elif receivedData.startswith('name'):
- self.clientName = receivedData.split(':')[-1]
- if MyServer.userInfo.has_key(self.clientName):
- self.request.sendall('valid')
- else:
- self.request.sendall('invalid')
- elif receivedData.startswith('pwd'):
- self.clientPwd = receivedData.split(':')[-1]
- ifself.clientPwd == MyServer.userInfo[self.clientName]:
- self.request.sendall('valid')
- time.sleep(5)
- sfile = open('PyNet.pdf', 'rb')
- whileTrue:
- data = sfile.read(1024)
- ifnot data:
- break
- whilelen(data) > 0:
- intSent = self.request.send(data)
- data = data[intSent:]
- time.sleep(3)
- self.request.sendall('EOF')
- else:
- self.request.sendall('invalid')
- elif receivedData == 'bye':
- break
- self.request.close()
- print 'Disconnected from', self.client_address
- if__name__ == '__main__':
- print 'Server is started\nwaiting for connection...\n'
- srv = SocketServer.ThreadingTCPServer(('localhost', 50000), MyServer)
- 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 程式碼- importsocket, time
- class MyClient:
- def__init__(self):
- print 'Prepare for connecting...'
- def connect(self):
- sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- sock.connect(('localhost', 50000))
- sock.sendall('Hi, server')
- self.response = sock.recv(8192)
- print 'Server:', self.response
- self.s = raw_input("Server: Do you want get the 'thinking in python' file?(y/n):")
- ifself.s == 'y':
- whileTrue:
- self.name = raw_input('Server: input our name:')
- sock.sendall('name:' + self.name.strip())
- self.response = sock.recv(8192)
- ifself.response == 'valid':
- break
- else:
- print 'Server: Invalid username'
- whileTrue:
- self.pwd = raw_input('Server: input our password:')
- sock.sendall('pwd:' + self.pwd.strip())
- self.response = sock.recv(8192)
- ifself.response == 'valid':
- print 'please wait...'
- f = open('b.pdf', 'wb')
- whileTrue:
- data = sock.recv(1024)
- if data == 'EOF':
- break
- f.write(data)
- f.flush()
- f.close()
- print 'download finished'
- break
- else:
- print 'Server: Invalid password'
- sock.sendall('bye')
- sock.close()
- print 'Disconnected'
- if__name__ == '__main__':
- client = MyClient()
- client.connect()
line-34到line-41處理檔案下載,client收到server的“EOF”訊號後,就知道檔案傳完了。
最後需要說明一下python的檔案,由於是內建型別,所以不想java那樣有那麼多的reader,writer,input,ouput啊。python中,在開啟或建立一個檔案時,主要是通過模式(mode)來區別的。
python的網路程式設計確實簡單,因為它提供了各種功能的已經寫好的類,直接繼承就Ok了。
python還在學習中,上面的例子跑通是沒問題,但寫得肯定不夠好,還得學習啊