前段时间有个外国的程序猿走红网络,这个哥们可以说是懒到了极点,上班请假给领导发短信写脚本,下班晚回家给老婆发短信写脚本,甚至于接个咖啡也要写脚本。这个哥们离职之后,他的同事在他的办公电脑上发现了这些脚本,并公布到了网上,引起众程序猿纷纷膜拜。最近我刚好学到shell脚本部分,就让我分析其中的一个跟领导请假的脚本吧。
#!/bin/sh -e # Exit early if any session with my username is found if who | grep -wq $USER; then exit fi # Phone numbers MY_NUMBER='+xxx' NUMBER_OF_BOSS='+xxx' EXCUSES=( 'Locked out' 'Pipes broke' 'Food poisoning' 'Not feeling well' ) rand=$[ $RANDOM % ${#EXCUSES[@]} ] RANDOM_EXCUSE=${EXCUSES[$rand]} MESSAGE="Gonna work from home. "$RANDOM_EXCUSE # Send a text message RESPONSE=`curl -fSs -u "$TWILIO_ACCOUNT_SID:$TWILIO_AUTH_TOKEN" \ -d "From=$MY_NUMBER" -d "To=$NUMBER_OF_BOSS" -d "Body=$MESSAGE" \ "https://api.twilio.com/2010-04-01/Accounts/$TWILIO_ACCOUNT_SID/Messages"` # Log errors if [ $? -gt 0 ]; then echo "Failed to send SMS: $RESPONSE" exit 1 fi
一 shebang声明
#!/bin/sh -e
脚本的shebang声明是在脚本直接执行,而不使用sh或者bash间接执行的时候,告诉解释程序此脚本的类型,如果脚本中没有此声明,那么脚本在执行的时候会使用默认的shell程序执行它。在此例中,脚本使用了/bin/sh作为解释程序。
在ubuntu系列linux里,为了加快系统启动速度,/bin/sh被设置成了指向dash的链接,而非bash
#ll /bin/sh lrwxrwxrwx 1 root root 4 2月 16 10:23 /bin/sh -> dash*
在Centos里,/bin/sh链接到了bash
#ll /bin/sh lrwxrwxrwx. 1 root root 4 2月 14 16:55 /bin/sh -> bash
所以在ubuntu里如果要使用bash要写成#!/bin/bash才行。
-e选项表示脚本中有任何错误都会终止执行脚本。
二 执行条件过滤
# Exit early if any session with my username is found if who | grep -wq $USER; then exit fi
脚本在主要代码执行前,首先要判断用户传入的参数、执行者的权限、文件目录是否存在等是否符合脚本执行的条件,所以会进行相应的判断,如果没达到符合的条件就会提示用户退出执行。在本例中,判断了脚本的所属者是否登录了系统,如果已经登录了就表明他已经来上班了,下面的请假过程就不用执行了。
三 声明变量
# Phone numbers MY_NUMBER='+xxx' NUMBER_OF_BOSS='+xxx'
定义字符串变量,分别存入本人和老板的电话号码。
EXCUSES=( 'Locked out' 'Pipes broke' 'Food poisoning' 'Not feeling well' )
定义请假理由的数组
rand=$[ $RANDOM % ${#EXCUSES[@]} ]
随机抽取请假编号。${#EXCUSES[@]}代表数值元素的个数,然后通过与随机数的取余运算获得随机的编号。$[ ]为算数运算符号,与let、$(())等价
RANDOM_EXCUSE=${EXCUSES[$rand]}
取得数组中的某一项。
MESSAGE="Gonna work from home. "$RANDOM_EXCUSE
将消息头与随机项合并组成消息。
三 执行主要任务
# Send a text message RESPONSE=`curl -fSs -u "$TWILIO_ACCOUNT_SID:$TWILIO_AUTH_TOKEN" \ -d "From=$MY_NUMBER" -d "To=$NUMBER_OF_BOSS" -d "Body=$MESSAGE" \ "https://api.twilio.com/2010-04-01/Accounts/$TWILIO_ACCOUNT_SID/Messages"`
其中的$TWILIO_ACCOUNT_SID和$TWILIO_AUTH_TOKEN为环境变量,通过curl命令执行了登录网站发送短信的过程,并把执行的结果返回给RESPONSE变量。
四 判断执行的结果
# Log errors if [ $? -gt 0 ]; then echo "Failed to send SMS: $RESPONSE" exit 1 fi
当脚本的主要代码执行完毕后,通过检测执行后的状态来判断脚本的执行结果,如果执行结果大于0,说明执行过程中出现了错误,后台执行的脚本会以发邮件的方式提示用户。
五 设置环境变量和周期执行
TWILIO_ACCOUNT_SID=ACxxxxxxxxxxxxxxxxxxxxxxxxxxxxx TWILIO_AUTH_TOKEN=yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy
此环境变量在用户未登录的情况下也要执行,所以不能设置在~/.bashrc ~/.bash_profil /etc/bashrc /etc/profile.d里,最好写在/etc/rc.local里
# Runs `hangover.sh` monday to friday at 8:45 am. 45 8 * * 1-5 /path/to/scripts/hangover.sh >> /path/to/hangover.log 2>&1
周一至周五的8点45分执行脚本,并输出日志。此任务没有用户名,应该是使用命令crontab -e编辑的。
六 总结
这位仁兄的代码简洁,注释清晰,内容包含了shell编程的主要步骤,值得我等新人研究学习。
原创文章,作者:realmaster,如若转载,请注明出处:http://www.178linux.com/72647