使用 Python 与远程计算机交互

Question

我刚刚成为我的研究小组集群的系统管理员，在这方面，我还是一个新手。我正在尝试制作一些工具来监控网络，并需要帮助开始使用 python（我的母语）实现它们。

例如，我想查看谁登录到远程计算机。手动地，我会 ssh 和 who，但是我如何将此信息放入脚本中进行操作？类似的，

import remote_info as ri
ri.open("foo05.bar.edu")
ri.who()

Out[1]: 
hutchinson tty7         2009-08-19 13:32 (:0)
hutchinson pts/1        2009-08-19 13:33 (:0.0)

类似于 cat /proc/cpuinfo 之类的东西来获取节点的处理器信息。一个起点真的很棒。谢谢。

Answer 1

这是一个简单、廉价的解决方案，可以帮助您开始

from subprocess import *
p = Popen('ssh servername who', shell=True, stdout=PIPE)
p.wait()
print p.stdout.readlines()

返回（例如）

['usr      pts/0        2009-08-19 16:03 (kakapo)\n',
 'usr      pts/1        2009-08-17 15:51 (kakapo)\n',
 'usr      pts/5        2009-08-17 17:00 (kakapo)\n']

和 cpuinfo：

p = Popen('ssh servername cat /proc/cpuinfo', shell=True, stdout=PIPE)

Answer 2

我一直在使用 Pexpect，它可以让你通过 ssh 进入机器，发送命令，读取输出，并对其做出反应，并取得成功。我什至围绕它启动了一个开源项目 Proxpect - 该项目已经很久没有更新了，但我离题了……

Answer 3

pexpect 模块可以帮助您与 ssh 交互。或多或少，您的示例如下所示。

child = pexpect.spawn('ssh servername')
child.expect('Password:')
child.sendline('ABCDEF')
(output,status) = child.sendline('who')

Answer 4

如果您的需求超出了简单的“ssh remote-host.example.org who”，那么有一个很棒的 Python 库，名为 RPyC。它具有所谓的“经典”模式，允许使用几行代码通过网络几乎透明地执行 Python 代码。对于可信环境来说非常有用的工具。

以下是来自 Wikipedia 的示例：

import rpyc
# assuming a classic server is running on 'hostname'
conn = rpyc.classic.connect("hostname")

# runs os.listdir() and os.stat() remotely, printing results locally
def remote_ls(path):
    ros = conn.modules.os
    for filename in ros.listdir(path):
        stats = ros.stat(ros.path.join(path, filename))
        print "%d\t%d\t%s" % (stats.st_size, stats.st_uid, filename)

remote_ls("/usr/bin")

如果您有兴趣，可以他们的 wiki 上有一个很好的教程。

但是，当然，如果您完全可以使用 Popen 进行 ssh 调用，或者只是不想运行单独的“RPyC”守护进程，那么这绝对是一种矫枉过正。

Answer 5

这涵盖了基础。请注意 sudo 用于需要更多权限的事情。我们将 sudo 配置为允许该用户执行这些命令，而无需输入密码。

另外，请记住，您应该运行 ssh-agent 才能使其“有意义”。但总而言之，它的效果确实很好。运行 deploy-control httpd configtest 将检查所有远程服务器上的 apache 配置。

#!/usr/local/bin/python

import subprocess
import sys

# The user@host: for the SourceURLs (NO TRAILING SLASH)
RemoteUsers = [
        "[email protected]",
        "[email protected]",
        ]

###################################################################################################
# Global Variables
Arg                             = None


# Implicitly verified below in if/else
Command = tuple(sys.argv[1:])

ResultList = []
###################################################################################################
for UH in RemoteUsers:
        print "-"*80
        print "Running %s command on: %s" % (Command, UH)

        #----------------------------------------------------------------------------------------------
        if Command == ('httpd', 'configtest'):
                CommandResult = subprocess.call(('ssh', UH, 'sudo /sbin/service httpd configtest'))

        #----------------------------------------------------------------------------------------------
        elif Command == ('httpd', 'graceful'):
                CommandResult = subprocess.call(('ssh', UH, 'sudo /sbin/service httpd graceful'))

        #----------------------------------------------------------------------------------------------
        elif Command == ('httpd', 'status'):
                CommandResult = subprocess.call(('ssh', UH, 'sudo /sbin/service httpd status'))

        #----------------------------------------------------------------------------------------------
        elif Command == ('disk', 'usage'):
                CommandResult = subprocess.call(('ssh', UH, 'df -h'))

        #----------------------------------------------------------------------------------------------
        elif Command == ('uptime',):
                CommandResult = subprocess.call(('ssh', UH, 'uptime'))

        #----------------------------------------------------------------------------------------------
        else:
                print
                print "#"*80
                print
                print "Error: invalid command"
                print
                HelpAndExit()

        #----------------------------------------------------------------------------------------------
        ResultList.append(CommandResult)
        print


###################################################################################################
if any(ResultList):
        print "#"*80
        print "#"*80
        print "#"*80
        print
        print "ERRORS FOUND.  SEE ABOVE"
        print
        sys.exit(0)

else:
        print "-"*80
        print
        print "Looks OK!"
        print
        sys.exit(1)

Answer 6

Fabric 是一种自动执行一些简单任务的简单方法，我当前使用的版本允许您像这样包装命令：

run('whoami', fail='ignore')

您可以为您需要的每台机器指定配置选项（config.fab_user、config.fab_password）（如果您想要自动处理用户名密码）。

有关 Fabric 的更多信息，请访问：

http://www.nongnu.org/fab/

新版本更Pythonic - 我不确定这是否对你来说更好......目前对我来说效果很好......