python使用argparse 、paramiko实现服务器管理器

使用argparse，paramiko两个包去实现简易的服务器管理器，完成两种方式的连接（ 密码和密钥 ），以及命令行交互，文件上传下载。

相比sys.argv的方式去判断传入的参数，如果参数较多那么argparse包更易维护和修改，远程控制模块paramiko可以很轻易的实现远控的功能

注意：paramiko中的SSHClient对象的exec_command在执行cd命令的时候没有效果，需要和其它命令一起执行，

　　　　例如：stdin, stdout, stderr = client.exec_command('cd test; ls')

话不多说，直接上代码：

 

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Riy


import logging
import argparse
import paramiko


logging.basicConfig(
    level=logging.DEBUG,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    filename='SSHlog.log',
    filemode='a'
)


class RiySSH:
    def ssh_parse(self):
        '''设置参数'''
        parser = argparse.ArgumentParser("Riy command line ssh server operate 服务器操作命令行工具...")
        parser.add_argument('-H', '--host', dest="HOST", action="store", help="要连接的主机")
        parser.add_argument('-P', '--port', dest="PORT", action="store", help="连接主机的端口")
        parser.add_argument('-u', '--username', dest="USER", action="store", help="连接的用户名")
        parser.add_argument('-p', '--password', dest="PWD", action="store", help="使用密码进行认证")
        parser.add_argument('-s', '--ssh', dest="SSH", action="store", help="使用密钥文件进行认证")
        parser.add_argument('--shell', dest="SHELL", action="store_true", help="进入命令交互模式")
        parser.add_argument('-U', '--upload', dest="UPLOAD", nargs="*", help="上传文件，需要追加文件名")
        parser.add_argument('-D', '--download', dest="DOWNLOAD", nargs="*", help="下载文件，需要追加文件名")
        parser.print_help()
        args = parser.parse_args()
        return args

    def connect_password(self, HOST, PORT, USER ,PWD=None, SSH=None):
        '''进行服务器连接'''
        client = paramiko.SSHClient()
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        private = paramiko.RSAKey.from_private_key_file(SSH)
        client.connect(hostname=HOST, port=PORT, username=USER, password=PWD, pkey=private)
        return client

    def connect_cmd(self, client):
        '''命令行交互'''
        while True:
            print('请输入shell命令，按q键退出：')
            CMD = input('>>')
            stdin, stdout, stderr = client.exec_command(CMD)
            print(stdout.read().decode('utf-8'))
            if CMD == 'q':
                break

    def download_file(self, client, remote_file_path, local_file_path):
        '''下载文件'''
        ftp_client = client.open_sftp()
        ftp_client.get(remote_file_path, local_file_path)

    def upload_file(self, client, local_file_path, remote_file_path):
        '''上传文件'''
        ftp_client = client.open_sftp()
        ftp_client.put(local_file_path, remote_file_path)


def main():
    a = RiySSH()
    args = a.ssh_parse()
    if args.PWD or args.SSH:
        HOST = args.HOST
        PORT = args.PORT
        USER = args.USER
        PWD = args.PWD
        SSH = args.SSH
        client = a.connect_password(HOST, PORT, USER, PWD, SSH)
        if args.UPLOAD:
            a.upload_file(client, args.UPLOAD[0], args.UPLOAD[1])
        if args.DOWNLOAD:
            a.download_file(client, args.DOWNLOAD[0], args.DOWNLOAD[1])
        if args.SHELL:
            a.connect_cmd(client)
        client.close()


if __name__ == '__main__':
    main()