怎样用python实现scp

怎样用python实现scp,第1张

scp 功能很强大,但需要人工输入 password, 当然可以通过把 公钥保存在远程主机的 ~/.ssh 目录中,而后就不用输入password,但这需要配置.

用 sshpass 可能在命令输入 password, 但 需稿宏要用 “sudo apt-get install sshpass” 安装

如果不想用上面两种方法,可以用 expect 编写脚本可以帮助我们自动交互

虽然 Python 也提供 pexpect 模块,但既然 expect 很简单,为何不直接用 os.system() 去执行呢?

下面是我编写的类,实现了远程复制

[html] view plain copy

class RemoteShell:

def __init__(self, host, user, pwd):

self.host = host

self.user = user

self.pwd = pwd

def put(self, local_path, remote_path):

scp_put = '''

spawn scp %s %s@%s:%s

expect "(yes/no)?" {

send "yes\r"

expect "password:"

send "%s\r"

} "password:" {send "%s\r"}

expect eof

exit'''

os.system("echo '%s' >scp_put.cmd" % (scp_put % (os.path.expanduser(local_path), self.user, self.host, remote_path, self.pwd, self.pwd)))

os.system('expect scp_put.cmd')

os.system('rm scp_put.cmd')

但发现每次文件都没有复制完,我想看expect 究竟做了什么事情,还好 expect 提供 -d 参数给出更多的信息。

最后发现是被复制文件桐蔽太大,expect 超时退出了

在脚本前加入 “set timeout -1" 就OK了

[html] view plain copy

scp_put = '''

set timeout -1

spawn scp %s %s@%s:%s

expect "(yes/no)?" {

send "yes\局敬州r"

expect "password:"

send "%s\r"

} "password:" {send "%s\r"}

expect eof

exit'''

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import pexpect

def ssh_cmd(ip, passwd, cmd):

ret = -1

ssh = pexpect.spawn('ssh root@%s "%s"' % (ip, cmd))

try:

i = ssh.expect(['password:', 'continue connecting (yes/no)?'], timeout=5)

if i == 0 :

ssh.sendline(passwd)

elif i == 1:

ssh.sendline('yes\n')

ssh.expect('password: ')

ssh.sendline(passwd)

ssh.sendline(cmd)

r = ssh.read()

print r

ret = 0

except pexpect.EOF:

print "EOF"

ssh.close()

ret = -1

except pexpect.TIMEOUT:

print "TIMEOUT"

ssh.close()

ret = -2

return ret

利用pexpect模块我们可以做很多事情,由于他提供了自动交互功能,因此我们可以实现ftp,telnet,ssh,scp等的自动登录,还是比较实用的。根据上面的代码相信读者已经知道怎么实现了(python就码腔谈是那么简单!)。

上面的代码去完成任务还是比较费时间的,因为程序要等待自动交互出现,另外ubuntu用ssh连接就是比较慢,要进行一系列的验证,这样才体现出ssh

的安全。我们要提高效率,在最短的时间内完成。后来我发现了python里面的paramiko模块,用这个实现ssh登录更加简单迟碰。看下面的代码:

复制代码 代码如下:

#-*- coding: utf-8 -*-

#!/usr/bin/python

import paramiko

import threading

def ssh2(ip,username,passwd,cmd):

try:

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

ssh.connect(ip,22,username,passwd,timeout=5)

for m in cmd:

stdin, stdout, stderr = ssh.exec_command(m)

# stdin.write("Y") #简单交互,输入 ‘Y'

out = stdout.readlines()

#屏幕输出

for o in out:

print o,

print '%s\tOK\n'%(ip)

ssh.close()

except :

print '%s\tError\n'%(ip)

if __name__=='__main__':

cmd = ['cal','echo hello!']#你要执行的命令列表

username = "" #用户名

passwd = "" #密码

threads = [] #多线程

print "Begin......"

for i in range(1,254):

ip = '192.168.1.'+str(i)

a=threading.Thread(target=ssh2,args=(ip,username,passwd,cmd))

a.start()

上面的程序还是有些圆孙技巧的:

1.

利用多线程,同时发出登录请求,同时去连接电脑,这样速度快很多,我试了一下,如果不用多线程,直接一个一个挨着执行的话,大约5~10秒钟才能对一台电

脑 *** 作完,具体时间要根据命令的来决定,如果是软件安装或者卸载时间要更长一些。这样下来怎么也要一二十分钟,用多线程后就快多了,所有的命令执行完用了

不到2分钟!

2.最好用root用户登录,因为安装或者卸载软件的时候如果用普通用户又会提示输入密码,这样又多了一次交互,处理起来就比较麻

烦!安装软件时apt-get install xxx

最好加上“-y”参数,因为有时安装或删除软件时提示是否继续安装或卸载,这又是一次自动交互!加上那个参数后就没有人机交互了。

3. 循环时循环所有ip,因为计算机的ip是路由器自动分配的,保险起见,最好全部都执行,保证没有遗漏的主机

4.远端执行命令时如果有交互,可以这样用 stdin.write("Y")来完成交互,“Y”就是输入“Y”。

5.把所有的命令放到一个列表里面,遍历列表可以依次执行列表里面的命令

6.为了更好的进行控制,最好在电脑上提前把root用户打开,装好ssh服务器并让其开机自动执行。

1、可以配置无密码访问或者用sshpass在shell中存密码

2、实例

ip.txt包含ip列表,每行余哗一个ip

test.sh保护修改配置的命令或者直接修改知首好,复制竖猛行到远程指定路径。

3、代码

#!/bin/sh

for

ip

in

`cat

ip.txt`

do

echo

${ip}

scp

-

P22

test.sh

root...


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/tougao/12301990.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-05-24
下一篇 2023-05-24

发表评论

登录后才能评论

评论列表(0条)

保存