From 3a10e9365e5946c806219b870a464983ab4d879c Mon Sep 17 00:00:00 2001 From: Cory Johns Date: Tue, 25 Oct 2016 14:41:16 -0400 Subject: [PATCH] Add agent and workload status properties to Unit --- juju/unit.py | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) 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. -- 2.25.1