通過使用 ftplib庫,批量上傳,下載文件


Python版本為:Python3.5

  1 from ftplib import FTP,error_perm  # 加載ftp模塊
  2 import os
  3 
  4 class FTPSync(object):
  5     conn = FTP()
  6 
  7     def __init__(self,host,port=21):
  8         self.conn.connect(host,port)
  9 
 10     def login(self,username,password):
 11         self.conn.login(username,password)
 12         self.conn.encoding = "GB2312"
 13         self.conn.set_debuglevel(2)             #打開調試級別2,顯示詳細信息
 14         self.conn.set_pasv(True)
 15         #0主動模式 1 #被動模式
 16         print(self.conn.welcome)
 17 
 18     def _ftp_list(self,line):
 19         li = line.split(' ')
 20         if self.ftp_dir_name == li[-1] and "<DIR>" in li:
 21             self._is_dir = True
 22 
 23     def _is_ftp_file(self, ftp_path):
 24         try:
 25             if ftp_path in self.conn.nlst(os.path.dirname(ftp_path)):
 26                 return True
 27             else:
 28                 return False
 29         except error_perm as e:
 30             return False
 31 
 32     def _is_ftp_dir(self,ftp_path):
 33         """
 34         用來判斷所給的路徑是文件還是文件夾
 35         """
 36         ftp_path = ftp_path.rstrip('/')
 37         ftp_parent_path = os.path.dirname(ftp_path)
 38         self.ftp_dir_name = os.path.basename(ftp_path)
 39         self._is_dir = False
 40         if ftp_path == '.' or ftp_path == './' or ftp_path == '':
 41             self._is_dir = True
 42         else:
 43             try:
 44                 self.conn.retrlines('LIST %s' %ftp_parent_path,self._ftp_list)
 45             except error_perm as e:
 46                 return self._is_dir
 47         return self._is_dir
 48 
 49 
 50     def get_file(self,ftp_path="",local_path="."):
 51         print(ftp_path)
 52         ftp_path = ftp_path.rstrip('/')
 53 
 54         file_name = os.path.basename(ftp_path)
 55 
 56         # 如果本地路徑是目錄,下載文件到該目錄
 57         if os.path.isdir(local_path):
 58             file_handler = open(os.path.join(local_path,file_name),'wb')
 59             self.conn.retrbinary("RETR %s"%(ftp_path),file_handler.write)
 60             file_handler.close()
 61 
 62         # 如果本地路徑不是目錄,但上層目錄存在,則按照本地路徑的文件名作為下載的文件名稱
 63         elif os.path.isdir(os.path.dirname(local_path)):
 64             file_handler = open(local_path,'wb')
 65             self.conn.retrbinary("RETR %s"%(ftp_path),file_handler.write)
 66             file_handler.close()
 67         # 如果本地路徑不是目錄,且上層目錄不存在,則退出
 68         else:
 69             print('EROOR:The dir:%s is not exist' %os.path.dirname(local_path))
 70 
 71 
 72     def get_dir(self,ftp_path,local_path=".",begin=True):
 73 
 74         if not self._is_ftp_dir(ftp_path):
 75             self.get_file(ftp_path=ftp_path, local_path=local_path)
 76             return
 77 
 78         if begin:
 79             local_path = os.path.join(local_path, os.path.basename(ftp_path))
 80 
 81         #如果本地目錄不存在,則創建目錄
 82         if not os.path.isdir(local_path):
 83             os.mkdir(local_path)
 84 
 85         #進入ftp目錄,開始遞歸查詢
 86         self.conn.cwd(ftp_path)
 87 
 88         ftp_files = self.conn.nlst()
 89 
 90         for file in ftp_files:
 91             local_file = os.path.join(local_path, file)
 92             #如果file ftp路徑是目錄則遞歸上傳目錄(不需要再進行初始化begin的標志修改為False)
 93             #如果file ftp路徑是文件則直接上傳文件
 94             if self._is_ftp_dir(file):
 95                 self.get_dir(file, local_file, False)
 96             elif "idea" in file:
 97                 pass
 98             else:
 99                 self.get_file(ftp_path=file, local_path=local_file)
