init.sh

#!/bin/bash

set -ex

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

pushd "$(dirname "$0")"

UBUNTU_VERSION=$(lsb_release -rs)

verlte() {
     [  "$1" = "`echo -e "$1\n$2" | sort -V | head -n1`" ]
}

verlte $UBUNTU_VERSION '20' && OLDER_UBUNTU=1 || OLDER_UBUNTU=0

if [ $OLDER_UBUNTU == 1 ]; then
  echo "Old version of Ubuntu detected ($UBUNTU_VERSION), will register an additional apt repository to install latest version of Python"
  ADD_PYTHON39_APTREPO=1
else
  ADD_PYTHON39_APTREPO=0
fi

if [ ""$AML_CloudName"" != "" ]; then
  echo "Running on AML machine: skipping venv creation by default"
  CREATE_VENV=0
else
  CREATE_VENV=1
fi


while getopts "nr" opt; do
  case $opt in
    n)
      echo "Skipping venv creation. Parameters: [$OPTARG]"
      CREATE_VENV=0
      ;;
    r)
      echo "Will add apt repo to install latest version of Python"
      ADD_PYTHON39_APTREPO=1
      ;;
    \?)
      echo "Syntax: init.sh [-n]" >&2
      echo "   -n    skip creation of virtual environment" >&2
      echo "   -r    register required apt repository to install latest version of Python for older versions of Ubuntu (e.g. 16)" >&2
      exit 1
      ;;
  esac
done


SUDO=''
if (( $EUID != 0 )); then
    SUDO='sudo -E'
fi

if [ ! -z "${VIRTUAL_ENV}" ]; then
    echo 'Running under virtual environment, skipping installation of global packages';
else


    if [ "${ADD_PYTHON39_APTREPO}" == "1" ]; then
        echo 'Adding APT repo ppa:deadsnakes/ppa'
        $SUDO apt install software-properties-common -y
        $SUDO apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/3bf863cc.pub
        $SUDO add-apt-repository ppa:deadsnakes/ppa -y
    fi

    $SUDO apt update

    $SUDO apt-get install curl -y

    # install nodejs 12.0 (required by pyright typechecker)
    curl -sL https://deb.nodesource.com/setup_16.x | $SUDO bash -

    # install all apt-get dependencies
    if [ $OLDER_UBUNTU == 1 ]; then
        # exclude package not available on older ubuntu
        cat apt-requirements.txt | grep -v python3-distutils | $SUDO  xargs apt-get install -y
    else
        cat apt-requirements.txt | $SUDO xargs apt-get install -y
    fi

    # $SUDO npm -g upgrade node
    $SUDO npm install -g --unsafe-perm=true --allow-root npm

    # Make sure that the desired version of python is used
    # in the rest of the script and when calling pyright to
    # generate stubs
    $SUDO update-alternatives --install /usr/bin/python python /usr/bin/python3.9 2
    $SUDO update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 2
    $SUDO update-alternatives --set python /usr/bin/python3.9
    $SUDO update-alternatives --set python3 /usr/bin/python3.9
    update-alternatives --query python
    export PATH="/usr/bin:${PATH}"

    # install pyright
    ./pyright.sh --version

    # installing orca required to export images with plotly
    ./install-orca.sh

    # install pip
    if [ ! -f "/tmp/get-pip.py" ]; then
        curl https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py
    fi

    python --version

    python /tmp/get-pip.py

    if [ "${CREATE_VENV}" == "1" ]; then
        echo 'Install virtualenv'
        python -m pip install --user virtualenv

        # Create a virtual environment
        python -m venv venv

        source venv/bin/activate
    fi
fi


if ./install-pythonpackages.sh; then
    echo 'Package dependencies successfully installed'
else
    echo 'An error occured when installing package dependencies'
    exit 1
fi

if [ "${CREATE_VENV}" == "1" ]; then
  # Register the `venv`` with jupyter
  python -m ipykernel install --user --name=venv
fi

if [ ""$GITHUB_ACTION"" == "" ] && [ -d ".git" ]; then

  echo 'running under a git enlistment -> configure pre-commit checks on every `git push` to run pyright and co'
  pre-commit install -t pre-push
fi

./createstubs.sh

popd