Implement model changes notification and hook

sugar/p2p/NotificationListener.py

@@ -0,0 +1,27 @@
+from Service import Service
+import network
+
+class NotificationListener:
+	TYPE = "_olpc_model_notification._udp"
+	ADDRESS = "224.0.0.222"
+	PORT = 6300
+	
+	def __init__(self, group, name):
+		server = network.GroupServer(NotificationListener.TYPE,
+									 NotificationListener.PORT,
+									 self._recv_multicast)
+		server.start()
+
+		service = Service(name, NotificationListener.TYPE,
+						  NotificationListener.ADDRESS,
+						  NotificationListener.PORT, True)
+		service.register(group)
+		
+		self._listeners = {}
+	
+	def add_listener(self, listener):
+		self._listeners.add(listener)
+	
+	def _recv_multicast(self, msg):
+		for listener in self._listeners:
+			listener(msg)

sugar/p2p/Notifier.py

@@ -0,0 +1,11 @@
+import network
+
+class Notifier:
+	def __init__(self, group, name):
+		service = group.get_service(name)
+		address = service.get_address()
+		port = service.get_port()
+		self._client = network.GroupClient(address, port)
+		
+	def notify(self, msg):
+		self._client.send_msg(msg)

sugar/p2p/model/LocalModel.py

@@ -1,10 +1,8 @@
 import socket
-import xmlrpclib
 
 from sugar.p2p.Service import Service
-
-MODEL_SERVICE_TYPE = "_olpc_model._tcp"
-MODEL_SERVICE_PORT = 6300
+from sugar.p2p.model.AbstractModel import AbstractModel
+import network
 
 class ModelRequestHandler(object):
 	def __init__(self, model):
@@ -16,26 +14,32 @@ class ModelRequestHandler(object):
 	def set_value(self, key, value):
 		return self._model.set_value(key, value)
 
-class LocalModel:
+class LocalModel(AbstractModel):
+	SERVICE_TYPE = "_olpc_model._tcp"
+	SERVICE_PORT = 6300
+
 	def __init__(self, group, model_id):
+		AbstractModel.__init__(self)
 		self._group = group
 		self._model_id = model_id
 		self._values = {}
 		
 		self._setup_service()
+		self._setup_notification()
 	
 	def get_value(self, key):
 		return self._values[key]
 		
 	def set_value(self, key, value):
 		self._values[key] = value
+		self._notify_model_change(key)
 
 	def _setup_service(self):
-		service = Service(self._model_id, MODEL_SERVICE_TYPE,
-						  '', MODEL_SERVICE_PORT)
+		service = Service(self._model_id, LocalModel.SERVICE_TYPE, '',
+						  LocalModel.SERVICE_PORT)
 		self._setup_server(service)
 		service.register(self._group)
-		
+	
 	# FIXME this is duplicated with StreamReader
 	def _setup_server(self, service):
 		started = False

sugar/p2p/model/RemoteModel.py

@@ -1,17 +1,26 @@
 import xmlrpclib
 
-from sugar.p2p.Service import Service
-import network
+from sugar.p2p.NotificationListener import NotificationListener
+from sugar.p2p.model.AbstractModel import AbstractModel
 
-class RemoteModel:
+class RemoteModel(AbstractModel):
 	def __init__(self, service):
+		AbstractModel.__init__(self)
+		
 		self._service = service
 		
 		addr = "http://%s:%d" % (service.get_address(), service.get_port())
 		self._client = xmlrpclib.ServerProxy(addr)
+		
+		self._setup_notification_listener()
 
 	def get_value(self, key):
 		return self._client.get_value(key)
 		
 	def set_value(self, key, value):
 		self._client.set_value(key, value)
+	
+	def _setup_notification_listener(self):
+		name = self._service.get_name()
+		self._notification = NotificationListener(self._group, name)
+		self._notification.add_listener(self._notify_model_change)

sugar/p2p/model/Store.py

@@ -15,7 +15,7 @@ class Store:
 		if self._local_models.has_key(model_id):
 			return self._local_models(model_id)
 		else:
-			service = self._group.get_service(model_id, MODEL_SERVICE_TYPE)
+			service = self._group.get_service(model_id, LocalModel.SERVICE_TYPE)
 			if service:
 				return RemoteModel(service)
 			else: