|

楼主 |
发表于 2023-10-31 21:08:42
|
显示全部楼层
学了2天学了点皮毛,打算用通过ftp协议来实现。
瞎整了一段代码,但是运行起来有点问题,请大佬帮我看看
- from ftplib import FTP
- import os
- import sys
- import time
- import socket
- class MyFTP():
- # 初始化FTP客户端
- def __init__(self, host, port=21):
- self.host = host
- self.port = port
- self.ftp = FTP()
- # 重新设置下编码方式
- self.ftp.encoding = 'gbk'
- self.log_file = open("log.txt", "a")
- self.file_list = []
- # FTP客户端登录
- def login(self, username, password):
- try:
- timeout = 60
- socket.setdefaulttimeout(timeout)
- # 0主动模式 1 #被动模式
- self.ftp.set_pasv(True)
- # 打开调试级别2,显示详细信息
- # self.ftp.set_debuglevel(2)
- self.debug_print('开始尝试连接到 %s' % self.host)
- self.ftp.connect(self.host, self.port)
- self.debug_print('成功连接到 %s' % self.host)
- self.debug_print('开始尝试登录到 %s' % self.host)
- self.ftp.login(username, password)
- self.debug_print('成功登录到 %s' % self.host)
- self.debug_print(self.ftp.welcome)
- except Exception as error:
- self.deal_error("FTP 连接或登录失败 ,错误描述为:%s" % error)
- pass
- #用于下载前判断远程文件和本地文件大小是否一致
- def is_same_size(self, local_file, remote_file):
- try:
- remote_file_size = self.ftp.size(remote_file)
- except Exception as error:
- # self.debug_print("is_same_size() 错误描述为:%s" % error)
- remote_file_size = -1
- try:
- local_file_size = os.path.getsize(local_file)
- except Exception as error:
- # self.debug_print("is_same_size() 错误描述为:%s" % error)
- local_file_size = -1
- self.debug_print('local_file_size:%d , remote_file_size:%d' % (local_file_size, remote_file_size))
- if remote_file_size == local_file_size:
- return 1
- else:
- return 0
- #从ftp下载单个文件
- def download_file(self, local_file, remote_file):
- self.debug_print("download_file()---> local_path = %s ,remote_path = %s" % (local_file, remote_file))
- if self.is_same_size(local_file, remote_file):
- self.debug_print('%s 文件大小相同,无需下载' % local_file)
- return
- else:
- try:
- self.debug_print('>>>>>>>>>>>>下载文件 %s ... ...' % local_file)
- buf_size = 1024
- file_handler = open(local_file, 'wb')
- self.ftp.retrbinary('RETR %s' % remote_file, file_handler.write, buf_size)
- file_handler.close()
- except Exception as error:
- self.debug_print('下载文件出错,出现异常:%s ' % error)
- return
- #从远程目录下载多个文件到本地目录
- def download_file_tree(self, local_path, remote_path):
- print("download_file_tree()---> local_path = %s ,remote_path = %s" % (local_path, remote_path))
- try:
- self.ftp.cwd(remote_path)
- except Exception as error:
- self.debug_print('远程目录%s不存在,继续...' % remote_path + " ,具体错误描述为:%s" % error)
- return
- if not os.path.isdir(local_path):
- self.debug_print('本地目录%s不存在,先创建本地目录' % local_path)
- os.makedirs(local_path)
- self.debug_print('切换至目录: %s' % self.ftp.pwd())
- self.file_list = []
- # 方法回调
- self.ftp.dir(self.get_file_list)
- remote_names = self.file_list
- self.debug_print('远程目录 列表: %s' % remote_names)
- for item in remote_names:
- file_type = item[0]
- file_name = item[1]
- local = os.path.join(local_path, file_name)
- if file_type == 'd':
- print("download_file_tree()---> 下载目录: %s" % file_name)
- self.download_file_tree(local, file_name)
- elif file_type == '-':
- print("download_file()---> 下载文件: %s" % file_name)
- self.download_file(local, file_name)
- self.ftp.cwd("..")
- self.debug_print('返回上层目录 %s' % self.ftp.pwd())
- return True
- # 从本地上传文件到ftp
- def upload_file(self, local_file, remote_file):
- if not os.path.isfile(local_file):
- self.debug_print('%s 不存在' % local_file)
- return
- if self.is_same_size(local_file, remote_file):
- self.debug_print('跳过相等的文件: %s' % local_file)
- return
- buf_size = 1024
- file_handler = open(local_file, 'rb')
- self.ftp.storbinary('STOR %s' % remote_file, file_handler, buf_size)
- file_handler.close()
- self.debug_print('上传: %s' % local_file + "成功!")
- # 从本地上传目录下多个文件到ftp
- def upload_file_tree(self, local_path, remote_path):
- if not os.path.isdir(local_path):
- self.debug_print('本地目录 %s 不存在' % local_path)
- return
- self.ftp.cwd(remote_path)
- self.debug_print('切换至远程目录: %s' % self.ftp.pwd())
- local_name_list = os.listdir(local_path)
- for local_name in local_name_list:
- src = os.path.join(local_path, local_name)
- if os.path.isdir(src):
- try:
- self.ftp.mkd(local_name)
- except Exception as error:
- self.debug_print("目录已存在 %s ,具体错误描述为:%s" % (local_name, error))
- self.debug_print("upload_file_tree()---> 上传目录: %s" % local_name)
- self.upload_file_tree(src, local_name)
- else:
- self.debug_print("upload_file_tree()---> 上传文件: %s" % local_name)
- self.upload_file(src, local_name)
- self.ftp.cwd("..")
- def close(self):
- self.debug_print("close()---> FTP退出")
- self.ftp.quit()
- self.log_file.close()
- # 打印日志
- def debug_print(self, s):
- self.write_log(s)
- # 处理错误异常
- def deal_error(self, error):
- log_str = '发生错误: %s' % error
- self.write_log(log_str)
- sys.exit()
- # 记录日志
- def write_log(self, log_str):
- time_now = time.localtime()
- date_now = time.strftime('%Y-%m-%d', time_now)
- format_log_str = "%s ---> %s \n " % (date_now, log_str)
- print(format_log_str)
- self.log_file.write(format_log_str)
- # 获取文件列表
- def get_file_list(self, line):
- file_arr = self.get_file_name(line)
- # 去除 . 和 ..
- if file_arr[1] not in ['.', '..']:
- self.file_list.append(file_arr)
- # 获取文件名
- def get_file_name(self, line):
- pos = line.rfind(':')
- while (line[pos] != ' '):
- pos += 1
- while (line[pos] == ' '):
- pos += 1
- file_arr = [line[0], line[pos:]]
- return file_arr
- if __name__ == "__main__":
- my_ftp = MyFTP("130.1.11.70")
- my_ftp.login("xxk", "xxk@2021")
- # 下载单个文件
- # my_ftp.download_file("D:/download", "./update/")
- # 下载目录
- my_ftp.download_file_tree(r"D:\download", "/update/")
- # 上传单个文件
- # my_ftp.upload_file("G:/f/", "/App/")
- # 上传目录
- # my_ftp.upload_file_tree("G:/f/", "/App/")
- my_ftp.close()
复制代码
ftp服务器上有4个文件
下载到本地只有1个文件
下载文件夹的时候会报错
以下是报错日志:
|
|