1#!/usr/bin/env bash 2# Copyright (c) 2021-2024 Huawei Device Co., Ltd. 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14 15set -e 16 17MY_USERNAME=${SUDO_USER} 18if [[ -z "${VENV_DIR}" && -n "${MY_USERNAME}" ]]; then 19 MY_HOME=$(grep "^${MY_USERNAME}:" /etc/passwd | cut -d: -f6) 20 if [[ ! -e "${MY_HOME}" ]]; then 21 MY_HOME=/home/${MY_USERNAME} 22 fi 23 VENV_DIR=${MY_HOME}/.venv-panda 24elif [[ -z "${VENV_DIR}" && -z "${MY_USERNAME}" ]]; then 25 VENV_DIR=/root/.venv-panda 26fi 27 28function activate_venv() 29{ 30 if [[ -d "${VENV_DIR}" ]]; then 31 source "${VENV_DIR}/bin/activate" 32 echo "${VENV_DIR} activated" 33 fi 34} 35 36function deactivate_venv() 37{ 38 if [[ -d "${VENV_DIR}" && -n "${VIRTUAL_ENV}" ]]; then 39 deactivate 40 fi 41} 42