python rc脚本

rc脚本

使用python写一个memcached的启动脚本,可以参照/etc/init.d/memcached脚本。
下面是启动脚本的大体结构,下一章节详细介绍类中方法的实现。

#!/usr/bin/python
import sys

class Process(object):
    '''memcached rc script'''
	def __init__(self, name, program, args, workdir):
	    self.name = name
		self.program = program
		self.args = args
		self.workdir = workdir
		
	def _init(self):
	
	def start(self):
	
	def stop(self):
	
	def restart(self):
	    self.stop()
		self.start()
		
	def status(self):
	
	def help(self):
	
def main():
    name = 'memcached'
	prog = '/usr/bin/memcached'
	args = '-u nobody -p 11211 -c 1024 -m 64'
	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':
	main()

脚本功能:使用python写一个memcached的start/stop/restart/status脚本。
前面构造了脚本的基础结构,下面是具体方法的实现。

命令行启动

#!/usr/bin/python

import sys
import os
from subprocess import Popen, PIPE

class Process(object):
    '''memcached rc script'''
    def __init__(self, name, program, args, workdir):
        self.name = name
        self.program = program
        self.args = args
        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 _writePid(self):
        if self.pid:
            with open(self._pidFile(), 'w') as fd:
                fd.write(str(self.pid))

    def start(self):
        pid = self._getPid()
        if pid:
            print "%s is running..." % self.name
            sys.exit()
        self._init()
        cmd = self.program + ' ' +self.args
        p = Popen(cmd, stdout=PIPE, shell=True) 
        self.pid = p.pid
        self._writePid()
        print "%s start Sucessful" % 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):
        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__

def main():
    name = 'memcached'
    prog = '/usr/bin/memcached'
    args = '-u nobody -p 11211 -c 1024 -m 64'
    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__':
    main()
使用命令行启动,父进程为当前bash的PID,当前shell关闭之后,该子进程由init接受,即父进程变更为1。
查询当前bash的PID

方法1

[root@linux02 advance]# echo $$
98509

方法2

[root@linux02 advance]# tty
/dev/pts/0
 [root@linux02 advance]# ps -ef|grep pts/0
root      98501   1666  0 Jun20 ?        00:00:10 sshd: root@pts/0 
root      98509  98501  0 Jun20 pts/0    00:00:01 -bash
root     109237  98509  0 11:35 pts/0    00:00:00 ps -ef
root     109238  98509  0 11:35 pts/0    00:00:00 grep pts/0

守护进程启动

#!/usr/bin/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 _writePid(self):
        if self.pid:
            with open(self._pidFile(), 'w') as fd:
                fd.write(str(self.pid))

    def _readConf(self, f):
        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()
        cmd = [self.program] + self._parseArgs() + ['-d','-P', self._pidFile()]
        print cmd
        p = Popen(cmd, stdout=PIPE) 
        #self.pid = p.pid
        #self._writePid()
        print "%s start Sucessful" % 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):
        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__

def main():
    name = 'memcached'
    prog = '/usr/bin/memcached'
    args = '-u nobody -p 11211 -c 1024 -m 64'
    wd = '/var/tmp/memcached'
    
    pm = Process(name = name,
                 program = prog,
                 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__':
    main()

注意点

  • 以-d方式启动,需添加-P参数,不需要自己写入pid文件。
  • 守护进程启动方式在命令行启动脚本基础上进行了改进,主要是针对配置文件。
  • 守护进程方式启动,会出现二次fork,最后结果就是其父进程PID为1。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值