X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2FN2VC.git;a=blobdiff_plain;f=examples%2Flocalcharm.py;h=978703ea72d292bd7723be9f6225b7e83b3cfb13;hp=9abfe419761ffa78fa4e0c266cc5d378ff67a04d;hb=65d71ab71ce1aaf9ca3aeeee7dd6cb8793278b3e;hpb=5cf8d29d1ec5ff9360920dd831ff6ead145e3d11

diff --git a/examples/localcharm.py b/examples/localcharm.py
index 9abfe41..978703e 100644
--- a/examples/localcharm.py
+++ b/examples/localcharm.py
@@ -10,9 +10,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()
 
@@ -24,13 +25,10 @@ async def run():
     )
 
     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())