Выполнять команды по ssh с Python

Question

Я пишу скрипт для автоматизации некоторых команд командной строки в Python. В настоящий момент я звоню так:

cmd = "some unix command" 
retcode = subprocess.call(cmd,shell=True) 

Однако мне нужно выполнить некоторые команды на удаленной машине. Вручную, я должен войти в систему, используя ssh, а затем запустите команды. Как я могу автоматизировать это в Python? Мне нужно войти с помощью (известного) пароля на удаленный компьютер, поэтому я не могу просто использовать cmd = ssh user@remotehost, мне интересно, есть ли модуль, который я должен использовать?

Answer 1

Я отсылаю вас к paramiko

см this question

ssh = paramiko.SSHClient() 
ssh.connect(server, username=username, password=password) 
ssh_stdin, ssh_stdout, ssh_stderr = ssh.exec_command(cmd_to_execute) 

Answer 2

Вы имели в виду Fabric? Это позволяет вам делать всевозможные удаленные вещи поверх SSH с помощью python.

Answer 3

Или вы можете просто использовать commands.getstatusoutput:

commands.getstatusoutput("ssh machine 1 'your script'") 

Я широко использовали его, и он прекрасно работает.

В Python 2.6+ используйте subprocess.check_output.

Answer 4

Я использовал paramiko сгусток (хороший) и pxssh (также приятно). Я бы тоже порекомендовал. Они работают немного по-другому, но имеют относительно большое совпадение в использовании.

Answer 5

Я нашел парамико слишком низкоуровневым, а Fabric не особенно хорошо подходит для использования в качестве библиотеки, поэтому я собрал свою собственную библиотеку под названием spur, которая использует paramiko для реализации немного более приятного интерфейса:

import spur 

shell = spur.SshShell(hostname="localhost", username="bob", password="password1") 
result = shell.run(["echo", "-n", "hello"]) 
print result.output # prints hello 

Если вам нужно запустить внутри оболочки:

shell.run(["sh", "-c", "echo -n hello"]) 

Answer 6

Все уже говорили (рекомендуется) с использованием paramiko и я просто делюсь код питона (API, можно сказать), что позволит Y ou выполнить несколько команд за один раз.

для выполнения команд на различного назначения узла: Commands().run_cmd(host_ip, list_of_commands)

Вы увидите один TODO, который я держал, чтобы остановить выполнение, если какой-либо из команд не в состоянии выполнить, я не знаю, как это сделать Это. пожалуйста, поделитесь своими знаниями

#!/usr/bin/python 

import os 
import sys 
import select 
import paramiko 
import time 


class Commands: 
    def __init__(self, retry_time=0): 
     self.retry_time = retry_time 
     pass 

    def run_cmd(self, host_ip, cmd_list): 
     i = 0 
     while True: 
     # print("Trying to connect to %s (%i/%i)" % (self.host, i, self.retry_time)) 
     try: 
      ssh = paramiko.SSHClient() 
      ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
      ssh.connect(host_ip) 
      break 
     except paramiko.AuthenticationException: 
      print("Authentication failed when connecting to %s" % host_ip) 
      sys.exit(1) 
     except: 
      print("Could not SSH to %s, waiting for it to start" % host_ip) 
      i += 1 
      time.sleep(2) 

     # If we could not connect within time limit 
     if i >= self.retry_time: 
      print("Could not connect to %s. Giving up" % host_ip) 
      sys.exit(1) 
     # After connection is successful 
     # Send the command 
     for command in cmd_list: 
      # print command 
      print "> " + command 
      # execute commands 
      stdin, stdout, stderr = ssh.exec_command(command) 
      # TODO() : if an error is thrown, stop further rules and revert back changes 
      # Wait for the command to terminate 
      while not stdout.channel.exit_status_ready(): 
       # Only print data if there is data to read in the channel 
       if stdout.channel.recv_ready(): 
        rl, wl, xl = select.select([ stdout.channel ], [ ], [ ], 0.0) 
        if len(rl) > 0: 
         tmp = stdout.channel.recv(1024) 
         output = tmp.decode() 
         print output 

     # Close SSH connection 
     ssh.close() 
     return 

def main(args=None): 
    if args is None: 
     print "arguments expected" 
    else: 
     # args = {'<ip_address>', <list_of_commands>} 
     mytest = Commands() 
     mytest.run_cmd(host_ip=args[0], cmd_list=args[1]) 
    return 


if __name__ == "__main__": 
    main(sys.argv[1:]) 

Спасибо!

Answer 7

Я рекомендую использовать ssh_decorate

Как просто:

from ssh_decorate import ssh_connect 
ssh=ssh_connect('user','password','server') 

#Run a python function 
@ssh 
def python_pwd(): 
    import os 
    return os.getcwd() 

print (python_pwd()) 

#Run a bash command 
ssh>>"ls"