Shell调用Python函数

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

 

以上为编写运维脚本的朋友提供帮助