Use DEFAULT_BUFFER enum consistently

Author: joguSD

awsshell/app.py

@@ -18,6 +18,7 @@
 from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 from prompt_toolkit.history import InMemoryHistory, FileHistory
 from prompt_toolkit.enums import EditingMode
+from prompt_toolkit.enums import DEFAULT_BUFFER
 
 from awsshell.ui import create_default_layout
 from awsshell.config import Config
@@ -444,7 +445,7 @@ def create_application(self, completer, history,
         )
 
     def on_input_timeout(self, cli):
-        if not self.show_help or cli.current_buffer_name != u'DEFAULT_BUFFER':
+        if not self.show_help or cli.current_buffer_name != DEFAULT_BUFFER:
             return
         document = cli.current_buffer.document
         text = document.text

awsshell/keys.py

@@ -11,6 +11,7 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 from prompt_toolkit.key_binding.manager import KeyBindingManager
+from prompt_toolkit.enums import DEFAULT_BUFFER
 from prompt_toolkit.keys import Keys
 
 
@@ -154,7 +155,7 @@ def handle_f9(event):
 
             """
             if event.cli.current_buffer_name == u'clidocs':
-                event.cli.focus(u'DEFAULT_BUFFER')
+                event.cli.focus(DEFAULT_BUFFER)
             else:
                 event.cli.focus(u'clidocs')
 

awsshell/toolbar.py

@@ -11,6 +11,7 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 from pygments.token import Token
+from prompt_toolkit.enums import DEFAULT_BUFFER
 
 
 class Toolbar(object):
@@ -87,7 +88,7 @@ def get_toolbar_items(cli):
             else:
                 show_help_token = Token.Toolbar.Off
                 show_help_cfg = 'OFF'
-            if cli.current_buffer_name == 'DEFAULT_BUFFER':
+            if cli.current_buffer_name == DEFAULT_BUFFER:
                 show_buffer_name = 'cli'
             else:
                 show_buffer_name = 'doc'