Python_实现ftp上传和下载

1.  #!/usr/bin/python
2.  #coding=gbk
3.  '''''''
4.      ftp自动下载、自动上传脚本，可以递归目录操作
5.  '''
6.
7.  from ftplib import FTP
8.  import os,sys,string,datetime,time
9.  import socket
10.
11. class MYFTP:
12.     def __init__(self, hostaddr, username, password, remotedir, port=21):
13.         self.hostaddr = hostaddr
14.         self.username = username
15.         self.password = password
16.         self.remotedir  = remotedir
17.         self.port     = port
18.         self.ftp      = FTP()
19.         self.file_list = []
20.         # self.ftp.set_debuglevel(2)
21.     def __del__(self):
22.         self.ftp.close()
23.         # self.ftp.set_debuglevel(0)
24.     def login(self):
25.         ftp = self.ftp
26.         try:
27.             timeout = 60
28.             socket.setdefaulttimeout(timeout)
29.             ftp.set_pasv(True)
30.             print '开始连接到 %s' %(self.hostaddr)
31.             ftp.connect(self.hostaddr, self.port)
32.             print '成功连接到 %s' %(self.hostaddr)
33.             print '开始登录到 %s' %(self.hostaddr)
34.             ftp.login(self.username, self.password)
35.             print '成功登录到 %s' %(self.hostaddr)
36.             debug_print(ftp.getwelcome())
37.         except Exception:
38.             deal_error("连接或登录失败")
39.         try:
40.             ftp.cwd(self.remotedir)
41.         except(Exception):
42.             deal_error('切换目录失败')
43.
44.     def is_same_size(self, localfile, remotefile):
45.         try:
46.             remotefile_size = self.ftp.size(remotefile)
47.         except:
48.             remotefile_size = -1
49.         try:
50.             localfile_size = os.path.getsize(localfile)
51.         except:
52.             localfile_size = -1
53.         debug_print('lo:%d  re:%d' %(localfile_size, remotefile_size),)
54.         if remotefile_size == localfile_size:
55.             return 1
56.         else:
57.             return 0
58.     def download_file(self, localfile, remotefile):
59.         if self.is_same_size(localfile, remotefile):
60.             debug_print('%s 文件大小相同，无需下载' %localfile)
61.             return
62.         else:
63.             debug_print('>>>>>>>>>>>>下载文件 %s ... ...' %localfile)
64.         #return
65.         file_handler = open(localfile, 'wb')
66.         self.ftp.retrbinary('RETR %s'%(remotefile), file_handler.write)
67.         file_handler.close()
68.
69.     def download_files(self, localdir='./', remotedir='./'):
70.         try:
71.             self.ftp.cwd(remotedir)
72.         except:
73.             debug_print('目录%s不存在，继续...' %remotedir)
74.             return
75.         if not os.path.isdir(localdir):
76.             os.makedirs(localdir)
77.         debug_print('切换至目录 %s' %self.ftp.pwd())
78.         self.file_list = []
79.         self.ftp.dir(self.get_file_list)
80.         remotenames = self.file_list
81.         #print(remotenames)
82.         #return
83.         for item in remotenames:
84.             filetype = item[0]
85.             filename = item[1]
86.             local = os.path.join(localdir, filename)
87.             if filetype == 'd':
88.                 self.download_files(local, filename)
89.             elif filetype == '-':
90.                 self.download_file(local, filename)
91.         self.ftp.cwd('..')
92.         debug_print('返回上层目录 %s' %self.ftp.pwd())
93.     def upload_file(self, localfile, remotefile):
94.         if not os.path.isfile(localfile):
95.             return
96.         if self.is_same_size(localfile, remotefile):
97.             debug_print('跳过[相等]: %s' %localfile)
98.             return
99.         file_handler = open(localfile, 'rb')
100.            self.ftp.storbinary('STOR %s' %remotefile, file_handler)
101.            file_handler.close()
102.            debug_print('已传送: %s' %localfile)
103.        def upload_files(self, localdir='./', remotedir = './'):
104.            if not os.path.isdir(localdir):
105.                return
106.            localnames = os.listdir(localdir)
107.            self.ftp.cwd(remotedir)
108.            for item in localnames:
109.                src = os.path.join(localdir, item)
110.                if os.path.isdir(src):
111.                    try:
112.                        self.ftp.mkd(item)
113.                    except:
114.                        debug_print('目录已存在 %s' %item)
115.                    self.upload_files(src, item)
116.                else:
117.                    self.upload_file(src, item)
118.            self.ftp.cwd('..')
119.
120.        def get_file_list(self, line):
121.            ret_arr = []
122.            file_arr = self.get_filename(line)
123.            if file_arr[1] not in ['.', '..']:
124.                self.file_list.append(file_arr)
125.
126.        def get_filename(self, line):
127.            pos = line.rfind(':')
128.            while(line[pos] != ' '):
129.                pos += 1
130.            while(line[pos] == ' '):
131.                pos += 1
132.            file_arr = [line[0], line[pos:]]
133.            return file_arr
134.    def debug_print(s):
135.        print (s)
136.    def deal_error(e):
137.        timenow  = time.localtime()
138.        datenow  = time.strftime('%Y-%m-%d', timenow)
139.        logstr = '%s 发生错误: %s' %(datenow, e)
140.        debug_print(logstr)
141.        file.write(logstr)
142.        sys.exit()
143.
144.    if __name__ == '__main__':
145.        file = open("log.txt", "a")
146.        timenow  = time.localtime()
147.        datenow  = time.strftime('%Y-%m-%d', timenow)
148.        logstr = datenow
149.        # 配置如下变量
150.        hostaddr = 'localhost' # ftp地址
151.        username = 'test' # 用户名
152.        password = 'test' # 密码
153.        port  =  21   # 端口号
154.        rootdir_local  = '.' + os.sep + 'bak/' # 本地目录
155.        rootdir_remote = './'          # 远程目录
156.
157.        f = MYFTP(hostaddr, username, password, rootdir_remote, port)
158.        f.login()
159.        f.download_files(rootdir_local, rootdir_remote)
160.
161.        timenow  = time.localtime()
162.        datenow  = time.strftime('%Y-%m-%d', timenow)
163.        logstr += " - %s 成功执行了备份\n" %datenow
164.        debug_print(logstr)
165.
166.        file.write(logstr)
167.        file.close()