Add relate example

[osm/N2VC.git] / juju / delta.py

diff --git a/juju/delta.py b/juju/delta.py
index ea15f24..0b142dd 100644 (file)
--- a/juju/delta.py
+++ b/juju/delta.py
@@ -3,11 +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)
@@ -21,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']
@@ -54,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