100 
101         #如果當前ftp目錄文件已經遍歷完畢返回上一層目錄
102         self.conn.cwd("..")
103 
104 
105     def get_all_dir(self):
106         ftp_files = self.conn.nlst()
107         for file in ftp_files:
108             self.get_dir(file,"D:\\ftp",True)
109 
110 
111     def put_file(self,local_path,ftp_path="."):
112         ftp_path = ftp_path.rstrip('/')
113         if os.path.isfile(local_path):
114             file_handler = open(local_path,'rb')
115             local_file_name = os.path.basename(local_path)
116 
117             #如果遠程路徑是個目錄,則上傳文件到這個目錄,文件名不變
118             if self._is_ftp_dir(ftp_path):
119                 self.conn.storbinary('STOR %s'%os.path.join(ftp_path,local_file_name), file_handler)
120 
121             #如果遠程路徑的上層是個目錄,則上傳文件,文件名按照給定命名
122             elif self._is_ftp_dir(os.path.dirname(ftp_path)):
123                 print('STOP %s'%ftp_path)
124                 self.conn.storbinary('STOR %s'%ftp_path, file_handler)
125             #如果遠程路徑不是目錄,且上一層的目錄也不存在,則提示給定遠程路徑錯誤
126             else:
127                 print('STOR %s'%ftp_path, file_handler)
128 
129     def put_dir(self,local_path,ftp_path=".",begin=True):
130         ftp_path = ftp_path.rstrip('/')
131 
132         if not os.path.isdir(local_path):
133             print('ERROR:The dir:%s is not exist' %local_path)
134             return
135 
136         #當本地目錄存在時上傳
137         #上傳初始化:如果給定的ftp路徑不存在需要創建,同時將本地的目錄存放在給定的ftp目錄下。
138         # 本地目錄下文件存放的路徑為ftp_path = ftp_path + os.path.basename(local_path)
139         #例如,將本地的文件夾a上傳到ftp的a/b目錄下,則本地a目錄下的文件將上傳的ftp的a/b/a目錄下
140         if begin:
141             if not self._is_ftp_dir(ftp_path):
142                 self.conn.mkd(ftp_path)
143             ftp_path = os.path.join(ftp_path,os.path.basename(local_path))
144 
145         # 如果上傳路徑是文件夾,則創建目錄
146         if not self._is_ftp_dir(ftp_path):
147             self.conn.mkd(ftp_path)
148 
149 
150         #進入本地目錄,開始遞歸查詢
151         os.chdir(local_path)
152         local_files = os.listdir('.')
153         for file in local_files:
154             ftp_file = os.path.join(ftp_path,file)
155             #如果file本地路徑是目錄則遞歸上傳文件(不需要再進行初始化begin的標志修改為False)
156             #如果file本地路徑是文件則直接上傳文件
157             if os.path.isdir(file):
158                 self.put_dir(file,ftp_file,False)
159             elif "idea" in file:
160                 pass
161             else:
162                 self.put_file(file,ftp_file)
163 
164         #如果當前本地目錄文件已經遍歷完畢返回上一層目錄
165         os.chdir('..')
166 
167 
168 if __name__ == "__main__":
169     ftp = FTPSync('FTPIP')
170     ftp.login('username',"password")
171     # ftp.get_file("ftppath")
172     # ftp.get_dir("ftppath","localpath",True)
173     # ftp.get_dir("","",True)
174     # ftp.get_all_dir("")
175     # ftp.put_file('localpath')
176     ftp.put_dir('localpath',begin=True)
177 
178 
179 #文件名命名的時候不要帶有空格,否則文件會傳輸不上去
View Code

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM