Addressed PR comments.

[osm/N2VC.git] / examples / unitrun.py

diff --git a/examples/unitrun.py b/examples/unitrun.py
index 0f8b556..a5b294b 100644 (file)
--- 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()