还是那个题目(题目和流程见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还在学习中,上面的例子跑通是没问题,但写得肯定不够好,还得学习啊