Fix copy-paste bug

[osm/N2VC.git] / juju / machine.py

diff --git a/juju/machine.py b/juju/machine.py
index a7cb6a9..04abc3b 100644 (file)
--- a/juju/machine.py
+++ b/juju/machine.py
@@ -1,7 +1,25 @@
+import logging
+
 from . import model
+from .client import client
+
+log = logging.getLogger(__name__)
 
 
 class Machine(model.ModelEntity):
+    async def destroy(self, force=False):
+        """Remove this machine from the model.
+
+        """
+        facade = client.ClientFacade()
+        facade.connect(self.connection)
+
+        log.debug(
+            'Destroying machine %s', self.id)
+
+        return await facade.DestroyMachines(force, [self.id])
+    remove = destroy
+
     def run(self, command, timeout=None):
         """Run command on this machine.
 
@@ -11,6 +29,24 @@ class Machine(model.ModelEntity):
         """
         pass
 
+    async def set_annotations(self, annotations):
+        """Set annotations on this machine.
+
+        :param annotations map[string]string: the annotations as key/value
+            pairs.
+
+        """
+        log.debug('Updating annotations on machine %s', self.id)
+
+        self.ann_facade = client.AnnotationsFacade()
+        self.ann_facade.connect(self.connection)
+
+        ann = client.EntityAnnotations(
+            entity=self.id,
+            annotations=annotations,
+        )
+        return await self.ann_facade.Set([ann])
+
     def scp(
             self, source_path, user=None, destination_path=None, proxy=False,
             scp_opts=None):