X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Faction.py;h=0f25647d78d147c13ef8258d9773bbcd24ec069b;hb=40933126953cd398c1f9783513ec706a14f78665;hp=7897a8b437b33aa2d7e97b1625fbff67cadac308;hpb=51e09af5d8c07e7454cd03b8bf30487a593efa61;p=osm%2FN2VC.git

diff --git a/examples/action.py b/examples/action.py
index 7897a8b..0f25647 100644
--- a/examples/action.py
+++ b/examples/action.py
@@ -10,6 +10,7 @@ This example:
 import asyncio
 import logging
 
+from juju import loop
 from juju.model import Model
 
 
@@ -24,7 +25,7 @@ async def run_action(unit):
     logging.debug("Action results: %s", action.results)
 
 
-async def run():
+async def main():
     model = Model()
     await model.connect_current()
     await model.reset(force=True)
@@ -40,13 +41,10 @@ async def run():
         await run_action(unit)
 
     await model.disconnect()
-    model.loop.stop()
 
 
-logging.basicConfig(level=logging.DEBUG)
-ws_logger = logging.getLogger('websockets.protocol')
-ws_logger.setLevel(logging.INFO)
-loop = asyncio.get_event_loop()
-loop.set_debug(False)
-loop.create_task(run())
-loop.run_forever()
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)
+    ws_logger = logging.getLogger('websockets.protocol')
+    ws_logger.setLevel(logging.INFO)
+    loop.run(main())