From 1c8a5da492c208861655cebaeda4b196237243e0 Mon Sep 17 00:00:00 2001
From: Wirawan Purwanto <wpurwant@turing.hpc.odu.edu>
Date: Wed, 27 Mar 2019 11:24:00 -0400
Subject: [PATCH] * Imported launch_jupyter from Turing.

  Originally furnished by Min Dong, 2019-03-01 14:30 EST.
---
 turing/calculations/launch_jupyter | 52 ++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100755 turing/calculations/launch_jupyter

diff --git a/turing/calculations/launch_jupyter b/turing/calculations/launch_jupyter
new file mode 100755
index 0000000..8acdb99
--- /dev/null
+++ b/turing/calculations/launch_jupyter
@@ -0,0 +1,52 @@
+#!/bin/bash -l
+
+jupyter_output=$(mktemp ~/.jupyter_output.XXXX)
+
+clean_up() {
+        job_id=$(awk '/srun: job/ {print $3;exit}' $jupyter_output)
+        scancel $job_id
+	rm $jupyter_output
+}
+
+trap clean_up EXIT
+
+set -m
+
+
+if [ x"$PYTHON_VER" = x ]; then
+        echo "Please load python and any needed python modules before launching juypter!"
+        echo "Example: "
+        echo "  enable_lmod"
+        echo "  module load python/3.6   # 2.7/3.7 is also avaiable"
+        exit 1
+fi
+
+echo " * loading jupyter module"
+
+enable_lmod
+module load python/$PYTHON_VER
+module load jupyter
+
+echo " * launching jupyter notebook server"
+(srun -n 1 -c 2 -t 240 -J juypter jupyter-notebook --no-browser --port=$((8000 + $RANDOM % 1000)) --ip=0.0.0.0 2>&1 | tee $jupyter_output > /dev/null) &
+
+echo 
+echo "   jupyter server will be on for 4 hour"
+echo "   please do not close this window, or jupyter will be terminated immediately"
+echo 
+
+
+sleep 10 
+
+job_id=$(awk     '/srun: job/ {print $3;exit}' $jupyter_output)
+tokens=$(awk -F: '/  http:/ {print $3}'        $jupyter_output)
+host=$(squeue -j $job_id -O NodeList -h | sed 's/ //g')
+url="http://$host:$tokens"
+
+echo " * launching browser"
+(firefox $url 2>&1 )>/dev/null &
+
+echo "   if you closed browser, please use below to connect back"
+echo "      $url"
+
+wait