Python 下载FTP所有昨天生成的文件（递归）

点击(此处)折叠或打开

1. #!/usr/bin/env python
   
2. # -*- encoding: utf8 -*-
   
3. # Mr.Wang
   
4. 
   
5. import os
   
6. import sys
   
7. import ftplib
   
8. import time
   
9. 
   
10. mytime = time.strftime('%Y%m%d',time.localtime(time.time() - 24*60*60))
    
11. 
    
12. class FTPSync(object):
    
13.     def __init__(self):
    
14.         self.conn = ftplib.FTP('XXX.XXX.XXX.XXX', 'user', 'password')
    
15.         self.conn.cwd('/') # 远端FTP目录
    
16.         os.chdir('/data/ftp/') # 本地下载目录
    
17.     def get_dirs_files(self):
    
18.         u''' 得到当前目录和文件, 放入dir_res列表 '''
    
19.         dir_res = []
    
20.         self.conn.dir('.', dir_res.append)
    
21.         files = [f.split(None, 8)[-1] for f in dir_res if f.startswith('-')]
    
22.         dirs = [f.split(None, 8)[-1] for f in dir_res if f.startswith('d')]
    
23.         return (files, dirs)
    
24.     def walk(self, next_dir):
    
25.         print 'Walking to', next_dir
    
26.         self.conn.cwd(next_dir)
    
27.         try:
    
28.             os.mkdir(next_dir)
    
29.         except OSError:
    
30.             pass
    
31.         os.chdir(next_dir)
    
32.         ftp_curr_dir = self.conn.pwd()
    
33.         local_curr_dir = os.getcwd()
    
34.         files, dirs = self.get_dirs_files()
    
35.         print "FILES: ", files
    
36.         print "DIRS: ", dirs
    
37.         for f in files:
    
38.                         if mytime in f:
    
39.                                 print next_dir, ':', f
    
40.                                 outf = open(f, 'wb')
    
41.                                 try:
    
42.                                         self.conn.retrbinary('RETR %s' % f, outf.write)
    
43.                                 finally:
    
44.                                         outf.close()
    
45.         for d in dirs:
    
46.             os.chdir(local_curr_dir)
    
47.             self.conn.cwd(ftp_curr_dir)
    
48.             self.walk(d)
    
49.     def run(self):
    
50.         self.walk('.')
    
51. def main():
    
52.     f = FTPSync()
    
53.     f.run()
    
54. if __name__ == '__main__':
    
55.     main()