Updated examples to use juju.loop

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

diff --git a/examples/add_machine.py b/examples/add_machine.py
index 391df00..8ae2d40 100755 (executable)
--- a/examples/add_machine.py
+++ b/examples/add_machine.py
@@ -62,8 +62,9 @@ async def main():
         await model.disconnect()
 
 
-logging.basicConfig(level=logging.DEBUG)
-ws_logger = logging.getLogger('websockets.protocol')
-ws_logger.setLevel(logging.INFO)
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)
+    ws_logger = logging.getLogger('websockets.protocol')
+    ws_logger.setLevel(logging.INFO)
 
-loop.run(main())
+    loop.run(main())