Блог

Я пытаюсь запустить click cli через bash с помощью разных групп команд, выполняемых через одну коллекцию команд.

src/preprocessing_extract_data/scripts/main.py

 import click 

@click.group()
def run_preprocessing_extract_data():
    pass


@run_preprocessing_extract_data.command()
@click.option(
    "--start_date",
    type=click.DateTime(formats=["%Y-%m-%d"]),
    required=True,
    help="Start date for the pipeline",
)
@click.option(
    "--end_date",
    type=click.DateTime(formats=["%Y-%m-%d"]),
    required=True,
    help="End date for the pipeline",
)
def main(start_date, end_date):
    ...

if __name__ == "__main__":
    main()
 

src/scripts/main.py

 from click import CommandCollection

from src.preprocessing_extract_data.scripts.main import run_preprocessing_extract_data

if __name__ == "__main__":
    cmds = [
        run_preprocessing_extract_data,
        # a few more similar command groups
    ]
    cli = CommandCollection(sources=cmds)
    cli()
 

scripts/entrypoint.sh

 #!/bin/sh
start_date="$1"
end_date="$2"

python src/scripts/main.py run_preprocessing_extract_data --start_date=$start_date --end_date=$end_date 
 

Я запускаю его с помощью ./scripts/entrypoint.sh --start_date="2020-11-01" --end_date="2021-12-01" --today="2021-12-10" , но он продолжает выходить из строя и выдает следующую ошибку:

 Usage: main.py [OPTIONS] COMMAND [ARGS]...
Try 'main.py --help' for help.

Error: No such command 'run_preprocessing_extract_data'.
 

Из документов:

Реализацией по умолчанию для такой системы слияния является CommandCollection класс. Он принимает список других мультикоманд и делает команды доступными на том же уровне.

Следовательно, в вашем скрипте теперь есть команда main ; вы можете проверить это, запустив свой скрипт с --help аргументами (или вообще без аргументов): python src/scripts/main.py --help .

Следовательно, вы можете сделать следующее:

 python src/scripts/main.py main --start_date="$start_date" --end_date="$end_date"
 

Кстати, вызов вашего сценария оболочки должен выполняться без --start_date : ./scripts/entrypoint.sh "2020-11-01" "2021-12-01" .