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
76 lines (61 loc) · 1.7 KB
/
prompt_style.py
File metadata and controls
76 lines (61 loc) · 1.7 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
from __future__ import unicode_literals
from abc import ABCMeta, abstractmethod
from six import with_metaclass
__all__ = (
'PromptStyle',
'IPythonPrompt',
'ClassicPrompt',
)
class PromptStyle(with_metaclass(ABCMeta, object)):
"""
Base class for all prompts.
"""
@abstractmethod
def in_prompt(self):
" Return the input tokens. "
return []
@abstractmethod
def in2_prompt(self, width):
"""
Tokens for every following input line.
:param width: The available width. This is coming from the width taken
by `in_prompt`.
"""
return []
@abstractmethod
def out_prompt(self):
" 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_prompt(self):
return [
('class:in', 'In ['),
('class:in.number', '%s' % self.python_input.current_statement_index),
('class:in', ']: '),
]
def in2_prompt(self, width):
return [
('class:in', '...: '.rjust(width)),
]
def out_prompt(self):
return [
('class:out', 'Out['),
('class:out.number', '%s' % self.python_input.current_statement_index),
('class:out', ']:'),
('', ' '),
]
class ClassicPrompt(PromptStyle):
"""
The classic Python prompt.
"""
def in_prompt(self):
return [('class:prompt', '>>> ')]
def in2_prompt(self, width):
return [('class:prompt.dots', '...')]
def out_prompt(self):
return []