如何检查代码是在本地运行还是在Python中的群集上运行

Question

我有一个可以在本地计算机和计算群集上运行的python脚本。我正在寻找一种有效的方法来检查脚本，无论它是否在本地运行。最好的解决方案是什么？ 我曾考虑过传递一些硬件信息并检查它是否已更改，但是肯定有比这更优雅的解决方案。

Answer 1

您可以使用os模块

检查正在运行的计算机的主机名。

 import os
 os.system('hostname')

这会将主机名输出到控制台。

如果要捕获输出，请像这样使用.popen：

 a=os.popen('hostname').read()

这将返回一个字符串，您可能必须将其编码为utf-8以使其可读，如下所示：

 print(a.encode('utf-8'))

但这最后一点仅适用于您的操作系统语言带有特殊字符的情况

Answer 2

对于这样的配置，您可以在集群/本地计算机上设置一个环境变量，然后让脚本检查该变量以确定其位置：

babel macros