python paramiko

利用上班休息时间写了一个小程序,利用 paramiko 实现单次的连接服务器--->执行命令(待完善的功能,应该是一次连接批量执行 command,并且不支持实时命令返回结果, 后续进行完善,在此做此记录)

#!/usr/bin/env python
# _*_coding:utf-8_*_
#  author:  'Edward.Liu'
# dateTime:  '15/12/4'
#   motto:  'Good memory as bad written'

import paramiko
import getpass


def ssh_clients(hostname=None, port=22, username=None, password=None, command=None, nbytes=4096):
    client = paramiko.Transport((hostname, port))
    client.connect(username=username, password=password)

    stdout_data = []
    stderr_data = []
    session = client.open_channel(kind='session')
    session.exec_command(command)
    while True:
        if session.recv_ready():
            stdout_data.append(session.recv(nbytes))
        if session.recv_stderr_ready():
            stderr_data.append(session.recv_stderr(nbytes))
        if session.exit_status_ready():
            break

    print '33[32mExit status is: 33[0m', session.recv_exit_status()
    print "33[31m*********************33[0m"
    print ''.join(stdout_data)
    print ''.join(stderr_data)
    session.close()
    client.close()


if __name__ == '__main__':
    host_list = ['x.x.x.x', 'x.x.x.x', 'x.x.x.x']#服务器地址列表
    print "33[31m     **********IPaddress List**********      33[0m"
    for index, value in enumerate(host_list):
        print index, value
    chose_ip = raw_input("33[32mPlease Connection Host IP Or Host Name:33[0m").strip()
    if chose_ip.isdigit():
        chose_ip = int(chose_ip)
        hostname = host_list[chose_ip]
    username = raw_input("33[32mPlease Input Will Connection Host User:33[0m").strip()
    password = getpass.getpass("33[32mEnter You Password:33[0m").strip()
    if len(username) == 0 and len(password) == 0:
        username = 'user' #远程服务器用户,此处为默认用户适用一组服务器
        password = 'password' #远程服务器密码
    command = raw_input("33[32mPlease Input Will Running Command:33[0m").strip()
    ssh_clients(hostname=hostname, username=username, password=password, command=command)
    print "33[32m--------------------SSH_Connection Info--------------------33[0m"
    print "33[32m|    **********SSH_Connection-->Host:%s33[0m" % hostname
    print "33[32m|    **********SSH_Connection-->User:%s33[0m" % username
    print "33[32m|    **********SSH_Connection-->Passwd:%s33[0m" % password
    print "33[32m|    **********SSH_Connection-->RunCMD:%s33[0m" % command
    print "33[32m--------------------SSH_Connection END--------------------33[0m"