Python 部署项目(Tomcat 容器)

此前书写了多实例的 Tomcat 启动等操作的脚本,今天完善 Tomcat 多实例部署(本脚本只提供思路)

脚本内容:

#!/usr/bin/env python
# _*_coding:utf-8_*_
# Author: "Edward.Liu"

# Import libary~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
import subprocess
import time
import sys
import signal
import os
import argparse
import contextlib
import zipfile


# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
class Tomcat(object):
    def __init__(self, tomcat_exe):
        self.tomcat_exe = tomcat_exe
        self.Tomcat_Home = "/software/%s" % tomcat_exe
        self.Tomcat_Log_Home = "/software/%s/logs" % tomcat_exe
        self.counnt = 10
        # deploy options
        self.timeStr = time.strftime("%Y-%m-%d-%H:%M")
        self.source_files = "/software/cybershop-front-0.0.1-SNAPSHOT.war"
        self.dest_dir = "/software/upload_project/%s-%s" % (
            self.timeStr, self.source_files.split('/')[2].split('.war')[0])
        self.dest_deploy_dir = "/software/deploy-front/%s" % self.source_files.split('/')[2].split('.war')[0]
        self.images_Home = "/software/newupload1"
        self.static_images_lins = "%s/assets/upload" % self.dest_dir
        self.static_Home = "/data/www"
        self.static_home_link = "%s/www" % self.dest_dir
        # deploy options --->end

    # Get Tomcat_PID~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def get_tomcat_pid(self):
        # 自定义获取程序 pid 与启动命令
        p = subprocess.Popen(['ps', '-Ao', 'pid,command'], stdout=subprocess.PIPE)
        out, err = p.communicate()
        for line in out.splitlines():
            if 'java' in line:
                if self.tomcat_exe in line:
                    pid = int(line.split(None, 1)[0])
                    return pid
                    # 获取 END

    # Start Tomcat Process~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def start_tomcat(self):
        if self.get_tomcat_pid() is not None:
            print "33[32m %s Is Started 33[0m" % self.tomcat_exe
        else:
            # Start Tomcat
            command_start_tomcat = "%s/bin/startup.sh" % self.Tomcat_Home
            p = subprocess.Popen(command_start_tomcat, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE, shell=True)
            stdout, stderr = p.communicate()
            print stdout, stderr

    # Stop Tomcat process~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def stop_tomcat(self):
        wait_sleep = 0
        if self.get_tomcat_pid() is None:
            print "33[32m %s is Not Running33[0m" % self.tomcat_exe + "~" * 20
        else:
            command_stop_tomcat = "%s/bin/shutdown.sh" % self.Tomcat_Home
            p = subprocess.Popen(command_stop_tomcat, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE, shell=True)
            stdout, stderr = p.communicate()
            while (self.get_tomcat_pid() is not None):
                print "waiting for processes to exit
"
                wait_sleep += 1
                time.sleep(1)
                if wait_sleep == self.counnt:
                    os.kill(self.get_tomcat_pid(), signal.SIGKILL)
                    print "33[32m Stop Tomcat is sucessful 33[0m"
                    break

    # View TomcatLogs~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def tomcat_log(self):
        command_tomcat_log = "tail -f %s/catalina.out " % self.Tomcat_Log_Home
        p = subprocess.Popen(command_tomcat_log, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
        returncode = p.poll()
        try:
            while returncode is None:
                line = p.stdout.readline()
                returncode = p.poll()
                line = line.strip()
                print line
            print returncode
        except KeyboardInterrupt:
            print 'ctrl+d or z'

    # Unzip Project_name~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def unzip(self):
        ret = 0
        try:
            with contextlib.closing(zipfile.ZipFile(self.source_files)) as zf:
                if not os.path.exists(self.dest_dir):
                    print "33[32mPath %s Is Not Exists Creating33[0m" % self.dest_dir
                    os.makedirs(self.dest_dir)
                    zf.extractall(self.dest_dir)
                    os.remove(self.source_files)
                    ret = 2

        except IOError:
            print "33[31m%s Is Not Exists Please send Files33[0m" % self.source_files
        return ret
    # Create Soft Links
    def soft_link(self):
        if os.path.islink(self.dest_deploy_dir):
            os.unlink(self.dest_deploy_dir)
            print "33[32mCreating Static Files/Images Link33[0m "
            os.symlink(self.images_Home, self.static_images_lins)
            os.symlink(self.static_Home, self.static_home_link)
            print self.dest_dir
            print self.dest_deploy_dir
            os.symlink(self.dest_dir, self.dest_deploy_dir)
        else:
            print "33[32mCreating Static Files/Images Link33[0m "
            os.symlink(self.images_Home, self.static_images_lins)
            os.symlink(self.static_Home, self.static_home_link)
            print self.dest_dir
            print self.dest_deploy_dir
            os.symlink(self.dest_dir, self.dest_deploy_dir)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
            description="eg: '%(prog)s' -c tomcat-front|tomcat -d {start|stop|status|restart|log|deploy}")
    # ADD Tomcat Apps ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    parser.add_argument('-c', '--app_name', nargs='+', dest='choices',
                        choices=('tomcat-front', 'tomcat-mobile'))  # choices 规定只能书写此处标出的, nargs='+' 至少有一个参数
    parser.add_argument('-d', '--Handle', action='store', nargs='?', dest='handle', default='log',
                        help='Input One of the {start|stop|status|restart|log|deploy}')  # nargs='?' 有一个货没有参数都可以
    parser.add_argument('-v', '--version', action='version', version='%(prog)s 1.0')

    args = parser.parse_args()
    if len(sys.argv) <= 4:
        parser.print_help()
    else:
        try:
            Handle = Tomcat(args.choices[0])
            if args.handle == 'log':
                Handle.tomcat_log()
            elif args.handle == 'start':
                Handle.start_tomcat()
            elif args.handle == 'stop':
                Handle.stop_tomcat()
            elif args.handle == 'restart':
                Handle.stop_tomcat()
                time.sleep(5)
                Handle.start_tomcat()
            elif args.handle == 'deploy':
                Handle.stop_tomcat()
                if Handle.unzip() != 0:
                    Handle.soft_link()
                Handle.start_tomcat()
            elif args.handle == 'status':
                if Handle.get_tomcat_pid() is not None:
                    print "33[32m %s Is Running is PID:33[0m" % Handle.tomcat_exe + "33[31m %s 33[0m" % Handle.get_tomcat_pid()
                else:
                    print "33[32m %s Not Running Or Not Exist 33[0m" % Handle.tomcat_exe
            else:
                print "33[31mYou Input parameter Is Not Exist33[0m"
        except TypeError:
            parser.print_help()