Merge pull request #3 from drmikecrowe/master

Support xonsh>=0.9.18 -- fixes #2 (ptk_shell_type)
vaaaaanquish · May 9, 2020 · 7b6ed71 · 7b6ed71
2 parents c4a7e2e + 30fb38d
commit 7b6ed71
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/xontrib/powerline2.xsh b/xontrib/powerline2.xsh
@@ -2,7 +2,11 @@ import os
 from os import path
 from collections import namedtuple
 from time import strftime
-from xonsh.platform import ptk_shell_type
+try:
+    from xonsh.platform import ptk_shell_type
+    ptk2 = ptk_shell_type() == 'prompt_toolkit2'
+except ImportError:
+    ptk2 = True
 
 
 __all__ = ()
@@ -30,7 +34,7 @@ $PL_DEFAULT_COLORS = {
                 "full_rtns": ("WHITE", "RED", "#444"),
             }
 
-if ptk_shell_type() == 'prompt_toolkit2':
+if ptk2:
     $PTK_STYLE_OVERRIDES['bottom-toolbar'] = 'noreverse'
 
 modes = {