近期利用lsync同步发现较慢,所以在脚本后面通过管道来同步文件.
内容如下:
#!/usr/bin/python
import paramiko
import os
import sys
import time
import multiprocessing
import datetime
ip_list = []
room_id = sys.argv[1]
cur_time = datetime.datetime.now()
def Upfile(host_ip,local_path,remote_path):
privatekey = os.path.expanduser('/root/.ssh/id_rsa')
key=paramiko.RSAKey.from_private_key_file(privatekey)
scp = paramiko.Transport((host_ip, 22))
scp.connect(username='root', pkey=key)
sftp = paramiko.SFTPClient.from_transport(scp)
src = local_path
des = remote_path
os.chdir(os.path.split(local_path)[0])
parent=os.path.split(local_path)[0]
for walker in os.walk(parent):
try:
sftp.mkdir(walker[0])
for file in walker[2]:
sftp.put(os.path.join(walker[0],file),os.path.join(walker[0],file))
print "execute time is %s and room is %s" %(cur_time,room_id)
except:
for file in walker[2]:
sftp.put(os.path.join(walker[0],file),os.path.join(walker[0],file))
scp.close()
def all_ip():
f = file('/opt/scripts/static_host.txt','r')
c = f.readlines()
for x in c:
ip = x.split('\n')[0]
ip_list.append(ip)
f.close()
if __name__=='__main__':
print "-"*50
print room_id
if not room_id:
sys.exit()
all_ip()
pool = multiprocessing.Pool(processes=5)
threads = []
print "Begin......"
for i in ip_list:
room_path = '/app/data/ckl/'+ room_id + '/'
pool.apply_async(Upfile,(i,room_path,room_path))
pool.close()
pool.join()
for res in threads:
print res.get()在php执行完成后中加管道来调用此脚本:
php /app/data/ckl/yiic json notify | sed -n 's/^Done: \(.*\)/\1/p' | xargs -I {} /opt/scripts/sync_rooms.py {}下面再贴网上来的一段:
import socket
import os
from stat import S_ISDIR
class SSHSession(object):
# Usage:
# Detects DSA or RSA from key_file, either as a string filename or a
# file object. Password auth is possible, but I will judge you for
# using it. So:
# ssh=SSHSession('targetserver.com','root',key_file=open('mykey.pem','r'))
# ssh=SSHSession('targetserver.com','root',key_file='/home/me/mykey.pem')
# ssh=SSHSession('targetserver.com','root','mypassword')
# ssh.put('filename','/remote/file/destination/path')
# ssh.put_all('/path/to/local/source/dir','/path/to/remote/destination')
# ssh.get_all('/path/to/remote/source/dir','/path/to/local/destination')
# ssh.command('echo "Command to execute"')
def __init__(self,hostname,username='root',key_file=None,password=None):
#
# Accepts a file-like object (anything with a readlines() function)
# in either dss_key or rsa_key with a private key. Since I don't
# ever intend to leave a server open to a password auth.
#
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.connect((hostname,22))
self.t = paramiko.Transport(self.sock)
self.t.start_client()
keys = paramiko.util.load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
key = self.t.get_remote_server_key()
# supposed to check for key in keys, but I don't much care right now to find the right notation
if key_file is not None:
if isinstance(key,str):
key_file=open(key,'r')
key_head=key_file.readline()
key_file.seek(0)
if 'DSA' in key_head:
keytype=paramiko.DSSKey
elif 'RSA' in key_head:
keytype=paramiko.RSAKey
else:
raise Exception("Can't identify key type")
pkey=keytype.from_private_key(key_file)
self.t.auth_publickey(username, pkey)
else:
if password is not None:
self.t.auth_password(username,password,fallback=False)
else: raise Exception('Must supply either key_file or password')
self.sftp=paramiko.SFTPClient.from_transport(self.t)
def command(self,cmd):
# Breaks the command by lines, sends and receives
# each line and its output separately
#
# Returns the server response text as a string
chan = self.t.open_session()
chan.get_pty()
chan.invoke_shell()
chan.settimeout(20.0)
ret=''
try:
ret+=chan.recv(1024)
except:
chan.send('\n')
ret+=chan.recv(1024)
ret+=chan.recv(1024)
ret+=chan.recv(1024)
for line in cmd.split('\n'):
chan.send(line.strip() + '\n')
ret+=chan.recv(1024)
return ret
def put(self,localfile,remotefile):
# Copy localfile to remotefile, overwriting or creating as needed.
self.sftp.put(localfile,remotefile)
def put_all(self,localpath,remotepath):
# recursively upload a full directory
os.chdir(os.path.split(localpath)[0])
parent=os.path.split(localpath)[1]
for walker in os.walk(parent):
try:
self.sftp.mkdir(os.path.join(remotepath,walker[0]))
except:
pass
for file in walker[2]:
self.put(os.path.join(walker[0],file),os.path.join(remotepath,walker[0],file))
def get(self,remotefile,localfile):
# Copy remotefile to localfile, overwriting or creating as needed.
self.sftp.get(remotefile,localfile)
def sftp_walk(self,remotepath):
# Kindof a stripped down version of os.walk, implemented for
# sftp. Tried running it flat without the yields, but it really
# chokes on big directories.
path=remotepath
files=[]
folders=[]
for f in self.sftp.listdir_attr(remotepath):
if S_ISDIR(f.st_mode):
folders.append(f.filename)
else:
files.append(f.filename)
print (path,folders,files)
yield path,folders,files
for folder in folders:
new_path=os.path.join(remotepath,folder)
for x in self.sftp_walk(new_path):
yield x
def get_all(self,remotepath,localpath):
# recursively download a full directory
# Harder than it sounded at first, since paramiko won't walk
#
# For the record, something like this would gennerally be faster:
# ssh user@host 'tar -cz /source/folder' | tar -xz
self.sftp.chdir(os.path.split(remotepath)[0])
parent=os.path.split(remotepath)[1]
try:
os.mkdir(localpath)
except:
pass
for walker in self.sftp_walk(parent):
try:
os.mkdir(os.path.join(localpath,walker[0]))
except:
pass
for file in walker[2]:
self.get(os.path.join(walker[0],file),os.path.join(localpath,walker[0],file))
def write_command(self,text,remotefile):
# Writes text to remotefile, and makes remotefile executable.
# This is perhaps a bit niche, but I was thinking I needed it.
# For the record, I was incorrect.
self.sftp.open(remotefile,'w').write(text)
self.sftp.chmod(remotefile,755)转载于:https://blog.51cto.com/ckl893/1785297
本文介绍了一个使用Python和Paramiko库实现的批量文件同步脚本。该脚本通过SSH连接远程服务器,并利用多进程提高文件传输效率。此外,还提供了一种在PHP脚本执行完成后触发该Python脚本的方法。
886

被折叠的 条评论
为什么被折叠?



