python+expect实现脚本自动登录远程服务器
起因
最近在折腾Ubuntu系统,有时候需要连接远程服务器,使用 ssh 命令连接较为复杂,需要记住每台机器的 ip 和 密码,当然 Ubuntu(linux) 下也有类似于 xshell 一样的管理工具,例如:finalShell,但是在命令行下没有方便管理的工具,因此打算自己实现一个简单的

选型
自动交互:
Linux环境下有 expect 命令允许通过自动接收命令输出并根据不同输出自动输入对应内容,满足我们自动输入密码、yes等命令的需要

操作交互:

shell读取文件并输入
python读取文件并输出
shell操作json较为复杂,如果使用普通空格等来作为分隔符,shell也需要复杂的切割逻辑,且 Linux 系统一般原生支持python,python可以很方便的操作文件并输出,因此选择python来交互

实现
安装软件
# 更新软件包
apt-get update
apt-get upgrade
# 安装expect
apt-get install expect
# 安装python3
apt-get install python3
服务器配置文件
# 创建目录
mkdir /usr/local/login
# 创建文件
touch /usr/local/login/server.json
/usr/local/login/server.json

[
{
"host": "服务器IP或域名",
"user": "服务器账号",
"password": "服务器密码",
"remark": "服务器说明"
}
]
expect登录脚本
touch /usr/local/login/ssh-except.exp
/usr/local/login/ssh-except.exp

#!/bin/expect
set ipaddr [lindex $argv 0]
set user [lindex $argv 1]
set passwd [lindex $argv 2]
set timeout 30
spawn ssh $user@$ipaddr
expect {
"yes" {send "yes\r";exp_continue}
"password" {send "$passwd\r"}
}
expect eof
用户交互脚本
# 创建登录脚本
touch /usr/local/bin/login
# 设置脚本可执行权限
chmod +x /usr/local/bin/login
/usr/local/bin/login

#!/usr/bin/python3
import json
import os

server_list=[]
with open("/usr/local/login/server.json", "r") as file:
server_list = json.load(file)
# 打印主界面
while True:
print("\n\n=====================================")
print("序号\t主机\t\t说明")
i = 1
print("0\t退出")
for server in server_list:
print(str(i)+"\t"+server['host']+"\t"+server['remark'])
i=i+1
print("=====================================")
index=input("请选择服务器编号:")
try:
server_index=int(index)
if server_index == 0:
break
if server_index > len(server_list) or server_index < 0:
print("编号不合法")
continue
server = server_list[server_index-1]
os.system("expect /usr/local/login/ssh-except.exp " + server['host'] + " "+ server['user'] +" "+ server['password'] +" ")
except Exception as e:
print("编号不合法")
continue