用python写测试脚本,从本地传文件至ftp远程路径
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了用python写测试脚本,从本地传文件至ftp远程路径相关的知识,希望对你有一定的参考价值。
转自:http://news.tuxi.com.cn/kf/article/jhtdj.htm
本文实例讲述了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程序设计有所帮助。
以下转自:http://blog.csdn.net/linda1000/article/details/8255771
Python中的ftplib模块
Python中默认安装的ftplib模块定义了FTP类,其中函数有限,可用来实现简单的ftp客户端,用于上传或下载文件
FTP的工作流程及基本操作可参考协议RFC959
ftp登陆连接
from ftplib import FTP #加载ftp模块
ftp=FTP() #设置变量
ftp.set_debuglevel(2) #打开调试级别2,显示详细信息
ftp.connect("IP","port") #连接的ftp sever和端口
ftp.login("user","password")#连接的用户名,密码
print ftp.getwelcome() #打印出欢迎信息
ftp.cmd("xxx/xxx") #更改远程目录
bufsize=1024 #设置的缓冲区大小
filename="filename.txt" #需要下载的文件
file_handle=open(filename,"wb").write #以写模式在本地打开文件
ftp.retrbinaly("RETR filename.txt",file_handle,bufsize) #接收服务器上文件并写入本地文件
ftp.set_debuglevel(0) #关闭调试模式
ftp.quit #退出ftp
ftp相关命令操作
ftp.cwd(pathname) #设置FTP当前操作的路径
ftp.dir() #显示目录下文件信息
ftp.nlst() #获取目录下的文件
ftp.mkd(pathname) #新建远程目录
ftp.pwd() #返回当前所在位置
ftp.rmd(dirname) #删除远程目录
ftp.delete(filename) #删除远程文件
ftp.rename(fromname, toname)#将fromname修改名称为toname。
ftp.storbinaly("STOR filename.txt",file_handel,bufsize) #上传目标文件
ftp.retrbinary("RETR filename.txt",file_handel,bufsize)#下载FTP文件
网上找到一个具体的例子:
# 例:FTP编程from ftplib import FTP
ftp = FTP()
timeout = 30
port = 21
ftp.connect(\'192.168.1.188\',port,timeout) # 连接FTP服务器
ftp.login(\'UserName\',\'888888\') # 登录
print ftp.getwelcome() # 获得欢迎信息
ftp.cwd(\'file/test\') # 设置FTP路径
list = ftp.nlst() # 获得目录列表
for name in list:
print(name) # 打印文件名字
path = \'d:/data/\' + name # 文件保存路径
f = open(path,\'wb\') # 打开要保存文件
filename = \'RETR \' + name # 保存FTP文件
ftp.retrbinary(filename,f.write) # 保存FTP上的文件
ftp.delete(name) # 删除FTP文件
ftp.storbinary(\'STOR \'+filename, open(path, \'rb\')) # 上传FTP文件
ftp.quit() # 退出FTP服务器
完整的模板:
#!/usr/bin/python# -*- coding: utf-8 -*-
import ftplib
import os
import socket
HOST = \'ftp.mozilla.org\'
DIRN = \'pub/mozilla.org/webtools\'
FILE = \'bugzilla-3.6.7.tar.gz\'
def main():
try:
f = ftplib.FTP(HOST)
except (socket.error, socket.gaierror):
print \'ERROR:cannot reach " %s"\' % HOST
return
print \'***Connected to host "%s"\' % HOST
try:
f.login()
except ftplib.error_perm:
print \'ERROR: cannot login anonymously\'
f.quit()
return
print \'*** Logged in as "anonymously"\'
try:
f.cwd(DIRN)
except ftplib.error_perm:
print \'ERRORL cannot CD to "%s"\' % DIRN
f.quit()
return
print \'*** Changed to "%s" folder\' % DIRN
try:
#传一个回调函数给retrbinary() 它在每接收一个二进制数据时都会被调用
f.retrbinary(\'RETR %s\' % FILE, open(FILE, \'wb\').write)
except ftplib.error_perm:
print \'ERROR: cannot read file "%s"\' % FILE
os.unlink(FILE)
else:
print \'*** Downloaded "%s" to CWD\' % FILE
f.quit()
return
if __name__ == \'__main__\':
main()追问
还是没看明白,你只要告诉我从本地上传到ftp远程路径的是什么就可以,谢谢
追答# 例:FTP编程from ftplib import FTP
ftp = FTP()
timeout = 30
port = 21
ftp.connect(\'192.168.1.188\',port,timeout) # 连接FTP服务器
ftp.login(\'user\',\'pass\') # 登录
print(ftp.getwelcome()) # 获得欢迎信息
ftp.cwd(\'/\') # 设置FTP路径
name = \'filename.txt\'
path = name # 本地文件路径
ftp.storbinary(\'STOR \' + name, open(path, \'rb\')) # 上传FTP文件
ftp.quit() # 退出FTP服务器
我用的是python3.5 测试通过 如果是2.7版本就是print那改一下就行
FTP搭建与配置
FTP是File Transfer Protocol (文件传输协议)的英文简称,用于在Internet上控制文件的双向传输
FTP的主要作用就是让用户连接一个远程计算机(这些计算机上运行着FTP服务器的程序),并查看远程计算机中的文件,然后把文件从远程计算机复制到本地计算机,或把本地计算机的文件传送到远程计算机
小公司用的多,大企业不用FTP,因为不安全(使用自动化管理平台)
搭建FTP服务流程
centos上自带vsftpd
yum install -y vsftpd
useradd -s /sbin/nologin virftp 创建普通用户 virftp随意只是个名字
vim /etc/vsftpd/vsftpd_login //内容如下,奇数行为用户名,偶数行为密码,多个用户就写多行
testuser1
shenguanchu1
chmod 600 /etc/vsftpd/vsftpd_login 安全考虑
db_load -T -t hash -f /etc/vsftpd/vsftpd_login /etc/vsftpd/vsftpd_login.db 转化为计算机能识别的二进制文件
mkdir /etc/vsftpd/vsftpd_user_conf 创建虚拟用户配置文件所在目录
cd /etc/vsftpd/vsftpd_user_conf
vim testuser1 配置文件名字与用户需相同 //加入如下内容
local_root=/home/virftp/testuser1 定义虚拟用户家目录
anonymous_enable=NO 是否允许匿名用户
write_enable=YES 是否允许可写
local_umask=022 创建目录权限
anon_upload_enable=NO 是否允许匿名用户上传
anon_mkdir_write_enable=NO 是否允许匿名用户创建目录、写
idle_session_timeout=600 空闲时间段超时时间
data_connection_timeout=120 数据传输超时时间
max_clients=10 最大客户端
mkdir /home/virftp/testuser1 创建虚拟用户家目录
touch /home/virftp/tesruser1/shenguanchu.txt 创建一个测试文件
chmod -R virftp:virftp /home/virftp 修改权限
vim /etc/pam.d/vsftpd //在最前面加上 定义密码文件在哪 认证用的文件
auth sufficent /lib64/security/pam_userdb.so db=/etc/vsftpd/vsftpd_login
account sufficient /lib64/security/pam_userdb.so db=/etc/vsftpd/vsftpd_login
注意32位和64位
vim /etc/vsftpd/vsftpd.conf
将anonymous_enable=YES改成anonymous_enable=NO
将#anon_upload_enable=YES改成anon_upload_enable=NO
将#anon_mkdir_write_enable=YES改成anon_mkdir_write_enable=NO
再增加如下内容
chroot_local_user=YES
guest_enable=YES
guest_username=virftp
virtual_use_local_privs=YES
user_config_dir=/etc/vsftpd/vsftpd_user_conf
allow_writeable_chroot=YES
然后systemctl start vsftpd //启动vsftp服务 centos7以下用service vsftpd start
window客户端 安装软件 filezilla client 开源免费软件
Linux上安装lftp
测试FTP
yum install -y lftp
lftp [email protected]
执行命令ls,看是否正常输出
若不正常查看日志/var/log/messages和/var/log/secure
也可window上安装XFTP方便安全
以上是关于用python写测试脚本,从本地传文件至ftp远程路径的主要内容,如果未能解决你的问题,请参考以下文章
想将ubuntu下某个文件夹下的文件定时上传至FTP或者共享文件夹,也可以直接传至另一台LINUX上使用。