Merge pull request #17 from tmckayus/addchecks

Add a check for Python3 and validate NGC api key (with helm)

cloud-scripts/build-cuopt-server.sh

@@ -21,17 +21,37 @@
 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 # DEALINGS IN THE SOFTWARE.
 
+if ! command -v python3 &> /dev/null; then
+   echo Python3 must be installed and on your path
+   exit 1
+fi
+
 DIR=$(dirname $(realpath "$0"))
 if [ "$API_KEY" == "" ]; then
     read -sp 'Enter a NGC api-key to access cuOpt resources: ' API_KEY
 fi
 
+if command -v helm &> /dev/null; then
+    echo helm installed, checking API_KEY
+    rm -f /tmp/cuopt-22.10.1.tgz
+    helm fetch https://helm.ngc.nvidia.com/nvidia/cuopt/charts/cuopt-22.10.1.tgz --username='$oauthtoken' --password=$API_KEY -d /tmp
+    if [ "$?" -eq 0 ]; then
+        rm -f /tmp/cuopt-22.10.1.tgz
+        echo API_KEY is valid
+    else
+        echo Failed to download cuopt helm chart, API_KEY is invalid. Please try again with a valid NGC api key.
+        exit 1
+    fi
+else
+    echo helm is not installed, skipping API_KEY check
+fi
+
 TF_VAR_api_key=$API_KEY terraform apply --auto-approve
 if [ "$?" -ne 0 ]; then
     exit -1
 fi
 terraform output --json outputs > values.json
-$DIR/utilities/parse.py values.json values.sh
+python3 $DIR/utilities/parse.py values.json values.sh
 source values.sh
 
 nmsg="The address of the cuOpt notebook server is $ip:30001"

cloud-scripts/utilities/parse.py

@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python3
 
 # SPDX-FileCopyrightText: Copyright (c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: MIT