用python写测试脚本，从本地传文件至ftp远程路径

转自：

创新互联主要从事网站设计制作、做网站、网页设计、企业做网站、公司建网站等业务。立足成都服务山城,十年网站建设经验,价格优惠、服务专业,欢迎来电咨询建站服务:13518219792

本文实例讲述了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')

('test','test')

#上传文件，不重命名

#('111.txt','a/b')

#上传文件，重命名

#('111.txt','a/112.txt')

#下载文件，不重命名

#('/a/111.txt',r'D:\\')

#下载文件，重命名

#('/a/111.txt',r'D:\112.txt')

#下载到已经存在的文件夹

#('a/b/c',r'D:\\a')

#下载到不存在的文件夹

#('a/b/c',r'D:\\aa')

#上传到已经存在的文件夹

('b','a')

#上传到不存在的文件夹

('b','aa/B/')

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

以下转自：

Python中的ftplib模块

Python中默认安装的ftplib模块定义了FTP类，其中函数有限，可用来实现简单的ftp客户端，用于上传或下载文件

FTP的工作流程及基本操作可参考协议RFC959

ftp登陆连接

from ftplib import FTP #加载ftp模块

ftp=FTP() #设置变量

#打开调试级别2，显示详细信息

("IP","port") #连接的ftp sever和端口

("user","password")#连接的用户名，密码

print #打印出欢迎信息

("xxx/xxx") #更改远程目录

bufsize=1024 #设置的缓冲区大小

filename="filename.txt" #需要下载的文件

file_handle=open(filename,"wb").write #以写模式在本地打开文件

("RETR filename.txt",file_handle,bufsize) #接收服务器上文件并写入本地文件

#关闭调试模式

#退出ftp

ftp相关命令操作

#设置FTP当前操作的路径

#显示目录下文件信息

#获取目录下的文件

#新建远程目录

#返回当前所在位置

#删除远程目录

#删除远程文件

(fromname, toname)#将fromname修改名称为toname。

("STOR filename.txt",file_handel,bufsize) #上传目标文件

("RETR filename.txt",file_handel,bufsize)#下载FTP文件

网上找到一个具体的例子：

# 例：FTP编程  

from ftplib import FTP  

ftp = FTP()  

timeout = 30  

port = 21  

('192.168.1.188',port,timeout) # 连接FTP服务器  

('UserName','888888') # 登录  

print   # 获得欢迎信息   

('file/test')    # 设置FTP路径  

list =        # 获得目录列表  

for name in list:  

print(name)             # 打印文件名字  

path = 'd:/data/' + name    # 文件保存路径  

f = open(path,'wb')         # 打开要保存文件  

filename = 'RETR ' + name   # 保存FTP文件  

# 保存FTP上的文件  

# 删除FTP文件  

('STOR '+filename, open(path, 'rb')) # 上传FTP文件  

# 退出FTP服务器

完整的模板：

#!/usr/bin/python  

# -*- coding: utf-8 -*-  

import ftplib  

import os  

import socket  

HOST = ''  

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()

python测试函数有哪些

测试函数是用于自动化测试，使用python模块中的unittest中的工具来测试

附上书中摘抄来的代码：

#coding=utf-8import unittestfrom name_function import get_formatted_nameclass NamesTestCase(unittest.TestCase): def test_first_last_name(self): formatted_name=get_formatted_name('janis','joplin') self.assertEqual(formatted_name,'Janis Joplin') def test_first_last_middle_name(self): formatted_name=get_formatted_name('wolfgang','mozart','amadeus') self.assertEqual(formatted_name,'Wolfgang Amadeus Mozart')#注意下面这行代码，不写会报错哦~~~书中没有这行if __name__=="__main__": unittest.main()

python如何在命令行定义函数

Python在命令行定义函数的方法如下：

打开电脑运行窗体，输入cmd,点击确定

命令行窗口，输入python,进入python命令行，编写函数后，敲两次回车，即定义好了函数

测试函数可以正常使用

更多Python相关技术文章，请访问Python教程栏目进行学习！以上就是小编分享的关于python如何在命令行定义函数的详细内容希望对大家有所帮助，更多有关python教程请关注环球青藤其它相关文章！

用Python写入文件，若该行数据存在则不写入；若不存在，则写入。怎么才能在写入之前判断该数据是否存在

用open,该函数创建一个文件对象，这将用来调用与之关联的其他支持方式即可：

file object = open(file_name [, access_mode][, buffering])

file_name: file_name参数是一个字符串值，包含要访问的文件的名称。

access_mode: access_mode决定了文件必须被打开，即，读，写，追加等的可能值是下表中给定的一个完整的列表的模式。这是可选参数，默认文件存取方式为read (r)。

扩展资料：

file命令的函数意义：

file 命令读取用 File 参数或者 FileList 变量指定的文件，在每个文件上执行一系列测试，然后将它们按照类型分类。然后此命令将文件类型写入标准输出。文件可以是常规文件、目录、FIFO（指定的管道）、块特殊文件、字符特别文件、符号链接或者套接字类型。

1、对于长度为零的常规文件，将识别为空文件。

2、对于符号链接文件，缺省情况下此链接后跟符号链接引用的文件。

如果文件是 ASCII 码的格式，则 file 命令将检查前 1024 个字节然后确定文件类型。如果文件不是 ASCII 格式，则 file 命令将尝试区分二进制数据文件和包含扩展字符的文本文件。

参考资料来源：百度百科-File

参考资料来源：百度百科-DriveName

如何使用python编写测试脚本

1)doctest

使用doctest是一种类似于命令行尝试的方式，用法很简单，如下

复制代码代码如下:

def f(n):

"""

f(1)

1

f(2)

2

"""

print(n)

if __name__ == '__main__':

import doctest

doctest.testmod()

应该来说是足够简单了，另外还有一种方式doctest.testfile(filename)，就是把命令行的方式放在文件里进行测试。

2)unittest

unittest历史悠久，最早可以追溯到上世纪七八十年代了，C++，Java里也都有类似的实现，Python里的实现很简单。

unittest在python里主要的实现方式是TestCase,TestSuite。用法还是例子起步。

复制代码代码如下:

from widget import Widget

import unittest

# 执行测试的类

class WidgetTestCase(unittest.TestCase):

def setUp(self):

self.widget = Widget()

def tearDown(self):

self.widget.dispose()

self.widget = None

def testSize(self):

self.assertEqual(self.widget.getSize(), (40, 40))

def testResize(self):

self.widget.resize(100, 100)

self.assertEqual(self.widget.getSize(), (100, 100))

# 测试

if __name__ == "__main__":

# 构造测试集

suite = unittest.TestSuite()

suite.addTest(WidgetTestCase("testSize"))

suite.addTest(WidgetTestCase("testResize"))

# 执行测试

runner = unittest.TextTestRunner()

runner.run(suite)

简单的说，1构造TestCase(测试用例)，其中的setup和teardown负责预处理和善后工作。2构造测试集，添加用例3执行测试需要说明的是测试方法，在Python中有N多测试函数，主要的有：

TestCase.assert_(expr[, msg])

TestCase.failUnless(expr[, msg])

TestCase.assertTrue(expr[, msg])

TestCase.assertEqual(first, second[, msg])

TestCase.failUnlessEqual(first, second[, msg])

TestCase.assertNotEqual(first, second[, msg])

TestCase.failIfEqual(first, second[, msg])

TestCase.assertAlmostEqual(first, second[, places[, msg]])

TestCase.failUnlessAlmostEqual(first, second[, places[, msg]])

TestCase.assertNotAlmostEqual(first, second[, places[, msg]])

TestCase.failIfAlmostEqual(first, second[, places[, msg]])

TestCase.assertRaises(exception, callable, ...)

TestCase.failUnlessRaises(exception, callable, ...)

TestCase.failIf(expr[, msg])

TestCase.assertFalse(expr[, msg])

TestCase.fail([msg])

Python 编写并测试函数change(str1)，其功能是对参数str1进行大小写转换？

def change(str1):

new_str = str()

for i in range(len(str1)):

if(65 = ord(str1[i]) = 90):

a = str1[i].lower()

print(a,end='')

elif(97 = ord(str1[i]) = 122):

a = str1[i].upper()

print(a,end='')

else:

a = str1[i]

print(a,end='')

return new_str

str2 = str(input("要转换的字符串："))

print(change(str2))

当前文章：包含python测试函数写入的词条
当前地址：http://kswjz.com/article/dseppoo.html