Use countdown widget in main notebook

home/widgets.py

@@ -1,7 +1,7 @@
 """AiiDAlab basic widgets."""
 
-from datetime import datetime
 import subprocess
+from datetime import datetime
 from threading import Timer
 
 import ipywidgets as ipw

start.ipynb

@@ -46,6 +46,26 @@
     "    parsed_url = urlparse.parse_qs(url.query)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from yaml import safe_load\n",
+    "\n",
+    "from home.widgets import CountDownWidget\n",
+    "\n",
+    "# For temporary deployments (e.g. demo server), duration is read from a local\n",
+    "# config file. This will initiate the countdown widget.\n",
+    "demo_server_config_file = Path.home() / \".aiidalab\" / \"demo-server-config.yml\"\n",
+    "if demo_server_config_file.exists():\n",
+    "    demo_server_config = safe_load(demo_server_config_file.read_text())\n",
+    "    countdown = CountDownWidget(duration=demo_server_config.get(\"duration\", \"12:00:00\"))\n",
+    "    countdown.start()\n",
+    "    display(countdown)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,