X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Funitrun.py;h=a5b294bb9dd25e7eca61e2643896e498ab9bdf0b;hb=0b9e0ad499afba7c8a3b86a82e3cc5d42da4887d;hp=0f8b556a5b8a37d8b981d17dcd1d109178d3bbe1;hpb=1aaf1efd3ecb1fbb91c63dd88a90e15bf4268bd0;p=osm%2FN2VC.git

diff --git a/examples/unitrun.py b/examples/unitrun.py
index 0f8b556..a5b294b 100644
--- a/examples/unitrun.py
+++ b/examples/unitrun.py
@@ -12,30 +12,23 @@ import logging
 
 from juju.model import Model, ModelObserver
 
-
 async def run_stuff_on_unit(unit):
     print('Running command on unit', unit.name)
 
     # unit.run() returns a client.ActionResults instance
-    action_results = await unit.run('unit-get public-address')
-    action_result = action_results.results[0]
+    action = await unit.run('unit-get public-address')
+
+    print("Action results: {}".format(action.results))
 
-    print('Results from unit', unit.name)
-    print(action_result.__dict__)
+    # Inform asyncio that we're done.
+    await unit.model.disconnect()
+    unit.model.loop.stop()
 
 
 class MyModelObserver(ModelObserver):
     async def on_unit_add(self, delta, old, new, model):
         loop.create_task(run_stuff_on_unit(new))
 
-    async def on_action_change(self, delta, old, new, model):
-        print(delta.data)
-
-        action = new
-        if action.status == 'completed':
-            await action.model.disconnect()
-            action.model.loop.stop()
-
 
 async def run():
     model = Model()