Criando Scripts SSH com Python

# -*- coding: utf-8 -*-
'''
Criando Scripts SSH com Python

Exemplo de script Python que executa comandos utilizando uma conexão SSH

Artigo: https://www.linkedin.com/pulse/criando-scripts-ssh-com-phyton-diego-mendes-rodrigues/

Diego Mendes Rodrigues
'''

# Importando do Paramiko
import paramiko

# Dados de acesso
ip = '127.0.0.1'    # IP do Servidor Linux
usuario = 'root'        # Usuário de acesso
senha = '123456'  # Senha de acesso

# Criando a conexão SSH
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(ip, username=usuario, password=senha)

# Função com o Menu
def print_menu():
    print(30 * "-" , "OPÇÕES" , 30 * "-")
    print("1. apt-get update")
    print("2. apt-get upgrade")
    print("3. Arquivo /etc/passwd")
    print("4. Arquivo /etc/shadow")
    print("5. Espaço em disco (df -h)")
    print("6. Usuários logados no sistema")
    print("7. Informações do sistema")
    print("8. Exit")
    print(67 * "-")

loop = True

while loop:
    print_menu()
    opcao = input("Selecione sua opção [1-8]: ")

    if opcao=='1':
        # Executando o UPDATE
        print('\n----------------------------------\nExecutando o apt-get update\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('apt-get update\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='2':
        # Executando o UPGRADE
        print('\n----------------------------------\nExecutando o apt-get upgrade\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('apt-get upgrade\n')
        stdin.write('y\n')
        stdin.flush()
        print(*stdout.readlines(), sep='')

    elif opcao=='3':
        print('\n----------------------------------\nArquivo /etc/passwd\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('cat /etc/passwd\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='4':
        print('\n----------------------------------\nArquivo /etc/shadow\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('cat /etc/shadow\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='5':
        print('\n----------------------------------\nEspaço em disco\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('df -h\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='6':
        print('\n----------------------------------\nUsuários logados\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('who\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='7':
        print('\n----------------------------------\nInformações do sistema\n----------------------------------')
        stdin, stdout, stderr = ssh.exec_command('uname -a\n')
        print(*stdout.readlines(), sep='')

    elif opcao=='8':
        loop=False

    else:
        print('\nOpção incorreta. Tente novamente.\n')


# Encerrando a conexão SSH
ssh.close()