python - 终止当前的 ssh 连接

Question

我的机器上有一个 python 脚本。

我可以从 ssh 连接 和机器的控制台 运行它。

由于脚本更改了一些 IP 配置内容，我想在更改 IP 之前断开 ssh - 这样 ssh 不会挂起，并且会在 IP 更改之前正确关闭。

所以 - 在 python 中有没有办法检查脚本是否从 ssh 运行？如果是这样关闭ssh？

Answer 1

我没有 Linux 机器，但在 Mac 上进行了测试。如果您发出who am i 命令并看到一个IP 地址，那么您是在一个ssh 连接上。如果没有，您在控制台。此答案基于以下answer。

在控制台上：

$ who am i
hai      ttys000  Dec 23 13:00

在 ssh 连接上：

$ who am i
hai      ttys001  Jan  4 07:41  (192.168.1.19)

有了这些信息，就可以从 Python 运行命令并解析输出了。

更新

另一种方法是检查环境变量 SSH_CLIENT 或 SSH_TTY 是否存在，根据此处的answer。