ssh-test.py

import pexpect
PROMPT = ['#', '>>>', '> ','\$']
def send_command(child, cmd):
	child.sendline(cmd)
	child.expect(PROMPT)
	print child.before

def connect(user, host, password):
	ssh_newkey = 'Are you sure you want to continue connecting'
	connStr = 'ssh ' + user + '@' + host
	child = pexpect.spawn(connStr)
	ret = child.expect([pexpect.TIMEOUT, ssh_newkey, '[P|p]assword:'])
	if ret == 0 :
		print '[-] Error Connecting'
		return
	if ret == 1:
		child.sendline('yes')
		ret = child.expect([pexpect.TIMEOUT, '[P|p]assword:'])
	if ret == 0 :
		print '[-] Error Connecting'
		return
	child.sendline(password)
	child.expect(PROMPT)
	return child

def main():
	host = 'localhost'
	user = 'root'
	password = 'toor'
	child = connect(user, host, password)
	send_command(child, 'cat /etc/shadow | grep root')

if __name__ == '__main__':
	main()