sugar-toolkit-gtk3/sugar/p2p/model/LocalModel.py

import socket
import logging

from sugar.presence.Service import Service
from sugar.p2p.Notifier import Notifier
from sugar.p2p.model.AbstractModel import AbstractModel
from sugar.p2p import network

class ModelRequestHandler(object):
	def __init__(self, model):
		self._model = model

	def get_value(self, key):
		return self._model.get_value(key)

	def set_value(self, key, value):
		return self._model.set_value(key, value)

class LocalModel(AbstractModel):
	SERVICE_TYPE = "_olpc_model._tcp"
	SERVICE_PORT = 6300

	def __init__(self, activity, pservice, service):
		AbstractModel.__init__(self)
		self._pservice = pservice
		self._activity = activity
		self._service = service
		self._values = {}
		
		self._setup_service()
		self._notifier = Notifier(service)
	
	def get_value(self, key):
		return self._values[key]
		
	def set_value(self, key, value):
		self._values[key] = value
		self._notify_model_change(key)
		self._notifier.notify(key)

	def _setup_service(self):
		service = self._pservice.share_activity(self._activity,
				stype = LocalModel.SERVICE_TYPE,
				address = '', port = LocalModel.SERVICE_PORT)
		self._setup_server(service)
	
	# FIXME this is duplicated with StreamReader
	def _setup_server(self, service):
		started = False
		tries = 10
		port = service.get_port()
		while not started and tries > 0:
			try:
				logging.debug('Start model server on port %d' % (port))
				p2p_server = network.GlibXMLRPCServer(("", port))
				p2p_server.register_instance(ModelRequestHandler(self))
				started = True
			except(socket.error):
				port = port + 1
				tries = tries - 1
		service.set_port(port)