This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new f47d7b9  Allow specifying extras when using breeze 
initialize_local_virtualenv (#19178)
f47d7b9 is described below

commit f47d7b95fe68fe4c0c9db6503ddbf2ed13ea43dd
Author: Niko <[email protected]>
AuthorDate: Fri Oct 22 17:30:33 2021 -0700

    Allow specifying extras when using breeze initialize_local_virtualenv 
(#19178)
---
 LOCAL_VIRTUALENV.rst                    | 7 +++++++
 breeze                                  | 5 ++++-
 scripts/ci/libraries/_initialization.sh | 6 ++++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/LOCAL_VIRTUALENV.rst b/LOCAL_VIRTUALENV.rst
index e96fd6a..bf982f2 100644
--- a/LOCAL_VIRTUALENV.rst
+++ b/LOCAL_VIRTUALENV.rst
@@ -204,6 +204,13 @@ Activate your virtualenv, e.g. by using ``workon``, and 
once you are in it, run:
 
   ./breeze initialize-local-virtualenv
 
+By default Breeze installs the ``devel`` extra only. You can optionally 
control which extras are installed by exporting ``VIRTUALENV_EXTRAS`` before 
calling Breeze:
+
+.. code-block:: bash
+
+  export VIRTUALENV_EXTRAS="devel,google,postgres"
+  ./breeze initialize-local-virtualenv
+
 5. (optionally) run yarn build if you plan to run the webserver
 
 .. code-block:: bash
diff --git a/breeze b/breeze
index d85b17c..d332271 100755
--- a/breeze
+++ b/breeze
@@ -230,6 +230,7 @@ function breeze::setup_default_breeze_constants() {
 #    PYTHON_MAJOR_MINOR_VERSION
 #    AIRFLOW_HOME_DIR
 #    AIRFLOW_SOURCES
+#    VIRTUALENV_EXTRAS
 #    DEFAULT_CONSTRAINTS_BRANCH
 #    OSTYPE
 #
@@ -252,13 +253,15 @@ function breeze::initialize_virtualenv() {
         echo
         echo "Initializing the virtualenv: $(command -v python)!"
         echo
+        echo "Extras to be installed: ${VIRTUALENV_EXTRAS}"
+        echo
         echo "This will wipe out ${AIRFLOW_HOME_DIR} and reset all the 
databases!"
         echo
         "${AIRFLOW_SOURCES}/confirm" "Proceeding with the initialization"
         echo
         pushd "${AIRFLOW_SOURCES}" >/dev/null 2>&1 || exit 1
         set +e
-        pip install -e ".[devel]" \
+        pip install -e ".[${VIRTUALENV_EXTRAS}]" \
             --constraint 
"https://raw.githubusercontent.com/${CONSTRAINTS_GITHUB_REPOSITORY}/${DEFAULT_CONSTRAINTS_BRANCH}/constraints-source-providers-${PYTHON_MAJOR_MINOR_VERSION}.txt";
         res=$?
         set -e
diff --git a/scripts/ci/libraries/_initialization.sh 
b/scripts/ci/libraries/_initialization.sh
index cad80ff..0a14af7 100644
--- a/scripts/ci/libraries/_initialization.sh
+++ b/scripts/ci/libraries/_initialization.sh
@@ -564,6 +564,11 @@ function initialization::initialize_kubernetes_variables() 
{
     readonly API_SERVER_PORT
 }
 
+function initialization::initialize_virtualenv_variables() {
+    # The extras to install when initializing a virtual env with breeze
+    export VIRTUALENV_EXTRAS=${VIRTUALENV_EXTRAS:="devel"}
+}
+
 function initialization::initialize_git_variables() {
     # SHA of the commit for the current sources
     COMMIT_SHA="$(git rev-parse HEAD 2>/dev/null || echo "Unknown")"
@@ -638,6 +643,7 @@ function initialization::initialize_common_environment() {
     initialization::initialize_image_build_variables
     initialization::initialize_provider_package_building
     initialization::initialize_kubernetes_variables
+    initialization::initialize_virtualenv_variables
     initialization::initialize_git_variables
     initialization::initialize_github_variables
     initialization::initialize_test_variables

Reply via email to