A few small changes for being compatible with the latest prompt_toolk…

…it 2.0
donnemartin · Mar 8, 2018 · 3a32e19 · 3a32e19
1 parent 9d3c77a
commit 3a32e19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/haxor_news/haxor.py b/haxor_news/haxor.py
@@ -152,7 +152,7 @@ def _create_cli(self):
             enable_open_in_editor=True,
             enable_system_prompt=True,
             enable_suspend=True,
-            extra_key_bindings=self.key_manager.key_bindings)
+            key_bindings=self.key_manager.key_bindings)
 
     def _add_comment_pagination(self, document_text):
         """Add the command to enable comment pagination where applicable.

diff --git a/haxor_news/style.py b/haxor_news/style.py
@@ -15,7 +15,7 @@
 
 from pygments.token import Token
 from pygments.util import ClassNotFound
-from prompt_toolkit.styles.pygments import style_from_pygments_dict, style_from_pygments
+from prompt_toolkit.styles.pygments import style_from_pygments_dict, style_from_pygments_cls
 from prompt_toolkit.styles import merge_styles
 import pygments.styles
 
@@ -48,7 +48,7 @@ def style_factory(self, name):
 
         # Create styles dictionary.
         return merge_styles([
-            style_from_pygments(pygments_style),
+            style_from_pygments_cls(pygments_style),
             style_from_pygments_dict({
                 Token.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
                 Token.Menu.Completions.Completion: 'bg:#008888 #ffffff',