Updated examples to use juju.loop

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

diff --git a/examples/leadership.py b/examples/leadership.py
index c52de60..b231003 100644 (file)
--- a/examples/leadership.py
+++ b/examples/leadership.py
@@ -9,6 +9,7 @@ This example:
 import asyncio
 
 from juju.model import Model
+from juju import loop
 
 async def report_leadership():
     model = Model()
@@ -22,5 +23,6 @@ async def report_leadership():
 
     await model.disconnect()
 
-loop = asyncio.get_event_loop()
-loop.run_until_complete(report_leadership())
+
+if __name__ == '__main__':
+    loop.run(report_leadership())