From 854878a59276ccea1e77e4b0afc3415ca6fd2735 Mon Sep 17 00:00:00 2001
From: Frank Knoll <Knoll_Frank@web.de>
Date: Sat, 27 Jul 2024 18:33:22 +0200
Subject: [PATCH] refactoring

---
 ...DrugDescriptionsForPathologiesPersister.py | 13 +++++++++++
 src/HowBadIsMyBatch.ipynb                     | 23 +------------------
 2 files changed, 14 insertions(+), 22 deletions(-)
 create mode 100644 src/DrugsForPathologies/DrugDescriptionsForPathologiesPersister.py

diff --git a/src/DrugsForPathologies/DrugDescriptionsForPathologiesPersister.py b/src/DrugsForPathologies/DrugDescriptionsForPathologiesPersister.py
new file mode 100644
index 00000000000..ada1a3cf44b
--- /dev/null
+++ b/src/DrugsForPathologies/DrugDescriptionsForPathologiesPersister.py
@@ -0,0 +1,13 @@
+import shutil
+from IOUtils import IOUtils
+
+def saveDrugDescriptionsForPathologies(drugDescrByPathology, directory):
+    shutil.rmtree(directory, ignore_errors = True)
+    filenameByPathology = {}
+    i = 0
+    for pathology, drugDescr in drugDescrByPathology.items():
+        i += 1
+        filenameByPathology[pathology] = f'{i}'
+        drugDescr['PATHOLOGY'] = pathology
+        IOUtils.saveDictAsJson(drugDescr, f'{directory}/{i}.json')
+    return filenameByPathology    
diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb
index 636ef91a04b..399d9b1dc84 100644
--- a/src/HowBadIsMyBatch.ipynb
+++ b/src/HowBadIsMyBatch.ipynb
@@ -623,6 +623,7 @@
    "source": [
     "import pandas as pd\n",
     "import os\n",
+    "from DrugsForPathologies.DrugDescriptionsForPathologiesPersister import saveDrugDescriptionsForPathologies\n",
     "from DrugsForPathologies.HtmlUpdater import updateHtmlFile as updateDrugsForPathologiesHtmlFile\n"
    ]
   },
@@ -685,28 +686,6 @@
     "drugs4Pathologies[drugs4Pathologies['PATHOLOGY'] == 'bone cancer']"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4444d234",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import shutil\n",
-    "from IOUtils import IOUtils\n",
-    "\n",
-    "def saveDrugDescriptionsForPathologies(drugDescrByPathology, directory):\n",
-    "    shutil.rmtree(directory, ignore_errors = True)\n",
-    "    filenameByPathology = {}\n",
-    "    i = 0\n",
-    "    for pathology, drugDescr in drugDescrByPathology.items():\n",
-    "        i += 1\n",
-    "        filenameByPathology[pathology] = f'{i}'\n",
-    "        drugDescr['PATHOLOGY'] = pathology\n",
-    "        IOUtils.saveDictAsJson(drugDescr, f'{directory}/{i}.json')\n",
-    "    return filenameByPathology    \n"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,