python实现ssh登录send_python: 用EXPECT 实现 ssh 登录（1） | 学步园

#! /usr/bin/env python

# filename : remove_loadbalance_members.py

import pexpect

import sys

import logging

user = sys.argv[1]

host = sys.argv[2]

password = sys.argv[3]

poolName = sys.argv[4]

members = sys.argv[5]

logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s ', level=logging.INFO)

logging.Formatter('%(asctime)s %(levelname)s %(message)s ')

ssh_newkey = 'Are you sure you want to continue connecting (yes/no)?'

child = pexpect.spawn('ssh -l %s %s '%(user, host))

i = child.expect([pexpect.TIMEOUT, ssh_newkey, 'assword: '])

if i == 0:  # Timeout

errorString = 'Wrong Username or Password when logging : '+ host +', '+user+'/'+password

logging.error(errorString)

print "before >> ",child.before

if i == 1:  # SSH does not have the public key. Just accept it.

child.sendline ('yes')

child.expect ('assword: ')

i = child.expect([pexpect.TIMEOUT, 'assword: '])

if i == 0:  # Timeout

errorString = 'Wrong Username or Password when logging : '+ host +', '+user+'/'+password

logging.error(errorString)

child.sendline(password)

n = child.expect(['#', pexpect.TIMEOUT, pexpect.EOF])

if n != 0 :

errorString = 'Wrong Username or Password when logging : '+ host +', '+user+'/'+password

logging.error(errorString)

logging.info('Login '+host+' successfully !')

# 1. Create the LoadBalance

logging.info('Strat remove loadbalance members .')

command = 'b pool '+poolName+' member '+members+' delete '

child.sendline(command)

n = child.expect(['#', pexpect.TIMEOUT, pexpect.EOF])

if n != 0 :

errorString = 'Remove a loadbalance members poolName = '+poolName+' member = '+members+' failed .'

logging.error(errorString)

print "before >> ",child.before

logging.info('Remove a loadbalance members poolName = '+poolName+' member = '+members+'  successfully !')

logging.info('Quit SSH to send command Line : exit')

child.sendline('exit')

#child.expect([pexpect.EOF])