def get_airflow_process_command()

in images/airflow/2.9.2/healthcheck.py [0:0]

• 14 lines of code
• 7 McCabe index (conditional complexity)

def get_airflow_process_command(airflow_component: str):
    """
    Get airflow command substring for a given airflow component.

    :param airflow_component: Airflow component running on host. 
    :return: Airflow command substring.
    """
    if airflow_component == "SCHEDULER":
        return "/usr/local/airflow/.local/bin/airflow scheduler"

    if airflow_component == "WORKER":
        return "MainProcess] -active- (celery worker)"

    if airflow_component == "STATIC_ADDITIONAL_WORKER":
        return "MainProcess] -active- (celery worker)"

    if airflow_component == "DYNAMIC_ADDITIONAL_WORKER":
        return "MainProcess] -active- (celery worker)"

    if airflow_component == "ADDITIONAL_WEBSERVER":
        return "/usr/local/airflow/.local/bin/airflow webserver"

    if airflow_component == "WEB_SERVER":
        return "/usr/local/airflow/.local/bin/airflow webserver"

    exit_with_status(ExitStatus.INVALID_AIRFLOW_COMPONENT)