X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Ffuture.py;h=01803255494c8fb594e584251a20a5ee82f991ac;hb=d23810d57aa7f0fa67070781645f4898e20db0ed;hp=d0dd4631b8c6b8fe1ea110d8978a147e40a5d399;hpb=b64327bb58139f1e1f770b16492968423f011eab;p=osm%2FN2VC.git

diff --git a/examples/future.py b/examples/future.py
index d0dd463..0180325 100644
--- a/examples/future.py
+++ b/examples/future.py
@@ -6,9 +6,10 @@ import asyncio
 import logging
 
 from juju.model import Model
+from juju import loop
 
 
-async def run():
+async def main():
     model = Model()
     await model.connect_current()
     await model.reset(force=True)
@@ -40,9 +41,8 @@ async def run():
     )
 
 
-logging.basicConfig(level=logging.DEBUG)
-ws_logger = logging.getLogger('websockets.protocol')
-ws_logger.setLevel(logging.INFO)
-loop = asyncio.get_event_loop()
-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())