Add support for getting the level entity when prefabs are enabled. (#5680)

* Add support for getting the level entity when prefabs are enabled. Signed-off-by: Mike Balfour <82224783+mbalfour-amzn@users.noreply.github.com> * Made a little prettier. Signed-off-by: Mike Balfour <82224783+mbalfour-amzn@users.noreply.github.com> * Addressed PR feedback plus fixed the crash when the level component failed to add. Signed-off-by: Mike Balfour <82224783+mbalfour-amzn@users.noreply.github.com>
2021-11-16 19:23:53 -06:00
parent 92510df373
commit 551e6010dd
2 changed files with 22 additions and 7 deletions
@@ -57,6 +57,10 @@ def add_level_component(component_name):
                                                                 level_component_list, entity.EntityType().Level)
    level_component_outcome = editor.EditorLevelComponentAPIBus(bus.Broadcast, 'AddComponentsOfType',
                                                                [level_component_type_ids_list[0]])
+    if not level_component_outcome.IsSuccess():
+        print('Failed to add {} level component'.format(component_name))
+        return None
+
    level_component = level_component_outcome.GetValue()[0]
    return level_component