#183: Fix KeyError when 'current' is missing

juju · May 24, 2017 · f1a5bad · f1a5bad
1 parent 7806170
commit f1a5bad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/amulet/sentry.py b/amulet/sentry.py
@@ -544,7 +544,7 @@ def check_status(status):
                 service = status.get(service_name, {})
                 for unit_name, unit in service.items():
                     if unit['agent-status']:
-                        if unit['agent-status']['current'] != 'idle':
+                        if unit['agent-status'].get('current') != 'idle':
                             return False
                         since = datetime.strptime(unit['agent-status']['since'][:20], '%d %b %Y %H:%M:%S')
                         if (datetime.now() - since).total_seconds() < IDLE_THRESHOLD: