Python ftplib实现图片管理中心

前言：
在Windows机器上搭建图片服务中心，提供图片上传与下载，在Jenkins执行自动化测试任务中，对APP自动化生成的失败截图，可进行统一管理。

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

目录：

1. Windows搭建Ftp服务
2. Python ftplib讲解
3. 图片管理中心数据上传与下载

正文：
一. Windows搭建Ftp服务
参考文档：http://jingyan.baidu.com/article/455a9950e1e2fba167277862.html

核心步骤：
1.开启window ftp服务

2.配置ftp服务
计算机管理->服务与应用程序->IIS管理器->右键点击 网站->新建站点

3.开启成功检测

二. Python ftplib讲解
1.Python中默认安装的ftplib模块，其中函数有限，函数列举如下：

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文件

2.FTP.quit()与FTP.close()的区别：

FTP.quit():发送QUIT命令给服务器并关闭掉连接。这是一个比较“缓和”的关闭连接方式，但是如果服务器对QUIT命令返回错误时，会抛出异常。
FTP.close()：单方面的关闭掉连接，不应该用在已经关闭的连接之后，例如不应用在FTP.quit()之后。

3.ftp简单的上传与下载：

#-*- coding:utf-8 -*-
from ftplib import FTP
import time
import tarfile

def ftpconnect(host, username, password):
    ftp = FTP()
    #ftp.set_debuglevel(2)         #打开调试级别2，显示详细信息
    ftp.connect(host, 21)          #连接
    ftp.login(username, password)  #登录，如果匿名登录则用空串代替即可
    return ftp

def downloadfile(ftp, remotepath, localpath):
    bufsize = 1024                #设置缓冲块大小
    fp = open(localpath,'wb')     #以写模式在本地打开文件
    ftp.retrbinary('RETR ' + remotepath, fp.write, bufsize) #接收服务器上文件并写入本地文件
    ftp.set_debuglevel(0)         #关闭调试
    fp.close()                    #关闭文件

def uploadfile(ftp, remotepath, localpath):
    bufsize = 1024
    fp = open(localpath, 'rb')
    ftp.storbinary('STOR '+ remotepath , fp, bufsize) #上传文件
    ftp.set_debuglevel(0)
    fp.close()                                    

if __name__ == "__main__":
    ftp = ftpconnect("******", "***", "***")
    downloadfile(ftp, "***", "***")
    uploadfile(ftp, "***", "***")

    ftp.quit()

三. 图片管理中心数据上传
1.Jenkins执行完任务，将生成的失败截图上传图片管理中心

# -*- coding:utf-8 -*-
import sys
import os
from ftplib import FTP


# 需要创建的文件夹名称,与质量中心执行任务的编号一致
dirName = sys.argv[1]


# 建立连接
def ftp_connect():
    ftp = FTP()
    ftp.connect("10.9.8.*", "21", timeout=10)
    ftp.login("timen.xu", "xyt123456")
    return ftp


# 上传文件
def upload_file(ftp, remote_path, local_path):
    buf_size = 1024
    fp = open(local_path, 'rb')
    ftp.storbinary('STOR ' + remote_path, fp, buf_size)
    ftp.set_debuglevel(0)
    fp.close()


# 主方法
if __name__ == '__main__':
    ftp_conn = ftp_connect()
    ftp_conn.mkd("/" + dirName + "/")
    pic_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) + "/screenshots/"
    for pic_name in os.walk(pic_path):
        for pic in pic_name[2]:
            upload_file(ftp_conn, "/" + dirName + "/" + pic, pic_path + pic)
    ftp_conn.quit()

总结：
Python ftplib虽然函数有限，提供了简单的几个函数，但实现了最基本的上传下载功能，可以用于自动化部署文件下载，日志文件下载等。

参考：
Python ftplib模块