Merge pull request #621 from dbcli/koljonen/show_transaction_status

Add transaction status to toolbar

1 files changed, 8 insertions, 1 deletions

diff --git a/pgcli/pgtoolbar.py b/pgcli/pgtoolbar.py
index 84db715d..be720074 100644
--- a/pgcli/pgtoolbar.py
+++ b/pgcli/pgtoolbar.py
@@ -10,7 +10,8 @@ def _get_vi_mode(cli):
         InputMode.INSERT_MULTIPLE: 'M',
     }[cli.vi_state.input_mode]
 
-def create_toolbar_tokens_func(get_vi_mode_enabled, get_is_refreshing):
+def create_toolbar_tokens_func(get_vi_mode_enabled, get_is_refreshing,
+                               failed_transaction, valid_transaction):
     """
     Return a function that generates the toolbar tokens.
     """
@@ -43,6 +44,12 @@ def create_toolbar_tokens_func(get_vi_mode_enabled, get_is_refreshing):
         else:
             result.append((token.On, '[F4] Emacs-mode'))
 
+        if failed_transaction():
+            result.append((token.Transaction.Failed, '     Failed transaction'))
+
+        if valid_transaction():
+            result.append((token.Transaction.Valid, '     Transaction'))
+
         if get_is_refreshing():
             result.append((token, '     Refreshing completions...'))