还是那个题目(题目和流程见java版本),感觉光用java写一点新意也没有,恰巧刚学习了python,何不拿来一用,呵呵:
服务器端:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
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 .clientName]: 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[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 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文件传完了。
客户端:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
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还在学习中,上面的例子跑通是没问题,但写得肯定不够好,还得学习啊