20181207 rc脚本

#!/usr/bin/env python
import sys
import os
from subprocess import Popen, PIPE

class Process(object):
    '''memcached rc script'''
    args = {'USER': 'memcached',
            'PORT': 11211,
            'MAXCONN': 1024,
            'CACHESIZE': 64,
            'OPTIONS': ''}

    def __init__(self, name, program, workdir):
        self.name = name
        self.program = program
        self.workdir = workdir

    def _init(self): # 一个下划线是普通方法
        '''/var/tmp/memcached'''
        if not os.path.exists(self.workdir):
            os.mkdir(self.workdir)
            os.chdir(self.workdir)

    def _pidFile(self):
        '''/var/tmp/memcached/memcached.pid'''
        return os.path.join(self.workdir, "%s.pid" % self.name)

    def _writhPid(self):
        if self.pid:  # 公有属性可以在各个方法里调用
            with open(self._pidFile(), 'w') as fd:
                fd.write(str(self.pid))

    def _readConf(self, f): # 读取配置文件，如果配置文件不是上面args的默认值，直接用配置文件中的配置覆盖默认参数
        with open(f) as fd:
            lines = fd.readlines()
            return dict([i.strip().replace('"','').split('=') for i in lines]) #用这个字典值覆盖默认的值

    def _parseArgs(self): # 返回一个列表
        conf = self._readConf('/etc/sysconfig/memcached')
        if 'USER' in conf:
            self.args['USER'] = conf['USER']
        if 'PORT' in conf:
            self.args['PORT'] = conf['PORT']
        if 'MAXCONN' in conf:
            self.args['MAXCONN'] = conf['MAXCONN']
        if 'CACHESIZE' in conf:
            self.args['CACHESIZE'] = conf['CACHESIZE']
        options = ['-u', self.args['USER'],
                   '-p', self.args['PORT'],
                   '-m', self.args['CACHESIZE'],
                   '-c', self.args['MAXCONN']]
        os.system("chown %s %s" % (self.args['USER'], self.workdir))
        return options

    def start(self):
        pid = self._getPid()
        if pid:
            print "%s is running..." % self.name
            sys.exit()
        self._init()  #调用初始化方法，创建目录
        #没有加-d 父进程就是当前shell。加-d父进程直接就是1。列表更安全.self.args现在是个字典，需要转换为列表.
        # -d -P 会自动把进程号写到/var/tmp/memcached
        cmd = [self.program] + self._parseArgs() + ['-d', '-P', self._pidFile()]
        print cmd
        p = Popen(cmd, stdout=PIPE, shell=True)
        #self.pid = p.pid
        #self._writhPid()
        print '%s start Successful' % self.name

    def _getPid(self):
        p = Popen(['pidof', self.name], stdout=PIPE)
        pid = p.stdout.read().strip()
        return pid

    def stop(self):
        pid = self._getPid()
        if pid:
            os.kill(int(pid), 15)
            if os.path.exists(self._pidFile()):
                os.remove(self._pidFile())
            print "%s is stopped"  % self.name

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

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

    def help(self):
        print "Usage: $s {start|stop|status|restart}" % __file__ //__file__ 等同于shell $0 代表脚本本身


def main():
    name = 'memcached'
    prog = '/usr/bin/memcached'
    wd = '/var/tmp/memcached'
    pm = Process(name = name,
                 program = prog,
                 args = args,
                 workdir = wd)
    try:
        cmd = sys.argv[1]
    except IndexError, e:
        print "Option error"
        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__':  #当.py文件被直接运行时，if __name__ == '__main__'之下的代码块将被运行；当.py文件以模块形式被导入时，if __name__ == '__main__'之下的代码块不被运行
    main()