python—memcached启动脚本

死siua11

#!/usr/bin/env python　　
#encoding:utf-8
　　

　　
import sys
　　
import os
　　
from subprocess import Popen, PIPE
　　

　　
class Process(object):
　　
    '''memcached rc script'''
　　

　　
    argv = {"USER": "memcached",
　　
            "PORT": "11211",
　　
            "MAXCONN": "1024",
　　
            "CACHESIZE": "64",
　　
            "OPTIONS": ""}   # 默认启动参数
　　

　　
    def __init__(self,name,program,workdir):  # 初始化类，包含：self、程序名、程序启动文件、pid文件路径
　　
        self.name = name
　　
        self.program = program
　　
        self.workdir = workdir
　　

　　

　　
    def __readConf(self,f):    # 读取配置文件，返回字典
　　
        with open(f,"r") as fd:
　　
            lines = fd.readlines()
　　
            return dict([i.strip().split("=") for i in lines])
　　

　　
    def __parseArgv(self):    # 配置文件替换默认参数
　　
    '''/etc/sysconfig/memcached'''
　　
        conf = self.__readConf("/etc/sysconfig/memcached")
　　

　　
        if "USER" in conf:
　　
            self.argv["USER"] = conf["USER"]
　　

　　
        if "PORT" in conf:
　　
            self.argv["PORT"] = conf["PORT"]
　　

　　
        if "MAXCONN" in conf:
　　
            self.argv["MAXCONN"] = conf["MAXCONN"]
　　

　　
        if "CACHESIZE" in conf:
　　
           self.argv["CACHESIZE"] = conf["CACHESIZE"]
　　

　　
        options = ["-u",self.argv["USER"],
　　
                   "-p",self.argv["PORT"],
　　
                   "-m",self.argv["CACHESIZE"],
　　
                   "-c",self.argv["MAXCONN"]]  # 替换的参数，以列表形式返回
　　

　　
        os.system("chown %s %s" % (self.argv["USER"],self.workdir)) # 修改pid文件目录的属主为启动用户，才能写pid文件
　　
        print ("启动参数为：%s" % str(options))
　　
        return options
　　

　　
    def __workDir(self):   # 判断pid文件路径，没有则创建
　　
        '''/var/tmp/memcached'''
　　
        if not os.path.exists("/var/tmp/memcached"):
　　
            os.mkdir("/var/tmp/memcached")
　　
            os.chdir("/var/tmp/memcached")
　　

　　
    def __pidFile(self):   # 获取pid文件，带路径
　　
        '''/var/tmp/memcached/memcached.pid'''
　　
        return os.path.join(self.workdir,"%s.pid" % self.name)
　　

　　
    def start(self):
　　
        pid = self.__getPid()
　　
        if pid:
　　
            print ("%s is running" % self.name)
　　
            sys.exit()
　　

　　
        self.__workDir()
　　
        cmd = [self.program] + self.__parseArgv() + ["-d", "-P", self.__pidFile()]  # 启动参数，后台启动方式，按指定文件自动写pid
　　
        p = Popen(cmd,stdout=PIPE)   # 启动进程方式
　　
        print("%s start successful" % self.name)
　　

　　

　　
    def __getPid(self):
　　
        p = Popen(["pidof",self.name],stdout=PIPE)  # 获取pid方式
　　
        pid = p.stdout.read().strip()    # 过滤pid
　　
        return pid
　　

　　
    def stop(self):
　　
        pid = self.__getPid()
　　
        if pid:
　　
            os.kill(int(pid),15)         # 杀掉进程方式，pid为数值类型
　　
            if os.path.exists(self.__pidFile()):
　　
                os.remove(self.__pidFile())     # 删掉pid文件
　　
            print("%s is stopped" % self.name)
　　

　　

　　
    def restart(self):
　　
        self.stop()
　　
        self.start()
　　

　　

　　
    def status(self):
　　
        pid = self.__getPid()
　　
        if pid:
　　
            print ("%s is running" % self.name)
　　
        else:
　　
            print ("%s is not running" % self.name)
　　

　　

　　
    def help(self):
　　
        print ("Usage: %s {start|stop|restart|status}" % __file__)
　　

　　

　　
def main():
　　
    name = "memcached"
　　
    program = "/data/program/memcached/bin/memcached"
　　
    workdir = "/var/tmp/memcached"
　　

　　
    pm = Process(name = name,
　　
                 program = program,
　　
                 workdir = workdir)    # 传参，也可以按顺序只写实参
　　
    try:
　　
        cmd = sys.argv[1]  # 如启动参数有误，则抛出异常,并退出
　　

　　
    except Exception as e:
　　
        raise e
　　
        sys.exit()
　　

　　
    if cmd == "start":
　　
        pm.start()
　　

　　
    elif cmd == "stop":
　　
        pm.stop()
　　

　　
    elif cmd == "restart":
　　
        pm.restart()
　　

　　
    elif cmd == "status":
　　
        pm.status()
　　

　　
    else:
　　
        pm.help()
　　

　　
if __name__ == "__main__":
　　
    main()