Separate main and run functions
1 files changed, 5 insertions(+), 3 deletions(-) M check.py
M check.py +5 -3
@@ 74,7 74,7 @@ def view(state, t, styles) -> None: t.addstr(i + 2, 0, pformat(state), styles['plain']) -def main(t, styles): +def run(t, styles): state = { 'tasks': [(False, t) for t in TASKS], 'current_task_id': 0, @@ 88,7 88,9 @@ def main(t, styles): if not state['running']: break +def main(): + with TerminalAndStyles() as (t, styles): + run(t, styles) if __name__ == '__main__': - with TerminalAndStyles() as (t, styles): - main(t, styles) + main()