add cleanup cell for demo notebook

cx1111 · cx1111 · commit 8ee9cd472dff · 2018-04-06T15:28:43.000-04:00
diff --git a/demo.ipynb b/demo.ipynb
@@ -376,9 +376,6 @@
     "# Make a temporary download directory in your current working directory\n",
     "cwd = os.getcwd()\n",
     "dl_dir = os.path.join(cwd, 'tmp_dl_dir')\n",
-    "# Make sure to use a new directory\n",
-    "while os.path.exists(dl_dir):\n",
-    "    dl_dir = dl_dir+'1'\n",
     "\n",
     "# Download all the WFDB content\n",
     "wfdb.dl_database('ahadb', dl_dir=dl_dir)\n",
@@ -525,6 +522,34 @@
     "comparitor.plot(title='xqrs detected qrs vs reference annotations')"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Cleanup for all demos\n",
+    "for file in ['100x.hea', '100.dat',  # demo 10\n",
+    "             'ecg-record.hea', 'ecg-record.dat',  # demo 11 \n",
+    "             'test01_00s_framex.hea', 'test01_00s.dat',  # demo 12\n",
+    "             '100.cpy',  # demo 13\n",
+    "             'b001.cpy',  # demo 14\n",
+    "            ]:\n",
+    "    if os.path.isfile(file):\n",
+    "        os.remove(file)\n",
+    "\n",
+    "dl_dir = os.path.join(cwd, 'tmp_dl_dir')  # demo 17, 18\n",
+    "if os.path.isdir(dl_dir):\n",
+    "    shutil.rmtree(dl_dir)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
