Upgrade to prompt-toolkit 1.0.0

Author: jonathanslenders

awsshell/app.py

@@ -15,7 +15,6 @@
 from prompt_toolkit.filters import Always
 from prompt_toolkit.interface import CommandLineInterface, Application
 from prompt_toolkit.interface import AbortAction, AcceptAction
-from prompt_toolkit.utils import Callback
 from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 from prompt_toolkit.history import InMemoryHistory, FileHistory
 
@@ -279,7 +278,7 @@ def cli(self):
     def run(self):
         while True:
             try:
-                document = self.cli.run()
+                document = self.cli.run(reset_current_buffer=True)
                 text = document.text
             except InputInterrupt:
                 pass
@@ -423,7 +422,7 @@ def create_application(self, completer, history,
             buffer=self.create_buffer(completer, history),
             on_abort=AbortAction.RETRY,
             on_exit=AbortAction.RAISE_EXCEPTION,
-            on_input_timeout=Callback(self.on_input_timeout),
+            on_input_timeout=self.on_input_timeout,
             key_bindings_registry=self.key_manager.manager.registry,
         )
 

awsshell/style.py

@@ -14,7 +14,7 @@
 from pygments.style import Style
 from pygments.util import ClassNotFound
 from pygments.styles import get_style_by_name
-from prompt_toolkit.styles import default_style_extensions
+from prompt_toolkit.styles import default_style_extensions, style_from_dict
 
 
 class StyleFactory(object):
@@ -43,32 +43,26 @@ def style_factory(self, style_name):
         except ClassNotFound:
             style = get_style_by_name('vim')
 
-        class CliStyle(Style):
-            """Provides styles for the autocomplete menu and the toolbar.
+        # Create a style dictionary.
+        styles = {}
+        styles.update(style.styles)
+        styles.update(default_style_extensions)
+        t = Token
+        styles.update({
+            t.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
+            t.Menu.Completions.Completion: 'bg:#008888 #ffffff',
+            t.Menu.Completions.Meta.Current: 'bg:#00aaaa #000000',
+            t.Menu.Completions.Meta: 'bg:#00aaaa #ffffff',
+            t.Scrollbar.Button: 'bg:#003333',
+            t.Scrollbar: 'bg:#00aaaa',
+            t.Toolbar: 'bg:#222222 #cccccc',
+            t.Toolbar.Off: 'bg:#222222 #696969',
+            t.Toolbar.On: 'bg:#222222 #ffffff',
+            t.Toolbar.Search: 'noinherit bold',
+            t.Toolbar.Search.Text: 'nobold',
+            t.Toolbar.System: 'noinherit bold',
+            t.Toolbar.Arg: 'noinherit bold',
+            t.Toolbar.Arg.Text: 'nobold'
+        })
 
-            :type styles: dict
-            :param styles: Contains pygments style info.
-            """
-
-            styles = {}
-            styles.update(style.styles)
-            styles.update(default_style_extensions)
-            t = Token
-            styles.update({
-                t.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
-                t.Menu.Completions.Completion: 'bg:#008888 #ffffff',
-                t.Menu.Completions.Meta.Current: 'bg:#00aaaa #000000',
-                t.Menu.Completions.Meta: 'bg:#00aaaa #ffffff',
-                t.Menu.Completions.ProgressButton: 'bg:#003333',
-                t.Menu.Completions.ProgressBar: 'bg:#00aaaa',
-                t.Toolbar: 'bg:#222222 #cccccc',
-                t.Toolbar.Off: 'bg:#222222 #696969',
-                t.Toolbar.On: 'bg:#222222 #ffffff',
-                t.Toolbar.Search: 'noinherit bold',
-                t.Toolbar.Search.Text: 'nobold',
-                t.Toolbar.System: 'noinherit bold',
-                t.Toolbar.Arg: 'noinherit bold',
-                t.Toolbar.Arg.Text: 'nobold'
-            })
-
-        return CliStyle
+        return style_from_dict(styles)

setup.py

@@ -7,7 +7,7 @@
 
 requires = [
     'awscli>=1.8.9,<2.0.0',
-    'prompt-toolkit==0.52',
+    'prompt-toolkit>=1.0.0,<1.1.0',
     'boto3>=1.2.1,<2.0.0',
     'configobj>=5.0.6,<6.0.0',
     'Pygments>=2.1.3,<3.0.0',