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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#!/usr/bin/env python
# coding=utf-8
"""
Create a CLI with a nested command structure as follows. The commands 'second' and 'third' navigate the CLI to the scope
of the submenu. Nesting of the submenus is done with the cmd2.AddSubmenu() decorator.
(Top Level)----second----->(2nd Level)----third----->(3rd Level)
| | |
---> say ---> say ---> say
"""
from __future__ import print_function
import sys
import cmd2
from IPython import embed
class ThirdLevel(cmd2.Cmd):
"""To be used as a third level command class. """
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.prompt = '3rdLevel '
self.top_level_attr = None
self.second_level_attr = None
def do_say(self, line):
print("You called a command in ThirdLevel with '%s'. "
"It has access to top_level_attr: %s "
"and second_level_attr: %s" % (line, self.top_level_attr, self.second_level_attr))
def help_say(self):
print("This is a third level submenu (submenu_ab). Options are qwe, asd, zxc")
def complete_say(self, text, line, begidx, endidx):
return [s for s in ['qwe', 'asd', 'zxc'] if s.startswith(text)]
@cmd2.AddSubmenu(ThirdLevel(),
command='third',
aliases=('third_alias',),
shared_attributes=dict(second_level_attr='second_level_attr', top_level_attr='top_level_attr'))
class SecondLevel(cmd2.Cmd):
"""To be used as a second level command class. """
def __init__(self, *args, **kwargs):
cmd2.Cmd.__init__(self, *args, **kwargs)
self.prompt = '2ndLevel '
self.top_level_attr = None
self.second_level_attr = 987654321
def do_ipy(self, arg):
"""Enters an interactive IPython shell.
Run python code from external files with ``run filename.py``
End with ``Ctrl-D`` (Unix) / ``Ctrl-Z`` (Windows), ``quit()``, '`exit()``.
"""
banner = 'Entering an embedded IPython shell type quit() or <Ctrl>-d to exit ...'
exit_msg = 'Leaving IPython, back to {}'.format(sys.argv[0])
embed(banner1=banner, exit_msg=exit_msg)
def do_say(self, line):
print("You called a command in SecondLevel with '%s'. "
"It has access to top_level_attr: %s" % (line, self.top_level_attr))
def help_say(self):
print("This is a SecondLevel menu. Options are qwe, asd, zxc")
def complete_say(self, text, line, begidx, endidx):
return [s for s in ['qwe', 'asd', 'zxc'] if s.startswith(text)]
@cmd2.AddSubmenu(SecondLevel(),
command='second',
aliases=('second_alias',),
shared_attributes=dict(top_level_attr='top_level_attr'))
class TopLevel(cmd2.Cmd):
"""To be used as the main / top level command class that will contain other submenus."""
def __init__(self, *args, **kwargs):
cmd2.Cmd.__init__(self, *args, **kwargs)
self.prompt = 'TopLevel '
self.top_level_attr = 123456789
def do_ipy(self, arg):
"""Enters an interactive IPython shell.
Run python code from external files with ``run filename.py``
End with ``Ctrl-D`` (Unix) / ``Ctrl-Z`` (Windows), ``quit()``, '`exit()``.
"""
banner = 'Entering an embedded IPython shell type quit() or <Ctrl>-d to exit ...'
exit_msg = 'Leaving IPython, back to {}'.format(sys.argv[0])
embed(banner1=banner, exit_msg=exit_msg)
def do_say(self, line):
print("You called a command in TopLevel with '%s'. "
"TopLevel has attribute top_level_attr=%s" % (line, self.top_level_attr))
def help_say(self):
print("This is a top level submenu. Options are qwe, asd, zxc")
def complete_say(self, text, line, begidx, endidx):
return [s for s in ['qwe', 'asd', 'zxc'] if s.startswith(text)]
if __name__ == '__main__':
root = TopLevel()
root.cmdloop()
|
