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 #文件名命名的時候不要帶有空格,否則文件會傳輸不上去