memcached管理工具(python)

#!/usr/bin/env python
# -*- conding: utf-8 -*-

import os
import sys
import shlex
from subprocess import Popen, PIPE

work_dir = "/var/tmp"

class Process(object):
	args = []
	'''memcached rc script'''
	def __init__(self,name,
					program,
					work_dir,
					args=[]
				):
		self.name = name
		self.program = program
		self.args += []
		self.work_dir = work_dir
	
	def init(self):
 		if not os.path.exists(self.work_dir):
			os.mkdir(self.work_dir)
			os.chdir(self.work_dir)

	def _get_status(self):
		pass

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

	def _writePid(self):
		with open(self._pidFile(), 'w') as fd:
			fd.write(str(self.pid))
	
	def parser_args(self):
		config = readConfig()
		if 'USER' in config:
			self.args += ['-u', config['USER']]
		if 'PORT' in config:
			self.args += ['-p', config['PORT']]
		if 'MAXCONN' in config:
			self.args += ['-c', config['MAXCONN']]
		if 'CACHESIZE' in config:
			self.args += ['-m', config['CACHESIZE']]
		if 'DAEMON' in config:
			self.args += ['-d', config['DAEMON']]
		if 'PID' in config:
			self.args += ['-p', config['PID']]
		print self.args
		return self.args

	def start(self):
		p_check = Popen(['pidof', 'memcached'], stdout=PIPE)
		if p_check.stdout.read():
			print "memcached is already running..."
			sys.exit()
		self.init()
		self.args = self.parser_args()
		cmd = [self.program] + self.args
		p = Popen(cmd, stdout=PIPE)
		self.pid = p.pid
		self._writePid()
		print "memcached start DONE"

	def _readPid(self):
		if os.path.exists(self._pidFile()):
			with open(self._pidFile()) as fd:
				return int(fd.read())
			
	def _checkPid(self, pid):
		return os.path.exists('/proc/%s' % pid)

	def stop(self):
		pid = self._readPid()
		if self._checkPid(pid):
			os.kill(pid, 15)
			os.remove(self._pidFile())
			print "memcached stop DONE"
	
	def status(self):
		if self._checkPid(self._readPid()):
			print "memcached is running..."
		else:
			print "memcached not running"

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

def readConfig():
	f = open('/etc/sysconfig/memcached')
	lines = f.readlines()
	return dict([i.strip().replace('"','').split('=') for i in lines])

def main():
	name = 'memcached'
	program = '/usr/bin/memcached'
	#args = '-u memcached -p 11211'
	wd = '/var/tmp/memcached'
	pm = Process(name, program, wd)
	try:
		m = sys.argv[1]
	except IndexError,e:
		print "options error"
		sys.exit()
	if sys.argv[1] == 'start':
		pm.start()
	elif sys.argv[1] == 'stop':
		pm.stop()
	elif sys.argv[1] == 'status':
		pm.status()
	elif sys.argv[1] == 'restart':
		pm.restart()
		

if __name__ == '__main__':
	main()