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
|
#!/usr/bin/env python
"""A simple example demonstrating the following how to emit a non-zero exit code in your cmd2 application."""
import cmd2
class ReplWithExitCode(cmd2.Cmd):
"""Example cmd2 application where we can specify an exit code when existing."""
def __init__(self) -> None:
super().__init__()
@cmd2.with_argument_list
def do_exit(self, arg_list: list[str]) -> bool:
"""Exit the application with an optional exit code.
Usage: exit [exit_code]
Where:
* exit_code - integer exit code to return to the shell
"""
# If an argument was provided
if arg_list:
try:
self.exit_code = int(arg_list[0])
except ValueError:
self.perror(f"{arg_list[0]} isn't a valid integer exit code")
self.exit_code = 1
return True
if __name__ == '__main__':
import sys
app = ReplWithExitCode()
sys_exit_code = app.cmdloop()
app.poutput(f'{sys.argv[0]!r} exiting with code: {sys_exit_code}')
sys.exit(sys_exit_code)
|
