python实现支持目录FTP上传下载文件的方法
admin
2023-07-31 02:26:56
0

本文实例讲述了python实现支持目录FTP上传下载文件的方法。分享给大家供大家参考。具体如下:

该程序支持ftp上传下载文件和目录、适用于windows和linux平台。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import ftplib
import os
import sys
class FTPSync(object):
  conn = ftplib.FTP()
  def __init__(self,host,port=21):    
    self.conn.connect(host,port)    
  def login(self,username,password):
    self.conn.login(username,password)
    self.conn.set_pasv(False)
    print self.conn.welcome
  def test(self,ftp_path):
    print ftp_path
    print self._is_ftp_dir(ftp_path)
    #print self.conn.nlst(ftp_path)
    #self.conn.retrlines( \'LIST ./a/b\')
    #ftp_parent_path = os.path.dirname(ftp_path)
    #ftp_dir_name = os.path.basename(ftp_path)
    #print ftp_parent_path
    #print ftp_dir_name
  def _is_ftp_file(self,ftp_path):
    try:
      if ftp_path in self.conn.nlst(os.path.dirname(ftp_path)):
        return True
      else:
        return False
    except ftplib.error_perm,e:
      return False
  def _ftp_list(self, line):
    list = line.split(\' \')
    if self.ftp_dir_name==list[-1] and list[0].startswith(\'d\'):
      self._is_dir = True
  def _is_ftp_dir(self,ftp_path):
    ftp_path = ftp_path.rstrip(\'/\')
    ftp_parent_path = os.path.dirname(ftp_path)
    self.ftp_dir_name = os.path.basename(ftp_path)
    self._is_dir = False
    if ftp_path == \'.\' or ftp_path== \'./\' or ftp_path==\'\':
      self._is_dir = True
    else:
      #this ues callback function ,that will change _is_dir value
      try:
        self.conn.retrlines(\'LIST %s\' %ftp_parent_path,self._ftp_list)
      except ftplib.error_perm,e:
        return self._is_dir    
    return self._is_dir
  def get_file(self,ftp_path,local_path=\'.\'):
    ftp_path = ftp_path.rstrip(\'/\')
    if self._is_ftp_file(ftp_path):    
      file_name = os.path.basename(ftp_path)
      #如果本地路径是目录,下载文件到该目录
      if os.path.isdir(local_path):
        file_handler = open(os.path.join(local_path,file_name), \'wb\' )
        self.conn.retrbinary(\"RETR %s\" %(ftp_path), file_handler.write) 
        file_handler.close()
      #如果本地路径不是目录,但上层目录存在,则按照本地路径的文件名作为下载的文件名称
      elif os.path.isdir(os.path.dirname(local_path)):
        file_handler = open(local_path, \'wb\' )
        self.conn.retrbinary(\"RETR %s\" %(ftp_path), file_handler.write) 
        file_handler.close()
      #如果本地路径不是目录,且上层目录不存在,则退出
      else:
        print \'EROOR:The dir:%s is not exist\' %os.path.dirname(local_path)
    else:
      print \'EROOR:The ftp file:%s is not exist\' %ftp_path
  def put_file(self,local_path,ftp_path=\'.\'):
    ftp_path = ftp_path.rstrip(\'/\')
    if os.path.isfile( local_path ):           
      file_handler = open(local_path, \"r\")
      local_file_name = os.path.basename(local_path)
      #如果远程路径是个目录,则上传文件到这个目录,文件名不变
      if self._is_ftp_dir(ftp_path):
        self.conn.storbinary(\'STOR %s\'%os.path.join(ftp_path,local_file_name), file_handler)
      #如果远程路径的上层是个目录,则上传文件,文件名按照给定命名
      elif self._is_ftp_dir(os.path.dirname(ftp_path)): 
        print \'STOR %s\'%ftp_path        
        self.conn.storbinary(\'STOR %s\'%ftp_path, file_handler)
      #如果远程路径不是目录,且上一层的目录也不存在,则提示给定远程路径错误
      else:        
        print \'EROOR:The ftp path:%s is error\' %ftp_path
      file_handler.close()
    else:
      print \'ERROR:The file:%s is not exist\' %local_path
  def get_dir(self,ftp_path,local_path=\'.\',begin=True): 
    ftp_path = ftp_path.rstrip(\'/\')
    #当ftp目录存在时下载    
    if self._is_ftp_dir(ftp_path):
      #如果下载到本地当前目录下,并创建目录
      #下载初始化:如果给定的本地路径不存在需要创建,同时将ftp的目录存放在给定的本地目录下。
      #ftp目录下文件存放的路径为local_path=local_path+os.path.basename(ftp_path)
      #例如:将ftp文件夹a下载到本地的a/b目录下,则ftp的a目录下的文件将下载到本地的a/b/a目录下
      if begin:
        if not os.path.isdir(local_path):
          os.makedirs(local_path)
        local_path=os.path.join(local_path,os.path.basename(ftp_path))
      #如果本地目录不存在,则创建目录
      if not os.path.isdir(local_path):
        os.makedirs(local_path)
      #进入ftp目录,开始递归查询
      self.conn.cwd(ftp_path)
      ftp_files = self.conn.nlst()
      for file in ftp_files:
        local_file = os.path.join(local_path, file)
        #如果file ftp路径是目录则递归上传目录(不需要再进行初始化begin的标志修改为False)
        #如果file ftp路径是文件则直接上传文件
        if self._is_ftp_dir(file):
          self.get_dir(file,local_file,False)
        else:
          self.get_file(file,local_file)
      #如果当前ftp目录文件已经遍历完毕返回上一层目录
      self.conn.cwd( \"..\" )
      return
    else:
      print \'ERROR:The dir:%s is not exist\' %ftp_path
      return

  def put_dir(self,local_path,ftp_path=\'.\',begin=True):
    ftp_path = ftp_path.rstrip(\'/\')
    #当本地目录存在时上传
    if os.path.isdir(local_path):      
      #上传初始化:如果给定的ftp路径不存在需要创建,同时将本地的目录存放在给定的ftp目录下。
      #本地目录下文件存放的路径为ftp_path=ftp_path+os.path.basename(local_path)
      #例如:将本地文件夹a上传到ftp的a/b目录下,则本地a目录下的文件将上传的ftp的a/b/a目录下
      if begin:        
        if not self._is_ftp_dir(ftp_path):
          self.conn.mkd(ftp_path)
        ftp_path=os.path.join(ftp_path,os.path.basename(local_path))          
      #如果ftp路径不是目录,则创建目录
      if not self._is_ftp_dir(ftp_path):
        self.conn.mkd(ftp_path)

      #进入本地目录,开始递归查询
      os.chdir(local_path)
      local_files = os.listdir(\'.\')
      for file in local_files:
        #如果file本地路径是目录则递归上传目录(不需要再进行初始化begin的标志修改为False)
        #如果file本地路径是文件则直接上传文件
        if os.path.isdir(file):          
          ftp_path=os.path.join(ftp_path,file)
          self.put_dir(file,ftp_path,False)
        else:
          self.put_file(file,ftp_path)
      #如果当前本地目录文件已经遍历完毕返回上一层目录
      os.chdir( \"..\" )
    else:
      print \'ERROR:The dir:%s is not exist\' %local_path
      return
if __name__ == \'__main__\':
  ftp = FTPSync(\'192.168.1.110\')
  ftp.login(\'test\',\'test\')
  #上传文件,不重命名
  #ftp.put_file(\'111.txt\',\'a/b\')
  #上传文件,重命名
  #ftp.put_file(\'111.txt\',\'a/112.txt\')
  #下载文件,不重命名
  #ftp.get_file(\'/a/111.txt\',r\'D:\\\\\')
  #下载文件,重命名
  #ftp.get_file(\'/a/111.txt\',r\'D:\\112.txt\')
  #下载到已经存在的文件夹
  #ftp.get_dir(\'a/b/c\',r\'D:\\\\a\')
  #下载到不存在的文件夹
  #ftp.get_dir(\'a/b/c\',r\'D:\\\\aa\')
  #上传到已经存在的文件夹
  ftp.put_dir(\'b\',\'a\')
  #上传到不存在的文件夹
  ftp.put_dir(\'b\',\'aa/B/\')

希望本文所述对大家的Python程序设计有所帮助。

相关内容

热门资讯

500 行 Python 代码... 语法分析器描述了一个句子的语法结构,用来帮助其他的应用进行推理。自然语言引入了很多意外的歧义,以我们...
定时清理删除C:\Progra... C:\Program Files (x86)下面很多scoped_dir开头的文件夹 写个批处理 定...
65536是2的几次方 计算2... 65536是2的16次方:65536=2⁶ 65536是256的2次方:65536=256 6553...
Mobi、epub格式电子书如... 在wps里全局设置里有一个文件关联,打开,勾选电子书文件选项就可以了。
scoped_dir32_70... 一台虚拟机C盘总是莫名奇妙的空间用完,导致很多软件没法再运行。经过仔细检查发现是C:\Program...
小程序支付时提示:appid和... [Q]小程序支付时提示:appid和mch_id不匹配 [A]小程序和微信支付没有进行关联,访问“小...
pycparser 是一个用... `pycparser` 是一个用 Python 编写的 C 语言解析器。它可以用来解析 C 代码并构...
微信小程序使用slider实现... 众所周知哈,微信小程序里面的音频播放是没有进度条的,但最近有个项目呢,客户要求音频要有进度条控制,所...
python查找阿姆斯特朗数 题目解释 如果一个n位正整数等于其各位数字的n次方之和,则称该数为阿姆斯特朗数。 例如1^3 + 5...
Apache Doris 2.... 亲爱的社区小伙伴们,我们很高兴地向大家宣布,Apache Doris 2.0.0 版本已于...