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()