Use LocalExecutor for 2.7 and above too instead of hardcoding 2.6

airflowctl/modes/virtualenv.py

@@ -12,6 +12,7 @@
 import typer
 import yaml
 from dotenv import load_dotenv
+from packaging import version
 from rich import print
 from rich.console import Console
 
@@ -315,12 +316,23 @@ def _setup_env_vars_to_run_airflow(self):
                 settings = yaml.safe_load(f)
             self.airflow_version = settings.get("airflow_version")
 
-        # TODO: Check for Airflow version >= 2.6 instead of hardcoding it
-        if self.airflow_version and self.airflow_version.startswith("2.6"):
+        if (
+            self.airflow_version
+            and is_valid_pep440_version(self.airflow_version)
+            and version.parse(self.airflow_version) >= version.parse("2.6.0")
+        ):
             os.environ["AIRFLOW__CORE__EXECUTOR"] = "LocalExecutor"
             os.environ["_AIRFLOW__SKIP_DATABASE_EXECUTOR_COMPATIBILITY_CHECK"] = "1"
 
 
+def is_valid_pep440_version(version_str: str) -> bool:
+    try:
+        version.parse(version_str)
+        return True
+    except version.InvalidVersion:
+        return False
+
+
 def create_virtualenv_with_specific_python_version(venv_path: Path, python_version: str):
     if isinstance(venv_path, str):
         venv_path = Path(venv_path)

pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "airflowctl"
-version = "0.2.4"
-description = "A lightweight Python package to simplify the installation of Apache Airflow effortlessly."
+version = "0.2.5"
+description = "A CLI tool to streamline getting started with Apache Airflow™ and managing multiple Airflow projects."
 authors = [
     "Kaxil Naik <[email protected]>",
 ]
@@ -33,6 +33,7 @@ rich = "^13.5.2"
 pyyaml = "^6.0.1"
 python-dotenv = ">=0.21.0"
 psutil = "^5.9.5"
+packaging = "^23.1"
 
 [tool.poetry.scripts]
 airflowctl = "airflowctl.cli:app"