Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 1 addition & 22 deletions SoftLayer/shell/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,7 @@

import click
from prompt_toolkit import auto_suggest as p_auto_suggest
from prompt_toolkit import history as p_history
from prompt_toolkit import shortcuts as p_shortcuts
from pygments import token

from SoftLayer.CLI import core
from SoftLayer.CLI import environment
Expand Down Expand Up @@ -49,33 +47,14 @@ def cli(ctx, env):
app_path = click.get_app_dir('softlayer_shell')
if not os.path.exists(app_path):
os.makedirs(app_path)
history = p_history.FileHistory(os.path.join(app_path, 'history'))
complete = completer.ShellCompleter(core.cli)

while True:
def get_prompt_tokens(_):
"""Returns tokens for the command prompt"""
tokens = []
try:
tokens.append((token.Token.Username, env.client.auth.username))
tokens.append((token.Token.At, "@"))
except AttributeError:
pass

tokens.append((token.Token.Host, "slcli-shell"))
if env.vars['last_exit_code']:
tokens.append((token.Token.ErrorPrompt, '> '))
else:
tokens.append((token.Token.Prompt, '> '))

return tokens

try:
line = p_shortcuts.prompt(
completer=complete,
history=history,
complete_while_typing=True,
auto_suggest=p_auto_suggest.AutoSuggestFromHistory(),
get_prompt_tokens=get_prompt_tokens,
)

# Parse arguments
Expand Down