Improve error checking for astro dev upgrade-check (#396)

* improve error checking for `astro dev upgrade-check` * Update cmd/airflow.go Co-authored-by: Kaxil Naik <[email protected]> * Update cmd/airflow.go Co-authored-by: Greg Neiheisel <[email protected]> Co-authored-by: Kaxil Naik <[email protected]> Co-authored-by: Greg Neiheisel <[email protected]>
astronomer · Dec 11, 2020 · 748ca2e · 748ca2e
1 parent 70ef968
commit 748ca2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmd/airflow.go b/cmd/airflow.go
@@ -398,7 +398,7 @@ func airflowUpgradeCheck(cmd *cobra.Command, args []string) error {
 	cmd.SilenceUsage = true
 
 	// Add airflow command, to simplify astro cli usage
-	args = append([]string{"bash", "-c", "pip install --no-deps 'apache-airflow-upgrade-check'; airflow upgrade_check"})
+	args = append([]string{"bash", "-c", "pip install --no-deps 'apache-airflow-upgrade-check'; python -c 'from packaging.version import Version\nfrom airflow import __version__\nif Version(__version__) < Version(\"1.10.14\"):\n  print(\"Please upgrade your image to Airflow 1.10.14 first, then try again.\");exit(1)\nelse:\n  from airflow.upgrade.checker import __main__;__main__()'"})
 	return airflow.Run(config.WorkingPath, args, "root")
 }