通过python socket实现FTP的部分功能

直接上代码吧
client端：

点击(此处)折叠或打开

1. #!/usr/bin/env python
   
2. #!/usr/bin/env python
   
3. #-*- coding: utf-8 -*-
   
4. import socket,os
   
5. import tab,re
   
6. 
   
7. HOST='172.15.111.80'
   
8. PORT=9999
   
9. s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
   
10. Match=re.compile('get|upload')
    
11. try:
    
12.     s.connect((HOST,PORT))
    
13.     print ''' \033[32;1m 请选择你要操作的命令：
    
14.         list: 显示下载列表
    
15.         get xxx: 下载某文件
    
16.         upload xxx: 上传某文件
    
17.         \033[0m'''
    
18. 
    
19.     while True:
    
20. 
    
21.         cmd=raw_input('>>>: ').strip()
    
22.         s.sendall(cmd)
    
23.         Input = cmd.split()
    
24.         if len(Input) == 1 and Input[0] == 'list':            ###目录文件展示
    
25.             s.sendall('ls -al')
    
26.             print '\033[32;1m ftp文件服务器列表：\033[0m'
    
27.             List = s.recv(4096)
    
28.             print List
    
29. 
    
30.         elif len(Input) == 2 and Match.match(cmd):            ###下载或者上传功能实现，目前只做了下载，上传功能类似
    
31.             file_total = s.recv(1024)
    
32.             receive_size = 0
    
33.             if 'No exist' in file_total:                      ###通过对服务端发送的类型进行判断，如果发送的是no exist就打印错误，否则的话接收到的应该是文件字符串大小
    
34.                 print file_total
    
35.                 continue
    
36.             else:
    
37.                 while receive_size < int(file_total):         ###一行行接收
    
38.                     f = open(Input[1],'wb')
    
39.                     data = s.recv(1024)
    
40.                     receive_size += len(data)
    
41.                     f.write(data)
    
42.                     f.close()
    
43.                 if receive_size == int(file_total):
    
44.                     print '传输完毕！'
    
45. 
    
46.         else:
    
47.             print '\033[31;1m 请给出两个正确的参数，例如：get abc.txt 或者upload abc.txt\033[0m'
    
48.             continue
    
49. finally:
    
50.     s.close()

server端：

点击(此处)折叠或打开

1. #!/usr/bin/env python
   
2. import socket,os,tab
   
3. import commands
   
4. 
   
5. HOST='172.15.111.80'
   
6. PORT=9999
   
7. s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
   
8. s.bind((HOST,PORT))
   
9. s.listen(5)
   
10. while True:
    
11.     conn,addr=s.accept()
    
12.     print 'Connect address: ' , addr
    
13.     list = os.listdir(os.getcwd())
    
14.     while True:
    
15.         data = conn.recv(1024).strip()
    
16.         cwd = data.split()
    
17.         if cwd[0] == 'list':                                ###对传递的第一个参数进行判断，如果为list，就传回目录下的文件列表
    
18.             Command = conn.recv(1024).strip()
    
19.             data = commands.getstatusoutput(Command)
    
20.             conn.sendall(data[1])
    
21.         elif cwd[0] == 'get':                                ###如果第一个参数是get，再继续判断第二个参数，如果存在，就返回字符串大小，并且一行行传回字符串，client端一行行接收；
22.             if cwd[1] in list:
    
23.                 f = open(cwd[1],'rb')
    
24.                 file_size = str(os.stat(cwd[1]).st_size)
    
25.                 conn.sendall(file_size)
    
26.                 for line in f:
    
27.                     conn.sendall(line),
    
28.             else:                                            ###如果第二个参数不在list中，那就返回不存在
    
29.                 conn.sendall('No exist this file')
    
30. 
    
31.         elif len(data) == 0:                                  ###如果输入为空，重新接收，不然就卡在那不动了
    
32.             continue
    
33. conn.close()

还有一个上传的功能没做，跟下载差不多，有一些优化也没做，例如：
1、多线程传输没做，功能点SocketServer，（data = SocketServer.ThreadingTCPServer((ip,port),类);   data.serve_forever()强化记忆一下)
2、其实这个可以用函数包起来，也可以调整一下
3、我看有人用md5校验，这个功能也不错，遇到网络丢包问题，可以使用