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