需求背景：

近日，在安裝某軟件過程，發(fā)現(xiàn)在安裝過程需要輸入一些信息才能繼續(xù)下一步操作，在機(jī)器數(shù)量較少情況下，我們可以單臺登錄上去完成安裝操作，但當(dāng)機(jī)器數(shù)量超過一定時(shí)，如果再手動(dòng)登錄操作，就會(huì)產(chǎn)生大量重復(fù)性操作，既不能帶來有效學(xué)習(xí)能力提升，同時(shí)也會(huì)極大產(chǎn)生不確定性，引發(fā)工作效率下降，那么如何自動(dòng)化完成某些操作呢，尤其是帶有交互功能的步驟呢，例如需要輸入賬號密碼？

1. EOF 多文本輸入

需求案例 1

新交付了一批機(jī)器，每臺機(jī)器只分配了一塊落盤 ，現(xiàn)在根據(jù)需求對該盤進(jìn)行分區(qū)并實(shí)現(xiàn)掛載，如何實(shí)現(xiàn)？

需求分析：

對于一個(gè)盤，實(shí)現(xiàn)分區(qū)掛載到不同目錄，通常思路有兩條：

方法一： 將整塊盤作為一個(gè)PV ，整合成VG卷，再根據(jù)劃分不同LV卷大小分給不同目錄方法二： 通過fdisk 將盤直接分割成對應(yīng)需求的大小，再對磁盤初始化，完成掛載

方案解決

這里我們?yōu)榱搜菔窘换スδ?，選擇方法二，實(shí)現(xiàn)腳本如下：

#!/bin/bash
fdisk /dev/sdb <<EOF
n
p
1
wq
EOF
 
 mkfs.xfs /dev/sdb1 &&  mkdir -p /data && mount /dev/sdb1 /data
echo '/dev/sdb1 /data xfs defaults 0 2' >> /etc/fstab

分析上述腳本，我們發(fā)現(xiàn)使用了 一個(gè)關(guān)鍵字 EOF

• EOF是END Of File的縮寫,表示自定義終止符.既然自定義,那么EOF就不是固定的,可以隨意設(shè)置別名,在linux按ctrl-d就代表EOF.
• EOF一般會(huì)配合cat能夠多行文本輸出.

其用法如下:

<<EOF //開始

.... //需要輸入的內(nèi)容

EOF //結(jié)束

例如使用 cat、<<、EOF和>以交互方式編寫bash腳本，如下所示。

cat << EOF > script.sh
#!/bin/bash
 
printf "Hello\n"
printf "Wordl!\n"
EOF

合理 利用這三個(gè)，即可以完成對應(yīng)多文本交互輸入，例如修改用戶密碼，正常情況下，需要連續(xù)輸入兩次密碼，兩次密碼一致才能修改成功，如下：

上面我們學(xué)會(huì)了EOF 這個(gè)關(guān)鍵字，那么我們試試通過它來修改密碼。腳本如下：

#!/bin/bash
 
cat << EOF| passwd 
新密碼
新密碼，與上述需一致
EOF
 
# or 不使用管道符
 
passwd << EOF
新密碼
新密碼，與上述需一致
EOF

實(shí)戰(zhàn)結(jié)果，成功修改密碼：

2. Expect 自動(dòng)交互

需求案例 2

  新交付了一批機(jī)器，需要給每臺機(jī)器分發(fā)文件，如何實(shí)現(xiàn)？

需求分析：

遠(yuǎn)程拷貝文件常用密令是scp 或者 rsync ,但是在給每臺機(jī)器傳輸時(shí)需要若輸入密碼，有的機(jī)器可能還需要輸入YES，錄入機(jī)器指紋信息，如下：

Expect 是在tcl基礎(chǔ)上的一個(gè)自動(dòng)化交互套件, 在一些需要交互輸入指令的場景下, 可通過腳本設(shè)置自動(dòng)進(jìn)行交互通信. 其交互流程主要有以下5步:

0 定義變量

1 spawn啟動(dòng)指定腳本或命令

2 expect匹配結(jié)果關(guān)鍵詞

3 send針對指定關(guān)鍵詞發(fā)送指定指令

4 執(zhí)行完成, 退出

但可惜的是os默認(rèn)沒有安裝，因此需要先安裝才能使用

Expect is a tcl application for automating and testing interactive applications such as telnet, ftp, passwd, fsck, rlogin, tip, etc. Expect makes it easy for a script to control another program and interact with it.

方案解決：

1. 先檢查本機(jī)是否安裝了expect ,如果沒有安裝，需要手動(dòng)安裝

# 檢查是否安裝了expect:
[root@localhost ~]# rpm -qi expect
Name        : expect
Version     : 5.45
Release     : 14.el7_1
Architecture: x86_64
Install Date: Fri 05 Aug 2022 07:26:04 AM CST
Group       : Development/Languages
....
 
# 如果沒有安裝, 使用yum安裝expect ,通常會(huì)順帶把依賴包tcl 也安裝了:
[root@localhost ~]# yum install -y expect  # -y  其實(shí)也是安裝過程中一個(gè)交互，發(fā)現(xiàn)沒，只是作為參數(shù)傳入了
[root@localhost ~]# yum install -y tcl  # 如果上述命令提示已安裝tcl了，此步可以忽略
 
