configured default value for cliColorDepth

aws · Mar 13, 2022 · 5b19dd5 · 5b19dd5
1 parent ea76de4
commit 5b19dd5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/awscli/autoprompt/prompttoolkit.py b/awscli/autoprompt/prompttoolkit.py
@@ -114,8 +114,9 @@ def create_application(self):
         # which can be used to configure different color schemes in aws cli. You can read more
         # about it in the documentation linked below:
         # https://python-prompt-toolkit.readthedocs.io/en/master/pages/advanced_topics/styling.html?highlight=classes#color-depths
-        cliColorPalette = {"black_and_white": ColorDepth.MONOCHROME, "ansi_colors": ColorDepth.ANSI_COLORS_ONLY,"256_colors": ColorDepth.DEFAULT, "true_colors": ColorDepth.TRUE_COLOR}
-        cliColorDepth = self._session.get_config_variable('color_palette')
+        cliColorDepth = {"black_and_white": ColorDepth.MONOCHROME, "ansi_colors": ColorDepth.ANSI_COLORS_ONLY,"256_colors": ColorDepth.DEFAULT, "true_colors": ColorDepth.TRUE_COLOR}
+        depthLevel = self._session.get_config_variable('color_palette')
+        depthLevel = depthLevel if depthLevel else "256_colors"
         self._create_buffers()
         input_buffer_container, doc_window, output_window = self._create_containers()
         layout = self._factory.create_layout(
@@ -127,7 +128,7 @@ def create_application(self):
         kb = kb_manager.keybindings
         app = Application(layout=layout, key_bindings=kb, full_screen=False,
                           output=self._output, erase_when_done=True,
-                          input=self._input, color_depth=cliColorPalette[cliColorDepth])
+                          input=self._input, color_depth=cliColorDepth[depthLevel])
         self._set_app_defaults(app)
         return app