1
#!/usr/bin/env python
2
"""A simple cmd2 application."""
3
import sys
4
from dataclasses import dataclass
5
from random import choice, randint
6
from typing import Optional
7

8
import cmd2
9
from cmd2.parsing import Statement
10

11

12
@dataclass(frozen=True)
13
class BuildStatus:
14
    id: int
15
    name: str
16
    status: str
17

18

19
class FirstApp(cmd2.Cmd):
20
    """A simple cmd2 application."""
21

22
    def __init__(self):
23
        self._status_cache = dict()
24

25
    def _start_build(self, name: str) -> BuildStatus:
26
        return BuildStatus(randint(10, 100), name, "Started")
27

28
    def _get_status(self, name: str) -> Optional[BuildStatus]:
29
        status = self._status_cache.get(name)
30
        status_types = ["canceled", "restarted", "error", "finished"]
31

32
        if status.status != "finished":
33
            status = BuildStatus(status.id, status.name, choice(status_types))
34
            self._status_cache[name] = status
35

36
        return status
37

38
    build_parser = cmd2.Cmd2ArgumentParser()
39
    build_parser.add_argument("name", help="Name of build to start")
40

41
    @cmd2.with_argparser(build_parser)
42
    def do_build(self, args: Statement):
43
        """Executes a long running process at an API endpoint"""
44
        status = self._start_build(args.name)
45
        self._status_cache[args.name] = status
46
        self.poutput(f"Build {args.name.upper()} successfully started with id : {status.id}")
47
        self.last_result = status
48

49
    status_parser = cmd2.Cmd2ArgumentParser()
50
    status_parser.add_argument("name", help="Name of build determine status of")
51

52
    @cmd2.with_argparser(status_parser)
53
    def do_status(self, args: Statement):
54
        """Shows the current status of a build"""
55
        status = self._get_status(args.name)
56
        self.poutput(f"Status for Build: {args.name} \n {status.status}")
57
        self.last_result = status
58

59

60
if __name__ == "__main__":
61
    import sys
62
    c = FirstApp()
63
    sys.exit(c.cmdloop())

1
import sys
2
import time
3

4
# start build
5
result = app('build tower')
6

7
# If there was an error then exit
8
if not result:
9
    print('Build failed')
10
    sys.exit()
11

12
# This is a BuildStatus dataclass object
13
build = result.data
14

15
print(f"Build {build.name} : {build.status}")
16

17
# Poll status
18
while True:
19
    # Perform status check
20
    result = app('status tower')
21

22
    # error checking
23
    if not result:
24
        print("Unable to determine status")
25
        break
26

27
    build_status = result.data
28

29
    # If the status shows complete then the script is done
30
    if build_status.status in ['finished', 'canceled']:
31
        print(f"Build {build.name} has completed")
32
        break
33

34
    print(f"Current Status: {build_status.status}")
35
    time.sleep(1)