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 34f7869..8ae2d40 100755 (executable)
--- a/examples/add_machine.py
+++ b/examples/add_machine.py
@@ -13,8 +13,8 @@ import logging
 from juju import loop
 from juju.model import Model
 
-MB = 1024 * 1024
-GB = MB * 1024
+MB = 1
+GB = 1024
 
 
 async def main():
@@ -54,15 +54,17 @@ async def main():
                         for unit in application.units))
 
         await application.remove()
-        await machine3.destroy()
-        await machine2.destroy()
-        await machine1.destroy()
+
+        await machine3.destroy(force=True)
+        await machine2.destroy(force=True)
+        await machine1.destroy(force=True)
     finally:
         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())