forked from prompt-toolkit/ptpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathprompt_style.py
More file actions
77 lines (62 loc) · 1.76 KB
/
prompt_style.py
File metadata and controls
77 lines (62 loc) · 1.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from __future__ import unicode_literals
from abc import ABCMeta, abstractmethod
from six import with_metaclass
from pygments.token import Token
__all__ = (
'PromptStyle',
'IPythonPrompt',
'ClassicPrompt',
)
class PromptStyle(with_metaclass(ABCMeta, object)):
"""
Base class for all prompts.
"""
@abstractmethod
def in_tokens(self, cli):
" Return the input tokens. "
return []
@abstractmethod
def in2_tokens(self, cli, width):
"""
Tokens for every following input line.
:param width: The available width. This is coming from the width taken
by `in_tokens`.
"""
return []
@abstractmethod
def out_tokens(self, cli):
" Return the output tokens. "
return []
class IPythonPrompt(PromptStyle):
"""
A prompt resembling the IPython prompt.
"""
def __init__(self, python_input):
self.python_input = python_input
def in_tokens(self, cli):
return [
(Token.In, 'In ['),
(Token.In.Number, '%s' % self.python_input.current_statement_index),
(Token.In, ']: '),
]
def in2_tokens(self, cli, width):
return [
(Token.In, '...: '.rjust(width)),
]
def out_tokens(self, cli):
return [
(Token.Out, 'Out['),
(Token.Out.Number, '%s' % self.python_input.current_statement_index),
(Token.Out, ']:'),
(Token, ' '),
]
class ClassicPrompt(PromptStyle):
"""
The classic Python prompt.
"""
def in_tokens(self, cli):
return [(Token.Prompt, '>>> ')]
def in2_tokens(self, cli, width):
return [(Token.Prompt.Dots, '...')]
def out_tokens(self, cli):
return []