今天就跟大家聊聊有關(guān)基于Python的shell自動(dòng)化框架ShutIt怎么用,可能很多人都不太了解,為了讓大家更加了解,小編給大家總結(jié)了以下內(nèi)容,希望大家根據(jù)這篇文章可以有所收獲。
成都創(chuàng)新互聯(lián)公司堅(jiān)持“要么做到,要么別承諾”的工作理念,服務(wù)領(lǐng)域包括:網(wǎng)站建設(shè)、成都網(wǎng)站制作、企業(yè)官網(wǎng)、英文網(wǎng)站、手機(jī)端網(wǎng)站、網(wǎng)站推廣等服務(wù),滿足客戶于互聯(lián)網(wǎng)時(shí)代的云溪網(wǎng)站設(shè)計(jì)、移動(dòng)媒體設(shè)計(jì)的需求,幫助企業(yè)找到有效的互聯(lián)網(wǎng)解決方案。努力成為您成熟可靠的網(wǎng)絡(luò)建設(shè)合作伙伴!
ShutIt是一個(gè)易于使用的基于shell的自動(dòng)化框架。它對基于python的expect庫(pexpect)進(jìn)行了包裝。你可以把它看作是“沒有痛點(diǎn)的expect”。它可以通過pip進(jìn)行安裝。
Hello World
讓我們從最簡單的例子開始吧。創(chuàng)建一個(gè)名為example.py的文件:
import shutit session = shutit.create_session('bash') session.send('echo Hello World', echo=True)
運(yùn)行這個(gè)文件:
python example.py
輸出:
python example.py echo "Hello World" echo "Hello World" Hello World Ians-MacBook-Air.local:ORIGIN_ENV:RhuebR2T#
“send”函數(shù)的***個(gè)參數(shù)是要運(yùn)行的命令。“echo”的參數(shù)將會(huì)輸出到終端上。默認(rèn)情況下,ShutIt是靜默的。
登錄服務(wù)器
如果你要登陸一臺服務(wù)器并執(zhí)行服務(wù)器上的命令。可以將example.py改為:
import shutit session = shutit.create_session('bash') session.login('ssh you@example.com', user='you', password='mypassword') session.send('hostname', echo=True) session.logout()
程序?qū)⒌卿浀竭@臺服務(wù)器上,并輸出主機(jī)名。
hostname hostname example.com example.com:cgoIsdVv:heDa77HB#
顯然,這很不安全!你可以這樣運(yùn)行:
import shutit session = shutit.create_session('bash') password = session.get_input('', ispass=True) session.login('ssh you@example.com', user='you', password=password) session.send('hostname', echo=True) session.logout()
它會(huì)讓你輸入密碼:
Input Secret: hostname hostname example.com example.com:cgoIsdVv:heDa77HB#
同樣的,“l(fā)ogin”方法在登錄后改變了提示符。你給了ShutIt一個(gè)登錄命令,并附帶用戶名和密碼(如果需要的話),然后,ShutIt會(huì)完成剩余的事情。
“l(fā)ogout”負(fù)責(zé)終止“l(fā)ogin”,并向屏幕輸出發(fā)生的任何變化。
登錄到多臺服務(wù)器
假設(shè)你有一個(gè)集群包含兩臺服務(wù)器,并希望同時(shí)登錄到這兩個(gè)服務(wù)器上去。則只需要?jiǎng)?chuàng)建兩個(gè)會(huì)話,并運(yùn)行類似的login和send命令:
import shutit session1 = shutit.create_session('bash') session2 = shutit.create_session('bash') password1 = session1.get_input('Password for server1', ispass=True) password2 = session2.get_input('Password for server2', ispass=True) session1.login('ssh you@one.example.com', user='you', password=password1) session2.login('ssh you@two.example.com', user='you', password=password2) session1.send('hostname', echo=True) session2.send('hostname', echo=True) session1.logout() session2.logout()
將輸出這樣的結(jié)果:
$ python example.py Password for server1 Input Secret: Password for server2 Input Secret: hostname hostname one.example.com one.example.com:Fnh3pyFj:qkrsmUNs# hostname hostname two.example.com two.example.com:Gl2lldEo:D3FavQjA#
實(shí)例:監(jiān)控多臺服務(wù)器
我們可以通過添加一些代碼邏輯來檢查命令的輸出,從而將上述代碼變成一個(gè)簡單的監(jiān)控工具:
import shutit capacity_command="""df / | awk '{print $5}' | tail -1 | sed s/[^0-9]//""" session1 = shutit.create_session('bash') session2 = shutit.create_session('bash') password1 = session.get_input('Password for server1', ispass=True) password2 = session.get_input('Password for server2', ispass=True) session1.login('ssh you@one.example.com', user='you', password=password1) session2.login('ssh you@two.example.com', user='you', password=password2) capacity = session1.send_and_get_output(capacity_command) if int(capacity) < 10: print('RUNNING OUT OF SPACE ON server1!') capacity = session2.send_and_get_output(capacity_command) if int(capacity) < 10: print('RUNNING OUT OF SPACE ON server2!') session1.logout() session2.logout()
在這里,我們用了“sendandget_output”方法來獲取capacity_command命令的輸出。
還有很多更加優(yōu)雅的方法可以完成上面的操作,但這取決于你想要Python有多聰明。
更復(fù)雜的IO – Expecting
假設(shè)你需要跟一個(gè)命令行程序進(jìn)行交互,并且要實(shí)現(xiàn)自動(dòng)化操作。在這里,我們使用telnet來舉一個(gè)簡單的例子:
import shutit session = shutit.create_session('bash') session.send('telnet', expect='elnet>', echo=True) session.send('open google.com 80', expect='scape character', echo=True) session.send('GET /', echo=True, check_exit=False) session.logout()
注意“expect”的參數(shù)。你只需要給出telnet提示符的一個(gè)子集來進(jìn)行匹配。
注意“check_exit”的參數(shù),后面我們會(huì)講到這個(gè)參數(shù)的。上面這段代碼將輸出:
$ python example.py telnet telnet> open google.com 80 Trying 216.58.214.14... Connected to google.com. Escape character is '^]'. GET / HTTP/1.0 302 Found Cache-Control: private Content-Type: text/html; charset=UTF-8 Referrer-Policy: no-referrer Location: http://www.google.co.uk/?gfe_rd=cr&ei=huczWcj3GfTW8gfq0paQDA Content-Length: 261 Date: Sun, 04 Jun 2017 10:57:10 GMT302 Moved 302 Moved
The document has moved here . Connection closed by foreign host.
現(xiàn)在回到“checkexit = false”上來。由于telnet命令會(huì)返回一個(gè)錯(cuò)誤的退出碼(1),我們不想讓腳本執(zhí)行失敗,這里的“checkexit = false”能讓ShutIt知道你并不關(guān)注這個(gè)退出碼。
如果你沒有傳入這個(gè)參數(shù),ShutIt會(huì)給你一個(gè)交互式的提示,如果你有終端接入的話。這被稱為“暫停點(diǎn)”。
暫停點(diǎn)
你可以隨便在什么時(shí)候通過調(diào)用以下方法來設(shè)置一個(gè)“暫停點(diǎn)”。
[...] session.pause_point('This is a pause point') [...]
當(dāng)腳本運(yùn)行到暫停點(diǎn)時(shí),同時(shí)按下“Ctrl”和“]”,則可以讓腳本繼續(xù)執(zhí)行。這對于調(diào)試非常有用:添加一個(gè)暫停點(diǎn),看看周圍,然后繼續(xù)。試試這個(gè):
import shutit session = shutit.create_session('bash') session.pause_point('Have a look around!') session.send('echo "Did you enjoy your pause point?"', echo=True)
程序輸出:
$ python example.py Have a look around! Ians-Air.home:ORIGIN_ENV:I00LA1Mq# bash imiell@Ians-Air:/space/git/shutit ? master + ? CTRL-] caught, continuing with run... 2017-06-05 15:12:33,577 INFO: Sending: exit 2017-06-05 15:12:33,633 INFO: Output (squashed): exitexitIans-Air.home:ORIGIN_ENV:I00LA1Mq# [...] echo "Did you enjoy your pause point?" echo "Did you enjoy your pause point?" Did you enjoy your pause point? Ians-Air.home:ORIGIN_ENV:I00LA1Mq#
更復(fù)雜的IO – Backgrounding
回到我們上面的“監(jiān)控多臺服務(wù)器”的例子上來。想象一下,我們要在每臺服務(wù)器上運(yùn)行一個(gè)長時(shí)間運(yùn)行的任務(wù)。默認(rèn)情況下,ShutIt會(huì)持續(xù)運(yùn)行很長時(shí)間。但是我們可以在后臺運(yùn)行任務(wù)來加快ShutIt的運(yùn)行速度。
在這里,你可以使用簡單的命令“sleep 60”來嘗試一個(gè)例子。
import shutit import time long_command="""sleep 60""" session1 = shutit.create_session('bash') session2 = shutit.create_session('bash') password1 = session1.get_input('Password for server1', ispass=True) password2 = session2.get_input('Password for server2', ispass=True) session1.login('ssh you@one.example.com', user='you', password=password1) session2.login('ssh you@two.example.com', user='you', password=password2) start = time.time() session1.send(long_command, background=True) session2.send(long_command, background=True) print('That took: ' + str(time.time() - start) + ' seconds to fire') session1.wait() session2.wait() print('That took: ' + str(time.time() - start) + ' seconds to complete')
我的筆記本電腦說,運(yùn)行這兩個(gè)命令只需花費(fèi)0.5秒,而腳本在一分鐘以后才運(yùn)行結(jié)束(使用了’wait’方法)。
雖然這個(gè)例子看起來是微不足道的,但是想像一下,如果你有數(shù)百臺這樣的服務(wù)器需要管理,那么你可以看到這幾行代碼和一個(gè)python import所帶來的強(qiáng)大的力量。
看完上述內(nèi)容,你們對基于Python的shell自動(dòng)化框架ShutIt怎么用有進(jìn)一步的了解嗎?如果還想了解更多知識或者相關(guān)內(nèi)容,請關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道,感謝大家的支持。