github
/
spug
mirror of https://github.com/openspug/spug
1
0
Fork 0
Code Issues Releases Wiki Activity

A 添加文件分发模块

pull/517/head
vapao 2022-06-26 22:10:13 +08:00
parent 8e7c8d8e74
commit 18313b286c
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spug_api/apps/exec/transfer.py
View File

@ -106,13 +106,19 @@ def _do_sync(rds, task, host):
fp.flush() fp.flush()
options = '-azv' if task.host_id else '-rzv' options = '-azv' if task.host_id else '-rzv'
target = f'{host.username}@{host.hostname}:{task.dst_dir}' argument = f'{task.src_dir}/ {host.username}@{host.hostname}:{task.dst_dir}'
command = f'rsync {options} -h -e "ssh -p {host.port} -i {fp.name}" {task.src_dir}/ {target}' command = f'rsync {options} -h -e "ssh -p {host.port} -o StrictHostKeyChecking=no -i {fp.name}" {argument}'
task = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) task = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
while True: while True:
message = task.stdout.readline() message = task.stdout.readline()
if not message: if not message:
break break
message = message.decode().rstrip('\r\n') message = message.decode().rstrip('\r\n')
if 'rsync: command not found' in message:
data = '\r\n\x1b[31m检测到该主机未安装rsync，可通过批量执行/执行任务模块进行以下命令批量安装\x1b[0m'
data += '\r\nCentos/Redhat: yum install -y rsync'
data += '\r\nUbuntu/Debian: apt install -y rsync'
rds.publish(token, json.dumps({'key': host.id, 'data': data}))
break
rds.publish(token, json.dumps({'key': host.id, 'data': message + '\r\n'})) rds.publish(token, json.dumps({'key': host.id, 'data': message + '\r\n'}))
rds.publish(token, json.dumps({'key': host.id, 'status': task.wait()})) rds.publish(token, json.dumps({'key': host.id, 'status': task.wait()}))