色粉盒撒娇 发表于 2017-5-4 06:58:03

用python实现的简单Server/Client文件传输

  还是那个题目(题目和流程见java版本),感觉光用java写一点新意也没有,恰巧刚学习了python,何不拿来一用,呵呵:
  服务器端:
python 代码

[*]import SocketServer, time  
[*]  
[*]class MyServer(SocketServer.BaseRequestHandler):   
[*]    userInfo = {   
[*]        'yangsq'    : 'yangsq',   
[*]        'hudeyong'  : 'hudeyong',   
[*]        'mudan'     : 'mudan' }   
[*]  
[*]    def handle(self):   
[*]        print 'Connected from', self.client_address   
[*]           
[*]        while True:   
[*]            receivedData = self.request.recv(8192)   
[*]            if not 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]   
[*]                if self.clientPwd == MyServer.userInfo:   
[*]                    self.request.sendall('valid')   
[*]                    time.sleep(5)   
[*]  
[*]                    sfile = open('PyNet.pdf', 'rb')   
[*]                    while True:   
[*]                        data = sfile.read(1024)   
[*]                        if not data:   
[*]                            break  
[*]                        while len(data) > 0:   
[*]                            intSent = self.request.send(data)   
[*]                            data = data   
[*]  
[*]                    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   
[*]        print  
[*]  
[*]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 代码

[*]import socket, 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):")   
[*]        if self.s == 'y':   
[*]            while True:   
[*]                self.name = raw_input('Server: input our name:')   
[*]                sock.sendall('name:' + self.name.strip())   
[*]                self.response = sock.recv(8192)   
[*]                if self.response == 'valid':   
[*]                    break  
[*]                else:   
[*]                    print 'Server: Invalid username'   
[*]  
[*]            while True:   
[*]                self.pwd = raw_input('Server: input our password:')   
[*]                sock.sendall('pwd:' + self.pwd.strip())   
[*]                self.response = sock.recv(8192)   
[*]                if self.response == 'valid':   
[*]                    print 'please wait...'   
[*]  
[*]                    f = open('b.pdf', 'wb')   
[*]                    while True:   
[*]                        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还在学习中,上面的例子跑通是没问题,但写得肯定不够好,还得学习啊
页: [1]
查看完整版本: 用python实现的简单Server/Client文件传输