From 01b96693ece2941d9fe28b19401aae6f909c590e Mon Sep 17 00:00:00 2001 From: Tim Van Steenburgh Date: Tue, 18 Oct 2016 16:20:18 -0400 Subject: [PATCH] Wait for entity if it doesn't exist yet --- juju/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/juju/model.py b/juju/model.py index 7781d34..5fb45b6 100644 --- a/juju/model.py +++ b/juju/model.py @@ -1325,7 +1325,10 @@ class BundleHandler(object): Annotations holds the annotations as key/value pairs. """ entity_id = self.resolve(id_) - entity = self.model.state.get_entity(entity_type, entity_id) + try: + entity = self.model.state.get_entity(entity_type, entity_id) + except KeyError: + entity = await self._wait_for_new(entity_type, entity_id) return await entity.set_annotations(annotations) -- 2.25.1