| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| # Sets up the python path for impala-python. This is needed because tests and other |
| # utility scripts depend on some modules external to infra/python/env-*. |
| # TODO: we should try to reduce our reliance on PYTHONPATH if possible. |
| # |
| # Setting USE_THRIFT11_GEN_PY will add Thrift 11 Python generated code rather than the |
| # default Thrift Python code. |
| # Used to allow importing testdata, test, etc modules from other scripts. |
| |
| # ${IMPALA_HOME}/bin has bootstrap_toolchain.py, required by bootstrap_virtualenv.py |
| export PYTHONPATH=${IMPALA_HOME}:${IMPALA_HOME}/bin |
| |
| # Generated Thrift files are used by tests and other scripts. |
| if [ "${USE_THRIFT11_GEN_PY:-}" == "true" ]; then |
| PYTHONPATH=${PYTHONPATH}:${IMPALA_HOME}/shell/build/thrift-11-gen/gen-py |
| THRIFT_HOME="${IMPALA_TOOLCHAIN_PACKAGES_HOME}/thrift-${IMPALA_THRIFT11_VERSION}" |
| else |
| PYTHONPATH=${PYTHONPATH}:${IMPALA_HOME}/shell/gen-py |
| fi |
| |
| PYTHONPATH=${PYTHONPATH}:${IMPALA_HOME}/infra/python/env-gcc${IMPALA_GCC_VERSION}/lib |
| |
| # There should be just a single version of python that created the |
| # site-packages directory. We find it by performing shell independent expansion |
| # of the following pattern: |
| # ${THRIFT_HOME}/python/lib{64,}/python*/site-packages |
| # Note: this could go wrong if we have used two different versions of |
| # Python to build Thrift on this machine, and the first version is not |
| # compatible with the second. |
| for PYTHON_DIR in ${THRIFT_HOME}/python/lib{64,}; do |
| [[ -d ${PYTHON_DIR} ]] || continue |
| for PKG_DIR in ${PYTHON_DIR}/python*/site-packages; do |
| PYTHONPATH=${PYTHONPATH}:${PKG_DIR}/ |
| done |
| done |
| |
| # Add Hive after Thrift because Hive supplies its own Thrift modules |
| PYTHONPATH=${PYTHONPATH}:${HIVE_HOME}/lib/py |