From: Cory Johns <johnsca@gmail.com>
Date: Tue, 25 Oct 2016 18:41:16 +0000 (-0400)
Subject: Add agent and workload status properties to Unit
X-Git-Tag: 0.1.0~65
X-Git-Url: https://osm.etsi.org/gitweb/?a=commitdiff_plain;h=3a10e9365e5946c806219b870a464983ab4d879c;p=osm%2FN2VC.git

Add agent and workload status properties to Unit
---

diff --git a/juju/unit.py b/juju/unit.py
index 78acaff..4b24331 100644
--- a/juju/unit.py
+++ b/juju/unit.py
@@ -1,4 +1,5 @@
 import logging
+from datetime import datetime
 
 from . import model
 from .client import client
@@ -7,6 +8,54 @@ log = logging.getLogger(__name__)
 
 
 class Unit(model.ModelEntity):
+    @property
+    def agent_status(self):
+        """Returns the current agent status string.
+
+        """
+        return self.data['agent-status']['current']
+
+    @property
+    def agent_status_since(self):
+        """Get the time when the `agent_status` was last updated.
+
+        """
+        since = self.data['agent-status']['since']
+        # Juju gives us nanoseconds, but Python only supports microseconds
+        since = since[:26]
+        return datetime.strptime(since, "%Y-%m-%dT%H:%M:%S.%f")
+
+    @property
+    def agent_status_message(self):
+        """Get the agent status message.
+
+        """
+        return self.data['agent-status']['message']
+
+    @property
+    def workload_status(self):
+        """Returns the current workload status string.
+
+        """
+        return self.data['workload-status']['current']
+
+    @property
+    def workload_status_since(self):
+        """Get the time when the `workload_status` was last updated.
+
+        """
+        since = self.data['workload-status']['since']
+        # Juju gives us nanoseconds, but Python only supports microseconds
+        since = since[:26]
+        return datetime.strptime(since, "%Y-%m-%dT%H:%M:%S.%f")
+
+    @property
+    def workload_status_message(self):
+        """Get the workload status message.
+
+        """
+        return self.data['workload-status']['message']
+
     def add_storage(self, name, constraints=None):
         """Add unit storage dynamically.