X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=examples%2Fcontroller.py;h=6002f683632fa3655d40887717a8eb7cfce70e9f;hb=1a3cee44420e79fda92943edf636eaddb393145e;hp=7e7f0dae3a5fdddb3392a3f72755aed1f17a1f35;hpb=b64327bb58139f1e1f770b16492968423f011eab;p=osm%2FN2VC.git

diff --git a/examples/controller.py b/examples/controller.py
index 7e7f0da..6002f68 100644
--- a/examples/controller.py
+++ b/examples/controller.py
@@ -12,9 +12,10 @@ import asyncio
 import logging
 
 from juju.controller import Controller
+from juju import loop
 
 
-async def run():
+async def main():
     controller = Controller()
     await controller.connect_current()
     model = await controller.add_model(
@@ -31,13 +32,10 @@ async def run():
     await model.disconnect()
     await controller.destroy_model(model.info.uuid)
     await controller.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())