# 查看expect的安裝路徑:
[root@localhost ~]#  which expect
/usr/bin/expect

2 .對應(yīng)功能腳本開發(fā)，本案例腳本參考如下：

[root@test01 ~]# cat scp.exp
#! /usr/bin/expect
set file [lindex $argv 0]
set file2 [lindex $argv 1]
spawn scp -rp $file $file2 root@192.168.31.89:/tmp
expect {
 "(yes/no)" {send "yes\r";exp_continue}
 "*password:*" {send "Password\r"}
 
}
expect eof
exit -onexit {
 send_user "bye \n"
}

3. 分析上述腳本，有幾個(gè)點(diǎn)需要說明

#!/usr/bin/expect

腳本文件的第一行指明expect 安裝位置，具體可以參考2 中命令查看，指明腳本解析器，和Shell類似，表示程序使用Expect解析，這里與一般bash 腳本不同，因此需要注意，通常我們會(huì)將expect腳本后綴修改成exp來和bash 腳本 sh區(qū)別

set 設(shè)置變量值

set file [lindex $argv 0] 將傳入的第一個(gè)參數(shù)賦給file ,類似第二、三個(gè)參數(shù)[lindex $argv 1] [lindex $argv 2] 等，后續(xù)調(diào)用時(shí)使用 $file ，和shell 一樣。特殊參數(shù)：

$argc表示傳參的個(gè)數(shù)，$argv0表示腳本的名字

spawn 表名要執(zhí)行的腳本或程序命令，如ssh、scp等

格式： spawn [選項(xiàng)] [需要自動(dòng)交互的命令或程序]

例如：spawn scp -rp $file $file2 root@192.168.31.89:/tmp #＜==執(zhí)行scp命令（注意開頭必須要有spawn， 否則無法實(shí)現(xiàn)交互）

expect

需和spawn 配合使用 ，表示匹配spawn指定的腳本或命令的輸出結(jié)果，如果與expect后面的字符串匹配，就執(zhí)行下面的send命令，表示對結(jié)果響應(yīng)反饋

有時(shí)命令的輸出提示信息有可能會(huì)變化，所以可以在expect中使用模糊匹配，比如*。

注意：匹配的動(dòng)作也可以放在下一行，這樣就不需要使用{}（大括號）了

send

在expect命令匹配指定的字符串后，發(fā)送指定的字符串給系統(tǒng)，這些命令可以支持一些特殊轉(zhuǎn)義符號，例如：\r表示回車、\n表示換行、\t表示制表符等

exp_continue

從命令的拼寫就可以看出命令的作用，即讓Expect程序繼續(xù)匹配的意思，如果需要一次匹配多個(gè)字符串，那么不同的匹配之間就要加上exp_continue，否則expect將不會(huì)自動(dòng)輸入指定的字符串。最后一個(gè)的結(jié)尾就不需要加上exp_continue了，因?yàn)榍懊娑家淹瓿闪?，它是最后一個(gè)啦

exit

功能類似于Shell中的exit，即直接退出腳本，還可以利用這個(gè)命令對腳本做一些關(guān)閉前提示等工作

send_user

打印Expect腳本信息，類似Shell里的echo. 例如打印變量信息，驗(yàn)證數(shù)據(jù)傳入是否正常

在掌握expect 基本使用方式后，我們寫一個(gè)批量查看機(jī)器負(fù)載信息的小腳本，加強(qiáng)記憶

#! /usr/bin/expect
    set time 30
    set ip [lindex $argv 0]
 
    spawn ssh root@$ip uptime
    expect {
        "*yes/no" { send "yes\r"; exp_continue }
        "*password:" { send "$password\r" }
    }
    expect eof

實(shí)戰(zhàn)結(jié)果：

小試牛刀

在學(xué)習(xí)完以上兩個(gè)方法，我們試著寫一個(gè)腳本，結(jié)合上述兩種方式，批量查看各機(jī)器目錄掛載情況，并列舉出來，參考腳本如下：

#!/bin/bash
 
ip="192.168.31.89"
username="root"
password="123456"
cmd=" df -PTh|grep ^/dev"
# 指定執(zhí)行引擎
expect <<EOF
    set time 30
    spawn ssh $username@$ip  $cmd
    expect {
        "*yes/no" { send "yes\r"; exp_continue }
        "*password:" { send "$password\r" }
    }
    expect eof
EOF

以上兩種方式就是日常在Linux Shell 腳本常用來做自動(dòng)化部署，解決腳本執(zhí)行過程參數(shù)交互問題，尤其expect 配合一些shell腳本執(zhí)行，十分便捷。除了以上使用方式，在企業(yè)生產(chǎn)中，我們還會(huì)用到ansible 腳本，ansible是新出現(xiàn)的自動(dòng)化運(yùn)維工具，基于Python開發(fā)，通過調(diào)用其模塊，實(shí)現(xiàn)批量系統(tǒng)配置、批量程序部署、批量運(yùn)行命令等功能，有興趣的小伙伴可以試試學(xué)習(xí)下。

到此這篇關(guān)于Linux Shell 自動(dòng)交互功能實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Linux Shell 自動(dòng)交互內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論