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