python解决ssh记不住ip的问题

hysaoh · 发表于 2023-1-11 21:45:26

本帖最后由 hysaoh 于 2023-1-11 21:48 编辑

楼主有好几台服务器，由于服务器过多导致记住服务器ip成了问题，于是写了这个脚本，能够更好的进行快乐的ssh。代码如下。
请注意，需要安装rich第三方库，其他均为自带库。
在我的windows terminal中效果如图

# -*- coding: utf-8 -*-
"""
@Author         ：Cat
@Date           : 2023年 01月 10日
@Introduction   ：A Lazy Cat
"""
import shlex
import subprocess
from rich.console import Console
from rich.table import Table

console = Console()

table = Table(title="服务器列表")
table.add_column("序号", style="cyan")
table.add_column("服务器", style="cyan")
table.add_column("IP", style="cyan")

serverList = {
    "阿里云服务器": "x.x.x.x",
    "腾讯云服务器": "x.x.x.x",
    "本地虚拟机：AlmaLinux": "192.168.197.129",
    "本地虚拟机：CentOS7网络络自动运维": "192.168.197.191"
}
keys = list(serverList.keys())
for i in range(len(keys)):
    table.add_row(str(i), keys[i], serverList[keys[i]])
console.print(table)
console.print("请选择要连接的服务器序号:", style="green", end="")
index = int(input())
console.print("正在连接服务器：" + keys[index], style="green")
cmd = "ssh root@" + serverList[keys[index]]
res = subprocess.Popen(shlex.split(cmd), shell=True).communicate()[1]