#!/usr/bin/python

import com

import model

import optparse

import pickle

import SocketServer

option_list = [

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

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

def spawn(host, port):

	return Worker(host, port, Handler, ())

class Handler(SocketServer.BaseRequestHandler):

	"""

	"""	

	def handle(self):

		data = self.request[0].strip()

		socket = self.request[1]

		model_size = int(data[:10])

		model = pickle.loads(data[10:10+model_size])

		max_iter = int(data[10+model_size:20+model_size])

		model.update(max_iter)

class Worker(SocketServer.UDPServer):

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

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

		self.host = host

		self.port = port

		self.handler = handler

		self.options = options

	def run(self):

		print "worker running at port %s" % self.port

		self.serve_forever()

if __name__ == "__main__":

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

		option_list=option_list)

	options, args = parser.parse_args()

	if len(args) > 0:

		port = int(args[0])

		worker = Worker("localhost", port, Handler, options)

		worker.run()