Shell调用Python函数
Shell脚本调用Python函数,本例子是通过Shell调用Python脚本,检查服务器上的某端口是否开启,编写的Python脚本port.py,脚本内容:
#!/usr/bin/python
# port.py
import socket
def scan(port):
s = socket.socket()
s.settimeout(0.1)
if s.connect_ex(('localhost', port)) == 0:
return 'port: ' + str(port) + ' open'
else:
return 'port: ' + str(port) + ' not open'
s.close()
if __name__ == '__main__':
scan()
在Shell脚本中,这样编写命令
cd /home/appuser
checkresult=`python -c 'import port; print port.scan(80)'`
echo $checkresult
语句解释:首先cd到要执行的Python脚本目录,然后使用 python -c 'import port; print port.scan(80)' 调用port模块中的scan函数,检查80端口是否正在使用。-c选项是告诉Python来执行一些Python命令
echo输出结果
port: 80 open
如果传入一个不是正在使用的端口号801
checkresult=`python -c 'import port; print port.scan(801)'`
echo输出结果
port: 801 not open
另外,`python -c 'import port; print port.scan(80)'` 也可以写成 $(python -c 'import port; print port.scan(80)')
实际使用中的一个例子,Shell脚本调用一个Python方法,通过socket检查远端的IP和端口是否可连通
Python脚本pysocket.py:
#!/usr/bin/python
import socket
def scan(ip,port):
s = socket.socket()
s.settimeout(2)
if s.connect_ex((str(ip), port)) == 0:
return 0
else:
return 1
s.close()
if __name__ == '__main__':
scan()
pysocket.py中的scan方法里,如果通过socket能连接上ip和端口的服务,则返回0值,否则返回1
Shell脚本
max_check=5
#检查服务是否启动
function check() {
#先进入到pysocket.py所在目录
cd /opt/data/py
i=1
while [ $i -le $max_check ]
do
echo "$i th check: " $1 " " $2
reval=$(python -c 'import pysocket; print pysocket.scan("'$1'",'$2')')
echo "check result: "$reval
if [ $reval -eq 0 ]; then
echo "check success"
return 0
else
echo "next check after 30 seconds"
sleep 30
fi
let i++
done
return 1
}
Shell中的check函数中,通过 $(python -c 'import pysocket; print pysocket.scan("'$1'",'$2')') 调用pysocket.py 的scan方法并传参数,如果连接不通,每次检查后等待30秒,最多检查5次。
Shell调用check函数
check 10.7.128.186 26666
以上为编写运维脚本的朋友提供帮助