error early if virtualenv is missing

apache · Aug 1, 2021 · 7bc1336 · 7bc1336
1 parent e8fc3ac
commit 7bc1336
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/airflow/operators/python.py b/airflow/operators/python.py
@@ -18,6 +18,7 @@
 import inspect
 import os
 import pickle
+import shutil
 import sys
 import types
 import warnings
@@ -325,6 +326,8 @@ def __init__(
                 "Passing op_args or op_kwargs is not supported across different Python "
                 "major versions for PythonVirtualenvOperator. Please use string_args."
             )
+        if not shutil.which("virtualenv"):
+            raise AirflowException('PythonVirtualenvOperator requires virtualenv, please install it.')
         super().__init__(
             python_callable=python_callable,
             op_args=op_args,