Python Tomcat Script(多实例)

之前书写过 Tomcat 单实例的 Python 脚本,本次增加 Tomcat 多实例的操作脚本.

1:准备

　　安装所需 Python 插件
　　　　A方法: pip install argparse

　　　　B方法:1.安装依赖包{setuptools}

　　　　　　　2.安装 argparse

#tar xzf setuptools-19.1.1.tar.gz && cd setuptools-19.1.1 && python setup.py install
#tar xzf argparse-1.4.0.tar.gz && cd argparse-1.4.0 && python setup.py install

2.脚本内容

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

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


# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
class Tomcat(object):
    def __init__(self, tomcat_exe):
        self.tomcat_exe = tomcat_exe
        self.Tomcat_Home = "/install/%s" % tomcat_exe
        self.Tomcat_Log_Home = "/install/%s/logs" % tomcat_exe
        self.counnt = 10
        self.Tomcat_Examples = ['tomcat', 'tomcat-mobile']

    # Get Tomcat_PID~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    def get_tomcat_pid(self):
        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

    # 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'


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
            description="eg: '%(prog)s' -c tomcat|tomcat-mobile -d {start|stop|status|restart|log}")
    # ADD Tomcat Apps ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    parser.add_argument('-c', '--app_name', nargs='+', dest='choices', choices=('tomcat', 'tomcat-mobile'))
    parser.add_argument('-d', '--Handle', action='store', nargs='?', dest='handle',
                        help='Input One of the {start|stop|status|restart|log}')
    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 == '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

        except TypeError:
            parser.print_help()