前言
最近因为业务需求,就写了这个脚本,脚本完成的任务是从FTP上下载一个目录,大家都知道从FTP上下载一个文件可用用get命令,下载多个文件可以用mget,但你要下载一个目录,对不起,不能,如果非要较劲,你说用lftp,然后用mirror命令也可以呀,这个我不较真,因为每个公司有每个公司的场景,选择合适的就行,因为我们这的FTP都是用ssl的,所以lftp就不能用了,脚本基本思路是建立ftp连接,然后登陆获取文件列表,根据返回的列表进行for循环,一个一个下载。
脚本如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#!/usr/bin/evn python
from ftplib import FTP_TLS, FTP
import socket
import ssl
import os
import sys
class IMPLICIT_FTP_TLS(FTP_TLS):
#构造函数初始化父类
def __init__( self , host = ' ', user=' ', passwd=' ', acct=' ', keyfile = None ,
certfile = None , timeout = 60 ):
FTP_TLS.__init__( self , host, user, passwd, acct, keyfile, certfile, timeout)
#建立服务器的FTPS连接
def connect( self , host = '', port = 0 , timeout = - 999 ):
if host ! = '':
self .host = host
if port > 0 :
self .port = port
if timeout ! = - 999 :
self .timeout = timeout
try :
#创建socket
self .sock = socket.create_connection(( self .host, self .port), self .timeout)
self .af = self .sock.family
#wrap_socket接收一个socket实例,返回SSLSocket实例,可以理解在普通socket上封装了一层ssl
self .sock = ssl.wrap_socket( self .sock, self .keyfile, self .certfile)
self . file = self .sock.makefile( 'rb' )
self .welcome = self .getresp()
except Exception as e:
print (e)
return self .welcome
def get_ftp_ver(version):
#如果没有这个目录就新建
if not os.path.isdir(version):
os.makedirs(version)
ftps = IMPLICIT_FTP_TLS()
ftps.connect(host = '10.0.0.8' , port = 666 )
ftps.login(user = "ftp_user" , passwd = "ftp_password" )
#建立安全的数据连接,之后才能返回数据。
ftps.prot_p()
ftps.cwd(version)
files = ftps.nlst()
#进入本地目录
os.chdir(version)
#循环下载每个文件
for file in files:
fp = open ( file , 'wb' )
ftps.retrbinary( 'RETR %s' % file , fp.write)
ftps.close()
if __name__ = = '__main__' :
get_ftp_ver(sys.argv[ 1 ])
|
脚本使用:
1
|
#python get_data.py version_20160920
|
后面跟的参数基本就是研发提供给你要更新的文件名称,然后运行就可以下载整个目录中的文件,脚本其它内容大家看注释就会明白。
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作能带来一定的帮助,如果有疑问大家可以留言交流。
原文链接:https://www.mindg.cn/?p=1627