在使用ftplib的时候,踩了很多坑,耗费了自己很多的时间,所以自己根据网上的博客和自己踩得坑,写了一个类,以后可以自己使用:
#coding:utf-8
import os
import sys
import ftplib
import configparser
import datetime
import chardet
class FTPSync(object):
conn = ftplib.FTP()
def __init__(self, host, port):
self.conn.connect(host, port)
def login(self, username, password):
'''
登录操作
:param username:用户名
:param password:密码
:return:
'''
self.conn.login(username, password)
self.conn.set_pasv(False)
def getFile(self,filename ,download_path):
'''
下载单个文件,这里下载的文件名和指定的文件名一样
:param filename:文件名
:param download_path:下载的路径
:return:
'''
if not os.path.isdir(download_path):
os.makedirs(download_path)
bufsize = 1024
fp = open(download_path + '/' + filename, 'wb+').write
self.conn.retrbinary('RETR %s' %filename, fp, bufsize)
def getFiles(self,ftp_path,download_path):
'''
下载多个文件
:param ftp_path:ftp文件下载的路径
:param download_path:本地文件下载的路径
:return:
'''
ftp_path = ftp_path.rstrip('/')
#切换文件的路径
self.conn.cwd(ftp_path)
li = self.conn.nlst()
for eachFile in li:
self.getFile(eachFile, download_path)
self.conn.close()
def create_path(self, path):
'''
创建dir路径,如果路径存在,不创建,如果存在,则进行创建
:param path:需要创建或者切换的路径
:return:
'''
#先切换到远程的根目录
self.conn.cwd('~')
dirs = path.rstrip('/')
base_dir, dirs = self.conn.pwd(), dirs.split('/')
for dir in dirs[1:]:
try:
self.conn.cwd(dir)
except Exception as e:
self.conn.mkd(dir)
#创建好路径之后切换到指定的路径下面,为下载做准备
self.conn.cwd(dir)
def putFile(self,ftp_path, uploadFile):
'''
进行单个文件的上传操作
:param ftp_path:远程ftp的路径
:param uploadFile:上传的本地路径
:return:
'''
#判断远程的目录是否存在,不存在则创建,且切换到指定的路径下
self.create_path(ftp_path)
print(self.conn.pwd())
#本地文件名和上传的文件名保持一致
local_file = uploadFile.split('/')[-1]
fp = open(uploadFile, "rb")
buf_size = 1024
self.conn.storbinary('STOR {}'.format(local_file), fp, buf_size)
fp.close()
该代码包括多个文件的下载,先切换到指定的文件夹下,下载这个文件夹下所有的文件
同时,也支持单个文件的下载,单个文件的上传