#!/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:

			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()