mirror of https://github.com/dbcli/pgcli
28 lines
1006 B
Python
28 lines
1006 B
Python
from prompt_toolkit.layout.toolbars import Toolbar
|
|
from prompt_toolkit.layout.utils import TokenList
|
|
from pygments.token import Token
|
|
|
|
class PGToolbar(Toolbar):
|
|
def __init__(self, token=None):
|
|
token = token or Token.Toolbar.Status
|
|
super(self.__class__, self).__init__(token=token)
|
|
|
|
def get_tokens(self, cli, width):
|
|
result = TokenList()
|
|
result.append((self.token, ' '))
|
|
if cli.buffers['default'].completer.smart_completion:
|
|
result.append((self.token.On, '[F2] Smart Completion: ON '))
|
|
else:
|
|
result.append((self.token.Off, '[F2] Smart Completion: OFF '))
|
|
|
|
if cli.buffers['default'].always_multiline:
|
|
result.append((self.token.On, '[F3] Multiline: ON'))
|
|
else:
|
|
result.append((self.token.Off, '[F3] Multiline: OFF'))
|
|
|
|
if cli.buffers['default'].always_multiline:
|
|
result.append((self.token,
|
|
' (Semi-colon [;] will end the line)'))
|
|
|
|
return result
|