Shell调用Python函数

ゝ一世哀愁。 2022-04-03 04:40 611阅读 0赞

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

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