| 12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- from __future__ import annotations
- from typing import Callable
- from traceback import print_tb
- from click.core import Command as ClickCommand
- def check_command_ran_without_error(result):
- print(result)
- assert result.exit_code == 0, print_tb(
- result.exc_info[2]
- ) # run command without errors
- def to_flags(cli_input: dict) -> list:
- """Turn dictionary of CLI input into a list of CLI flags ready for use in FlaskCliRunner.invoke().
- Example:
- cli_input = {
- "year": 2020,
- "country": "NL",
- }
- cli_flags = to_flags(cli_input) # ["--year", 2020, "--country", "NL"]
- runner = app.test_cli_runner()
- result = runner.invoke(some_cli_function, to_flags(cli_input))
- """
- return [
- item
- for sublist in zip(
- [f"--{key.replace('_', '-')}" for key in cli_input.keys()],
- cli_input.values(),
- )
- for item in sublist
- ]
- def get_click_commands(module) -> list[Callable]:
- return [
- getattr(module, attr)
- for attr in dir(module)
- if type(getattr(module, attr)) == ClickCommand
- ]
|
