object-python-交互式登錄(scp/ssh)

#! /usr/bin/env python2.6
# 2013-1-23
# Tong.yang
#

"""
This runs a command on a remote host using SSH/SCP. At the prompts enter hostname, user, password and the command
"""
import pexpect
import os
import getpass
from time import sleep

def ssh_command(files, user, host, password):
        ssh_key = 'Are you sure you want to continue connecting'
#       child = pexpect.spawn('ssh -l %s %s %s' % (user, host, command))
        child = pexpect.spawn('/usr/bin/scp %s@%s:%s /tmp/123/' % (user, host, files))
        i = child.expect([pexpect.TIMEOUT, ssh_key, 'password: '])

        if i == 0: # TIMEOUT
                print 'Error!'
                print 'SSH could not login, Here is what SSH said:'
                print child.before, child.after
                return None
        if i == 1: # SSH does not have the public key, just accept it.
                child.sendline('yes')
                child.expect('password: ')
                i = child.expect([TIMEOUT, 'password: '])
                if i == 0: # TIMEOUT
                        print 'Error!'
                        print 'SSH could not login, Here is what SSH said:'
                        print child.before, child.after
                        print None
        # input password
        child.sendline(password)
        sleep(80)
        return child

def main():
#       host = raw_input('Hostname:')
#       user = raw_input('user:')
#       password = getpass.getpass()
#       files = raw_input('Enter file the absolute path:')
#       for i in range(1, 10):
#               host = '192.168.0.' + str(i)
#               print host
        host = 'yourIP'
        user = 'youruser'
        password = 'yourpassword'
        files = '/tmp/1234.tar.gz'
        child = ssh_command(files, user, host, password)
        child.expect(pexpect.EOF)
        print child.before

if __name__ == '__main__':
        try:
                main()
                os.system('sh /bash/Cacti_html.sh')   # 註銷掉(沒有特殊要求的)
        except Exception, e:
                print str(e)
                #traceback.print_exc()
                os._exit(1)

------------------------------------------------------------------------------------------------
cat Cacti_html.sh
#! /bin/bash
# 2013-1-23
# 你們沒有用，但是還是寫出來參考
Tar_Dir="/tmp/123"
Dir="/data/www/cacti"

rm -rf /data/www/cacti/1234/
cd $Tar_Dir
tar zxf 1234.tar.gz -C $Dir
rm -rf /tmp/123/1234.tar.gz

備註：注意os.system是調用的shell腳本，沒有特殊需要，不用這個，在這裏是用Cacti_html.sh做了一個處理。
run:   python2.6 script-name