#!/usr/bin/python

 import com

 import worker

 import optparse

 import pickle

 import hashlib

 import subprocess

 import SocketServer

 option_list = [

 	optparse.make_option("-q", "--quiet", action="store_true", dest="quiet", 

 		help="quiet mode", default=False)]

 class Handler(SocketServer.BaseRequestHandler):

 	"""

 	"""	

 	def handle(self):

 		data = self.request[0]

 		socket = self.request[1]

 		#print "%s wrote:" % self.client_address[0]

 		#print data

 		key_found = False

 		for key, value in com.commands.iteritems():

 			if key in data:

 				key_found = True

 				args = data.partition(key)[2].split()				

 				if args:

 					value[0](socket, self.client_address, self.worker, args)

 				else:

 					value[0](socket, self.client_address, self.worker)

 		if not key_found:

 			print "unkown command: " + data

 			return

 			model_name, _, cell_data = data.partition(" ")

 			cell = pickle.loads(cell_data)

 			#self.cells[cell[0]] = cell[1]			

 			model = com.getModel(model_name)

 			#print cell

 			model.cells[cell[0]] = cell[1]

 			#print hashlib.sha256(str(self.cells)).hexdigest()

 class Server(SocketServer.UDPServer):

 	def __init__(self, host, port, handler, options, worker):

 		SocketServer.UDPServer.__init__(self, (host, port), handler)

 		self.host = host

 		self.port = port

 		self.handler = handler

 		self.options = options

 		self.handler.worker = worker

 		self.handler.cells = {}

 	def run(self):

 		self.serve_forever()

 def spawnWorker(server, n):

 	port = server[1] + 1

 	worker_list = [] 

 	for i in xrange(n):

 		print "starting worker %i" % (port + i)

 		#worker.spawn(server[0], port + i)

 		subprocess.Popen("python worker.py %i" % (port + i), shell=True)

 		worker_list.append((server[0], port + i)) 

 	return worker_list		

 if __name__ == "__main__":

 	parser = optparse.OptionParser(usage="usage: %prog [options] port worker",

 		option_list=option_list)

 	options, args = parser.parse_args()

 	if len(args) > 0:

 		host = "localhost"

 		port = int(args[0])

 		worker_count = int(args[1])

 		worker = spawnWorker((host, port), worker_count)

 		server = Server(host, port, Handler, options, worker)

 		server.run()