www.cnblogs.com/ruiyqinrui

开源、架构、Linux C/C++/python AI BI 运维开发自动化运维。 春风桃李花 秋雨梧桐叶。“力尽不知热 但惜夏日长”。夏不惜,秋不获。@ruiY--秦瑞

python爬虫,C编程,嵌入式开发.hadoop大数据,桉树,onenebula云计算架构.linux运维及驱动开发.

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理
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

 

posted on 2022-11-21 08:55  秦瑞It行程实录  阅读(48)  评论(0编辑  收藏  举报
www.cnblogs.com/ruiyqinrui