import sys
import warnings
from urllib.request import urlretrieve

assert (
    "google.colab" in sys.modules
), "This script is intended for use on Google Colab only."

urlretrieve(
    "https://underactuated.csail.mit.edu/setup/setup_underactuated_colab.py",  # noqa
    "setup_underactuated_colab.py",
)

import setup_underactuated_colab as new_setup  # noqa

warnings.warn(
    "jupyter_setup.py is deprecated.  Please use"
    " setup_underactuated_colab.py instead."
)


def setup_drake():
    new_setup.setup_drake(version="0.27.0", build="release")


def setup_underactuated():
    new_setup.setup_underactuated(
        underactuated_sha="2a15bde1136495c41226a92dfdebf3b3b53cf830",
        drake_version="0.27.0",
        drake_build="release",
    )