Python fabric으로 서버 관리 자동화

pip install fabric

install.py(fabfile.py) 스크립트 작성

from fabric.state import output
from fabric.api import run, env, put, sudo
 
#env.hosts=[
#    "root@server-001.test.com"
#]
#env.password = "passwd"
 
#env.warn_only = True
#output.status = False
#output.everything = False
env.disable_known_hosts = True
env.reject_unknown_hosts = False
 
def install_daemon(opt1, opt2, opt3):
    output.result = run("uname -a")
    #put('./install_daemon.sh', '~/')
    #output.result = run('/bin/bash ~/install_daemon.sh -a "%s" -b "%s" -c "%s"' % (opt1, opt2, opt3))
    return output.result.return_code
 
if __name__ == '__main__':
    from fabric.tasks import execute
    env.password = "password"
    result = execute(install_daemon, "opt1", "opt2", "opt3", hosts=["root@server-001.test.com"])
    for host in result.keys():
        return_code = result[host]
        print host, return_code

install_daemon.py 스크립트 작성

#!/bin/bash
 
while [[ $# -gt 1 ]]
do
key="$1"
 
case $key in
    -a|--opt1)
    OPT1="$2"
    shift # past argument
    ;;
    -b|--opt2)
    OPT2="$2"
    shift # past argument
    ;;
    -c|--opt3)
    OPT3="$2"
    shift # past argument
    ;;
    *)
            # unknown option
    ;;
esac
shift # past argument or value
done
 
if [ ! "$OPT1" ]
then
    echo "required opt1"
    exit -1
fi
 
if [ ! "$OPT2" ]
then
    echo "required opt2"
    exit -2
fi
 
if [ ! "$OPT3" ]
then
    echo "required opt3"
    exit -3
fi
 
PROCESS_NAME="daemon.py"
PROCESS_EXISTS=`ps auxwww |fgrep $PROCESS_NAME |grep -v grep |wc -l`
 
if [ "$PROCESS_EXISTS" -ne "0" ]
then 
    killall $PROCESS_NAME
    sleep 3
fi
 
NOW_DATETIME=`date +%Y%m%d%H%M%S`
USER=`whoami`
PWD=`cd ~;pwd`
 
INSTALL_DIR="/usr/local/daemon"
INSTALL_VER="1.1.0"
INSTALL_SRC="daemon-${INSTALL_VER}"
 
SRC_DIR="$PWD/src"
SRC_URL="http://download.test.com/src/${INSTALL_SRC}.tar.gz"
 
if [ -d "$INSTALL_DIR" ]
then
    mv ${INSTALL_DIR} ${INSTALL_DIR}_${NOW_DATETIME}
fi
 
if [ ! -d "$SRC_DIR" ]
then
    mkdir $SRC_DIR
fi
 
if [ -d "${SRC_DIR}/${INSTALL_SRC}" ]
then
    mv ${SRC_DIR}/${INSTALL_SRC} ${SRC_DIR}/${INSTALL_SRC}_${NOW_DATETIME}
fi
 
wget -q ${SRC_URL} -O "${SRC_DIR}/${INSTALL_SRC}.tar.gz"
if [ "$?" -ne "0" ]; then exit -4; fi
 
tar xvfz "${SRC_DIR}/${INSTALL_SRC}.tar.gz" -C $SRC_DIR
if [ "$?" -ne "0" ]; then exit -5; fi
 
cd ${SRC_DIR}/${INSTALL_SRC}
./configure --prefix=${INSTALL_DIR} \
--sbindir=${INSTALL_DIR}/bin \
--sysconfdir=${INSTALL_DIR}/etc \
--with-pid-dir=${INSTALL_DIR}/var/run
if [ "$?" -ne "0" ]; then exit -6; fi
 
make -j `grep processor /proc/cpuinfo | wc -l`
if [ "$?" -ne "0" ]; then exit -7; fi
make install
if [ "$?" -ne "0" ]; then exit -8; fi
mkdir -p ${INSTALL_DIR}/var/run
if [ "$?" -ne "0" ]; then exit -9; fi
mkdir -p ${INSTALL_DIR}/var/lock
if [ "$?" -ne "0" ]; then exit -10; fi
${INSTALL_DIR}/bin/daemon start
if [ "$?" -ne "0" ]; then exit -11; fi
sleep 3
 
PORT_EXISTS=`netstat -lnp | grep ':8001 '|fgrep ${PROCESS_NAME} |wc -l`
if [ "$PORT_EXISTS" -eq "0" ]
then
    echo "port is not open"
    exit -11
fi
BOOT_START=`egrep "${INSTALL_DIR}/bin/daemon\s+start" /etc/rc.d/rc.local |wc -l`
if [ "$BOOT_START" -eq "0" ]
then
    echo -e "# daemon start\n${INSTALL_DIR}/bin/daemon start" >> /etc/rc.d/rc.local
    if [ "$?" -ne "0" ]; then exit -12; fi
fi
exit 0

fab 커멘드로 설치 실행

fab -f install.py -H root@server-001.test.com install_daemon:opt1,opt2,opt

python 스크립트로 설치 실행

python install.py