gyorgyf@0: #!/usr/bin/env python
gyorgyf@0: # encoding: utf-8
gyorgyf@0: """
gyorgyf@0: mcserver.py
gyorgyf@0: 
gyorgyf@0: Created by George Fazekas on 2012-06-16.
gyorgyf@0: Copyright (c) 2012 . All rights reserved.
gyorgyf@0: """
gyorgyf@0: 
gyorgyf@0: import os,sys,optparse,signal
gyorgyf@0: import cherrypy as cp
gyorgyf@0: 
gyorgyf@0: from cherrypy.lib import static
gyorgyf@0: import subprocess as sp
gyorgyf@0: from subprocess import Popen as spopen
gyorgyf@0: 
gyorgyf@0: 
gyorgyf@0: op = optparse.OptionParser()
gyorgyf@0: op.add_option('-u', '--user', action="store", dest="USER", default="server", type="str")
gyorgyf@0: options, args = op.parse_args()
gyorgyf@0: CONFIG_FILE = "mc%(USER)s.cfg" %options.__dict__
gyorgyf@0: 
gyorgyf@0: print CONFIG_FILE
gyorgyf@0: 
gyorgyf@0: if not os.path.isfile(CONFIG_FILE) :
gyorgyf@0: 	print >> sys.stderr, "Config file not found."
gyorgyf@0: 	sys.exit(-1)
gyorgyf@0: 
gyorgyf@0: 
gyorgyf@0: class MoodConductor:
gyorgyf@0: 	
gyorgyf@0: 	# @cp.expose
gyorgyf@0: 	def index(self):
gyorgyf@0: 		return "Hello world!"
gyorgyf@0: 	# index.exposed = True
gyorgyf@0: 	
gyorgyf@0: 	@cp.expose
gyorgyf@0: 	def mood(self,x,y):
gyorgyf@0: 		print x,y
gyorgyf@0: 		return "OK"
gyorgyf@0: 
gyorgyf@0: 
gyorgyf@0: 
gyorgyf@0: def getProcessPids(port,kill=False):
gyorgyf@0: 	'''Get the pid of the offending Python process given a port after an unsuccessful restart.'''
gyorgyf@0: 	print "Running lsof -i :"+str(port)," ...\n\n"
gyorgyf@0: 	command = "lsof -i :"+str(port)
gyorgyf@0: 	w = spopen(command,stdout=sp.PIPE,stderr=sp.PIPE,shell=True)
gyorgyf@0: 	se = w.stderr.readlines()
gyorgyf@0: 	result = w.stdout.readlines()
gyorgyf@0: 	exitcode = w.wait()
gyorgyf@0: 	if not result : 
gyorgyf@0: 		print "getProcessPid:: Unable to obtain process pid. (lsof returned nothing. exitcode: %s)" %str(exitcode)
gyorgyf@0: 		return False
gyorgyf@0: 	import pprint
gyorgyf@0: 	pprint.pprint(result)
gyorgyf@0: 	
gyorgyf@0: 	# get heading:
gyorgyf@0: 	ix = None
gyorgyf@0: 	head = result[0].upper()
gyorgyf@0: 	if 'PID' in head:
gyorgyf@0: 		head = filter(lambda x: x != str(), head.split(' '))
gyorgyf@0: 		head = map(lambda x: x.strip().replace(' ',''), head)
gyorgyf@0: 		if 'PID' in head : ix = head.index('PID')
gyorgyf@0: 	# get process pid
gyorgyf@0: 	pids = []
gyorgyf@0: 	for line in result :
gyorgyf@0: 		if 'python' in line.lower() :
gyorgyf@0: 			line = filter(lambda x: x != str(), line.split(' '))
gyorgyf@0: 			line = map(lambda x: x.strip().replace(' ',''), line)
gyorgyf@0: 			try :
gyorgyf@0: 				if ix : 
gyorgyf@0: 					pids.append(int(line[ix]))
gyorgyf@0: 				else:
gyorgyf@0: 					numbers = filter(lambda x: x.isdigit(), line)
gyorgyf@0: 					pids.append(int(numbers[0]))
gyorgyf@0: 			except:
gyorgyf@0: 				print 'Error parsing lsof results.' 
gyorgyf@0: 				return False
gyorgyf@0: 	print 'Pids found: ',pids
gyorgyf@0: 	# kill if specified
gyorgyf@0: 	if kill :
gyorgyf@0: 		pids_killed = []
gyorgyf@0: 		import signal
gyorgyf@0: 		for pid in pids:
gyorgyf@0: 			print 'Killing process: ',pid
gyorgyf@0: 			try :
gyorgyf@0: 				os.kill(pid,signal.SIGKILL)
gyorgyf@0: 				pids_killed.append(pid)
gyorgyf@0: 			except :
gyorgyf@0: 				print 'Failed: ',pid
gyorgyf@0: 		if pids_killed :
gyorgyf@0: 			print 'Processes killed:',pids_killed,' Waiting 10 secods...'
gyorgyf@0: 			import time
gyorgyf@0: 			time.sleep(10)
gyorgyf@0: 			return True
gyorgyf@0: 	return False
gyorgyf@0: 
gyorgyf@0: 
gyorgyf@0: def main(argv=None):
gyorgyf@0: 	
gyorgyf@0: 	# Configure and start
gyorgyf@2: 	# port = 8030
gyorgyf@2: 	# cp.config.update({'server.socket_host': '127.0.0.1'})
gyorgyf@2: 	# cp.config.update({'server.socket_port': port})
gyorgyf@0: 	cp.config.update({'tools.staticdir.root': os.getcwd()})
gyorgyf@2: 	port = int(cp.server.socket_port)
gyorgyf@0: 	getProcessPids(port,kill=True)
gyorgyf@0: 	cp.tree.mount(MoodConductor(),script_name="/moodconductor",config=CONFIG_FILE)
gyorgyf@0: 	cp.quickstart()
gyorgyf@0: 
gyorgyf@0: if __name__ == "__main__":
gyorgyf@0: 	main()