61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
import socket
|
|
import logging
|
|
|
|
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)
|