Skip to content

ui

Full documentation pages are generated for docstring reference only and may contain symbols imported from other modules. Imported symbols are not distinguished from locally defined symbols and will appear in any module that they are imported into. For better information on where symbols should be imported from, review the sourcecode on the github.

FoSpy._dev.testing.ui

_get_opts_and_run 

_get_opts_and_run(test, options, choice_name)
Source code in FoSpy/_dev/testing/ui.py 
64
65
66
67
68
69
70
71
def _get_opts_and_run(test, options, choice_name):
    def func():
        args = get_options(options, choice_name)
        if not args:
            return args
        test.run(**args)

    return func

get_options 

get_options(options, choice_name)
Source code in FoSpy/_dev/testing/ui.py 
 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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
def get_options(options, choice_name):
    import questionary
    args = {o:v for o,v in options.values()}

    bools = [k for k,v in [(k,v[1]) for k,v in options.items()] if isinstance(v, bool)]
    strings = [k for k,v in [(k,v[1]) for k,v in options.items()] if isinstance(v, str)]

    toggles = { o:(f"[ ] {o}", f"[x] {o}") for o in bools }

    def update_string(args, choice):
        opt = options[choice][0]

        args[opt] = questionary.text(f"{choice} (current: {args[opt]}): ").ask()

        return args

    def update_bool(args, choice):
        opt = options[choice][0]

        args[opt] = not args[opt]

        return args

    exits = {
        "[Run Test]": lambda args: args,
        "[Back]": lambda args: None,
        "[Quit]": lambda args: False
    }

    while True:
        toggle_labels = { toggles[o][1] if args[options[o][0]] else toggles[o][0] : o for o in bools }
        string_labels = { f"{o}: ({args[options[o][0]]})" : o for o in strings }

        all_labels = {**toggle_labels, **string_labels}


        updates = {**{choice:update_bool for choice in toggle_labels.values()},
                   **{choice:update_string for choice in string_labels.values()}}

        choice = questionary.select(
            f"Additional Options for Test: {choice_name}",
            choices=[
                *all_labels,
                questionary.Separator(),
                *exits
            ]
        ).ask()

        if choice in exits:
            return exits[choice](args)

        choice = all_labels.get(choice, None)
        print(choice)
        args = updates.get(choice, lambda args, choice: args)(args, choice)

get_test 

get_test(**args)
Source code in FoSpy/_dev/testing/ui.py 
 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
def get_test(**args):
    import questionary
    from . import TESTS
    from ...config import values as cfg
    from ._utils import toggle_branches, get_current_branch

    cfg.DEV.branch = get_current_branch()

    show_advanced = args == {}
    advanced = False

    while True:
        current_branch = cfg.DEV.branch
        next_branch = "main" if current_branch == "dev" else "dev"
        branch_label = f"[Switch to {next_branch} feature branch (currently on {current_branch})]"

        choices = {}

        for name, (test, options) in TESTS.items():
            if advanced:
                choices[name] = _get_opts_and_run(test, options, name)
            elif not args:
                choices[name] = test.run
            else:
                def try_test(t=test, a=args):
                    try:
                        t.run(**a)
                    except Exception as e:
                        print("Failed to run test with provided arguments", e)

                choices[name] = try_test

        choices[questionary.Separator()] = None

        if show_advanced:
            advanced_label = "[x] Advanced options" if advanced else "[ ] Advanced options"
            choices[advanced_label] = "toggle_advanced"

        choices["[Quit]"] = False
        choices[branch_label] = toggle_branches

        label = questionary.select(
            "Select a test to run:",
            choices=list(choices.keys())
        ).ask()

        selection = choices[label]

        if selection == "toggle_advanced":
            advanced = not advanced
            continue

        elif callable(selection):
            out = selection()
            if out is None:
                continue
            return out

        else:
            return selection