Fix mode segment

2012-11-23 19:03:58 +01:00 · 2012-11-23 19:03:58 +01:00 · e18f3fab75
parent 0f5a497834
commit e18f3fab75
1 changed files with 6 additions and 3 deletions
--- a/vim/segments/core.py
+++ b/vim/segments/core.py
@ -47,10 +47,13 @@ def mode(override=None):
 	'''
 	mode = vim_funcs['mode']()

+	if not override:
+		return (mode, vim_modes[mode])
+
 	try:
-		return override[mode]
-	except TypeError:
-		return vim_modes[mode]
+		return (mode, override[mode])
+	except IndexError:
+		return (mode, vim_modes[mode])


 def modified_indicator(text=u'+'):