X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=juju%2Fdelta.py;h=0b142ddf0141a4e29e13493f8b8538f2b55551ad;hb=a56869a418a68b795817d4f14d4065d11784f09f;hp=c7f694d0d1c85873d64f7e5468cf3387a96889a5;hpb=173b900fcd95b2436af55df2618302146f4a2f40;p=osm%2FN2VC.git

diff --git a/juju/delta.py b/juju/delta.py
index c7f694d..0b142dd 100644
--- a/juju/delta.py
+++ b/juju/delta.py
@@ -3,10 +3,12 @@ from .client import client
 
 def get_entity_delta(d):
     _delta_types = {
+        'action': ActionDelta,
         'application': ApplicationDelta,
+        'annotation': AnnotationDelta,
         'machine': MachineDelta,
         'unit': UnitDelta,
-        'action': ActionDelta,
+        'relation': RelationDelta,
     }
 
     return _delta_types[d.entity](d.deltas)
@@ -20,6 +22,12 @@ class EntityDelta(client.Delta):
         return None
 
 
+class ActionDelta(EntityDelta):
+    def get_entity_class(self):
+        from .action import Action
+        return Action
+
+
 class ApplicationDelta(EntityDelta):
     def get_id(self):
         return self.data['name']
@@ -29,6 +37,15 @@ class ApplicationDelta(EntityDelta):
         return Application
 
 
+class AnnotationDelta(EntityDelta):
+    def get_id(self):
+        return self.data['tag']
+
+    def get_entity_class(self):
+        from .annotation import Annotation
+        return Annotation
+
+
 class MachineDelta(EntityDelta):
     def get_entity_class(self):
         from .machine import Machine
@@ -44,7 +61,7 @@ class UnitDelta(EntityDelta):
         return Unit
 
 
-class ActionDelta(EntityDelta):
+class RelationDelta(EntityDelta):
     def get_entity_class(self):
-        from .action import Action
-        return Action
+        from .relation import Relation
+        return Relation