ShutIt是一个易于使用的基于shell的自动化框架。它对基于python的expect库（pexpect）进行了包装。你可以把它看作是“没有痛点的expect”。它可以通过pip进行安装。

Hello World

让我们从最简单的例子开始吧。创建一个名为example.py的文件：

import shutit

session = shutit.create_session('bash')

session.send('echo Hello World', echo=True)

运行这个文件：

python example.py

输出：

python example.py

echo "Hello World"

echo "Hello World"

Hello World

Ians-MacBook-Air.local:ORIGIN_ENV:RhuebR2T#

“send”函数的第一个参数是要运行的命令。“echo”的参数将会输出到终端上。默认情况下，ShutIt是静默的。

登录服务器

如果你要登陆一台服务器并执行服务器上的命令。可以将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()

程序将登录到这台服务器上，并输出主机名。

hostname

hostname

example.com

example.com:cgoIsdVv:heDa77HB#

显然，这很不安全！你可以这样运行：

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()

它会让你输入密码：

Input Secret:

hostname

hostname

example.com

example.com:cgoIsdVv:heDa77HB#

同样的，“login”方法在登录后改变了提示符。你给了ShutIt一个登录命令，并附带用户名和密码（如果需要的话），然后，ShutIt会完成剩余的事情。

“logout”负责终止“login”，并向屏幕输出发生的任何变化。

登录到多台服务器

假设你有一个集群包含两台服务器，并希望同时登录到这两个服务器上去。则只需要创建两个会话，并运行类似的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()

将输出这样的结果：

$ python example.py

Password for server1

Input Secret:  

Password for server2

Input Secret:

hostname

hostname

one.example.com

one.example.com:Fnh2pyFj:qkrsmUNs# hostname

hostname

two.example.com

two.example.com:Gl2lldEo:D3FavQjA#

实例：监控多台服务器

我们可以通过添加一些代码逻辑来检查命令的输出，从而将上述代码变成一个简单的监控工具：

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()

更复杂的IO – Expecting

假设你需要跟一个命令行程序进行交互，并且要实现自动化操作。在这里，我们使用telnet来举一个简单的例子：

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”的参数。你只需要给出telnet提示符的一个子集来进行匹配。

注意“check_exit”的参数，后面我们会讲到这个参数的。上面这段代码将输出：

$ 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 GMT  

<HTML><HEAD><meta http-equiv="content-type" content="text/html;charset=utf-8">

<TITLE>302 Moved</TITLE></HEAD><BODY>

<H1>302 Moved</H1>

The document has moved

<A HREF="http://www.google.co.uk/?gfe_rd=cr&amp;ei=huczWcj3GfTW8gfq0paQDA">

here

</A>.

</BODY></HTML>

Connection closed by foreign host.

现在回到“checkexit = false”上来。由于telnet命令会返回一个错误的退出码（1），我们不想让脚本执行失败，这里的“checkexit = false”能让ShutIt知道你并不关注这个退出码。

如果你没有传入这个参数，ShutIt会给你一个交互式的提示，如果你有终端接入的话。这被称为“暂停点”。

暂停点

你可以随便在什么时候通过调用以下方法来设置一个“暂停点”。

[...]

session.pause_point('This is a pause point')

[...]

当脚本运行到暂停点时，同时按下“Ctrl”和“]”，则可以让脚本继续执行。这对于调试非常有用：添加一个暂停点，看看周围，然后继续。试试这个：

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#

更复杂的IO – Backgrounding

回到我们上面的“监控多台服务器”的例子上来。想象一下，我们要在每台服务器上运行一个长时间运行的任务。默认情况下，ShutIt会持续运行很长时间。但是我们可以在后台运行任务来加快ShutIt的运行速度。

在这里，你可以使用简单的命令“sleep 60”来尝试一个例子。

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')

我的笔记本电脑说，运行这两个命令只需花费0.5秒，而脚本在一分钟以后才运行结束（使用了’wait’方法）。

虽然这个例子看起来是微不足道的，但是想像一下，如果你有数百台这样的服务器需要管理，那么你可以看到这几行代码和一个python import所带来的强大的力量。

更多信息

ShutIt可以做很多事。更多信息，请参阅：

  • ShutIt (https://ianmiell.github.io/shutit/)

  • GitHub (https://github.com/ianmiell/shutit/blob/master/README.md)

ShutIt：一个基于 Python 的 shell 自动化框架的更多相关文章

  1. 符号执行-基于python的二进制分析框架angr

    转载:All Right 符号执行概述 在学习这个框架之前首先要知道符号执行.符号执行技术使用符号值代替数字值执行程序,得到的变量的值是由输入变 量的符号值和常量组成的表达式.符号执行技术首先由Kin ...

  2. python+request接口自动化框架

    python+request接口自动化框架搭建 1.数据准备2.用python获取Excel文件中测试用例数据3.通过requests测试接口4.根据接口返回的code值和Excel对比 但本章只讲整 ...

  3. 基于Selenium的Web自动化框架增强篇

    在写完上一篇“基于Selenium的Web自动化框架”(http://www.cnblogs.com/AlwinXu/p/5836709.html)之后一直没有时间重新审视该框架,正好趁着给同事分享的 ...

  4. 从零学习基于Python的RobotFramework自动化

    从零学习基于Python的RobotFramework自动化 一.        Python基础 1)      版本差异 版本 编码 语法 其他 2.X ASCII try: raise Type ...

  5. 基于python的webUI自动化-小白基础篇

    最近打算研究一下基于python的webUI自动化,先自学了一下相关基础知识,大概用了一个多月的时间,主要是找视频,一边看视频或者文档一边对照着敲代码运行. 重点强调:一定要一边看一边对照着敲代码运行 ...

  6. psutil一个基于python的跨平台系统信息跟踪模块

    受益于这个模块的帮助,在这里我推荐一手. https://pythonhosted.org/psutil/#processes psutil是一个基于python的跨平台系统信息监视模块.在pytho ...

  7. 【转】发布一个基于NGUI编写的UI框架

    发布一个基于NGUI编写的UI框架 1.加载,显示,隐藏,关闭页面,根据标示获得相应界面实例 2.提供界面显示隐藏动画接口 3.单独界面层级,Collider,背景管理 4.根据存储的导航信息完成界面 ...

  8. 【转】基于Selenium的web自动化框架(python)

    1 什么是selenium Selenium 是一个基于浏览器的自动化工具,它提供了一种跨平台.跨浏览器的端到端的web自动化解决方案.Selenium主要包括三部分:Selenium IDE.Sel ...

  9. 初识TPOT：一个基于Python的自动化机器学习开发工具

    1. TPOT介绍 一般来讲,创建一个机器学习模型需要经历以下几步: 数据预处理 特征工程 模型选择 超参数调整 模型保存 本文介绍一个基于遗传算法的快速模型选择及调参的方法,TPOT:一种基于Pyt ...

随机推荐

  1. Atitit. 包厢记时系统 的说明，教程，维护，故障排查手册v2 pb25.doc

    Atitit. 包厢记时系统 的说明,教程,维护,故障排查手册v2 pb25.doc 1. 服务器方面的维护1 1.1. 默认情况下,已经在系统的启动目录下增加了 个启动项目1 1.2. 后台服务.保 ...

  2. JMX简单样例

    一:创建maven项目,在pom.xml里面增加例如以下依赖 <dependency> <groupId>com.sun.jdmk</groupId> <ar ...

  3. dos指令 批处理文件

    windows下开发的时候难免写一些脚本,脚本的调用又难以避免的写批处理文件,也就是(.bat)文件!这个文件是什么呢?其实就是以下的这些dos命令.以下是从网上摘抄的,留以记录,待以后需要时查阅.也 ...

  4. Python内置函数之bool()

    该函数是一个类对象 class bool([x]) bool()只能传入一个参数. bool()用来判断对象是否为True,返回值为True或者False. 下面看看例子: >>> ...

  5. NPOI读取操作excel

    .读取using (FileStream stream = new FileStream(@"c:\客户资料.xls", FileMode.Open, FileAccess.Rea ...

  6. 关于WPF界面框架MahApps.Metro的一个BUG

    碰到了这个问题,记录一下,以便以后查阅: 在一个WPF项目中使用MahApps.Metro界面框架,其中有一个功能是嵌入一个带句柄的标记. 首先WPF是出了窗体和WebBrowser带有句柄外,其他控 ...

  7. 大数据(3) - 高可用 HDFS HA

    HDFS HA高可用 1 HA概述 1)所谓HA(high available),即高可用(7*24小时不中断服务). 2)实现高可用最关键的策略是消除单点故障.HA严格来说应该分成各个组件的HA机制 ...

  8. 初尝CDN:什么是分布式服务节点?

    什么是CDN?笔者初见时也是摸不着头脑,查阅了大量的资料才明白什么是CDN,笔者为您揭秘什么是CDN? CDN的全称是Content Delivery Network,即内容分发网络.CDN的通俗理解 ...

  9. CentOS上使用Squid+Stunnel搭建代理服务器教程

    这篇文章主要介绍了CentOS上使用Squid+Stunnel搭建代理服务器教程,同时文中也介绍了用户认证的方法,适合于多用户共同使用代理,这种功能在国内用还是比较exciting的~需要的朋友可以参 ...

  10. Java连接Sql Server 2008的简单数据库应用

    1.从微软官网下载JDBC驱动包 sqljdbc_4.0.2206.100_chs.exe,双击解压文件到指定目录,我的指定目录是: C:\Program Files\Microsoft JDBC D ...