python ftp测试

刚学Python，做的ftp测试

1，简单ftp

vim test.py

#!/usr/bin/env python
import time
from ftplib import FTP
local_dir_update="*****"
local_dir_download="*******"
remote_host="*.*.*.*"
port="21"
remote_user="name"
remote_passwd="password"
ftp=FTP()
ftp.set_debuglevel(2)
ftp.connect(remote_host,port)
ftp.login(remote_user,remote_passwd)
print ftp.getwelcome()
#ftp.cwd()
bufsize = 1024
filename='ha.txt'
file_handler = open(filename,'rb')
ftp.storbinary('STOR ha.txt',file_handler,bufsize)
ftp.set_debuglevel(0)
file_handler.close()
ftp.quit()

2.稍复杂点

vim myftp.py

#!/usr/bin/env python
import time,socket
from ftplib import FTP
class MyFtp:
        def __init__(self,hostaddr,username,password,remotedir='./',port=21):
                self.hostaddr=hostaddr
                self.username = username 
                self.password = password 
                self.remotedir  = remotedir 
                self.port     = port 
                self.ftp      = FTP() 
                self.file_list = []
        def __del__(self):
                self.ftp.quit()
        def login(self):
                ftp = self.ftp
                try:
                        timeout = 60
                        socket.setdefaulttimeout(timeout) 
                        ftp.set_pasv(True) 
                        print 'connecting %s...' %(self.hostaddr)
                        ftp.connect(self.hostaddr, self.port) 
                        ftp.login(self.username, self.password) 
                        print 'login is ok'
                except Exception,e: 
                        print str(e)
                try: 
                        ftp.cwd(self.remotedir) 
                except(Exception): 
                        print 'change dir is failed..'
        def put_files(self,filename):
                ftp = self.ftp
                self.filename=filename
                command='STOR '+self.filename
                file_handler = open(self.filename,'rb')
                bufsize=1024
                ftp.storbinary(command,file_handler,bufsize)
                print 'upload is ok'
                file_handler.close()
        def download_files(self,filename):
                ftp = self.ftp
                self.filename = filename
                command='RETR '+self.filename
                bufsize=1024
                ftp.retrbinary(command,open(self.filename,'wb').write,bufsize)
if __name__ == '__main__':
        remote_host="*******"
        remote_user="name"
        remote_passwd="*******"
        remotedir='./'
        port='21'
        filename='ha.txt'
        f = MyFtp(remote_host,remote_user,remote_passwd,remotedir,port)
        f.login()
#       f.put_files(filename)
#       f.download_files(filename)
        a=['ha.txt','ha1.txt','ha2.txt']
        for a1 in a:
                f.download_files(a1)
        print 'download is ok'

3.与上面配合写的一个，在一个特定目录下的文件执行脚本到服务器上取.zip文件。(工作中用到，才写）

vim download.py

#!/usr/bin/env python
import os,time
from  myftp import MyFtp
file_list=['xxxxxx','xxxxxx,'xxxxxx','xxxxxx','xxxxxx']
local_dir_update="/xxxxxx"
local_dir_download="/xxxxxx"
a=time.localtime(time.time()+24*60*60)
dt=time.strftime('%Y%m%d',a)
if os.path.exists(local_dir_download+os.sep+dt):
        print "ok"
else:
        os.mkdir(local_dir_download+os.sep+dt)

if __name__ == '__main__':
        remote_host="xxxxxx"
        remote_user="cxxxxxx"
        remote_passwd="xxxxxx"
        remotedir=dt
        port='21'
        filename='ha.txt'
        f = MyFtp(remote_host,remote_user,remote_passwd,remotedir,port)
        f.login()
#       f.put_files(filename)
#       f.download_files(filename)
        a=os.listdir(local_dir_update)
        os.chdir(local_dir_download+os.sep+dt)
        for a1 in a:
                if a1 in file_list:
                        f.download_files(a1+".zip")
        print 'download is ok'